最近一个项目用到了多个数据库,所以需要实现动态切换数据源来查询数据,http://www.cnblogs.com/lzrabbit/p/3750803.html这篇文章让我受益匪浅,提供了一种自动切换数据源的思路,但这种方式不支持事务,所以我进一步改进了这个方案,下面直入正题
多数据源配置:
#============================================================================ # DataBaseOne #============================================================================ jdbc.one.driver=com.mysql.jdbc.Driver jdbc.one.url=jdbc:mysql://127.0.0.1:3306/DataBaseOne?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true jdbc.one.username=root jdbc.one.password=root ============================================================================= #============================================================================ # DataBaseTwo #============================================================================ jdbc.two.driver=com.mysql.jdbc.Driver jdbc.two.url=jdbc:mysql://127.0.0.1:3306/DataBaseTwo?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true jdbc.two.username=root jdbc.two.password=root ============================================================================= #============================================================================ # DataBaseThree #============================================================================ jdbc.three.driver=com.mysql.jdbc.Driver jdbc.three.url=jdbc:mysql://127.0.0.1:3306/DataBaseThree?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true jdbc.three.username=root jdbc.mysql.password=root ============================================================================= #============================================================================ # 通用配置 #============================================================================ jdbc.initialSize=5 jdbc.minIdle=5 jdbc.maxIdle=20 jdbc.maxActive=100 jdbc.maxWait=100000 jdbc.defaultAutoCommit=false jdbc.removeAbandoned=true jdbc.removeAbandonedTimeout=600 jdbc.testWhileIdle=true jdbc.timeBetweenEvictionRunsMillis=60000 jdbc.numTestsPerEvictionRun=20 jdbc.minEvictableIdleTimeMillis=300000
Spring中使用多数据源:
<?xml version="1.0" encoding="UTF-8"?> <beans xmlns="http://www.springframework.org/schema/beans" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:context="http://www.springframework.org/schema/context" xmlns:aop="http://www.springframework.org/schema/aop" xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.0.xsd http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.0.xsd http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.0.xsd"> <!-- 自己根据情况补全其他配置,以下只提供了数据源配置 --> <!-- 多数据源配置 --> <bean id="propertyConfigurer" class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer"> <property name="location" value="classpath:jdbc.properties"/> </bean> <!-- 第一个数据源dataSourceOne --> <bean id="dataSourceOne" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close"> <property name="driverClassName" value="${jdbc.one.driver}"/> <property name="url" value="${jdbc.one.url}"/> <property name="username" value="${jdbc.one.username}"/> <property name="password" value="${jdbc.one.password}"/> <property name="initialSize" value="${jdbc.initialSize}"/> <property name="minIdle" value="${jdbc.minIdle}"/> <property name="maxIdle" value="${jdbc.maxIdle}"/> <property name="maxActive" value="${jdbc.maxActive}"/> <property name="maxWait" value="${jdbc.maxWait}"/> <property name="defaultAutoCommit" value="${jdbc.defaultAutoCommit}"/> <property name="removeAbandoned" value="${jdbc.removeAbandoned}"/> <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}"/> <property name="testWhileIdle" value="${jdbc.testWhileIdle}"/> <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}"/> <property name="numTestsPerEvictionRun" value="${jdbc.numTestsPerEvictionRun}"/> <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}"/> </bean> <!-- 第二个数据源dataSourceTwo --> <bean id="dataSourceTwo" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close"> <property name="driverClassName" value="${jdbc.two.driver}"/> <property name="url" value="${jdbc.two.url}"/> <property name="username" value="${jdbc.two.username}"/> <property name="password" value="${jdbc.two.password}"/> <property name="initialSize" value="${jdbc.initialSize}"/> <property name="minIdle" value="${jdbc.minIdle}"/> <property name="maxIdle" value="${jdbc.maxIdle}"/> <property name="maxActive" value="${jdbc.maxActive}"/> <property name="maxWait" value="${jdbc.maxWait}"/> <property name="defaultAutoCommit" value="${jdbc.defaultAutoCommit}"/> <property name="removeAbandoned" value="${jdbc.removeAbandoned}"/> <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}"/> <property name="testWhileIdle" value="${jdbc.testWhileIdle}"/> <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}"/> <property name="numTestsPerEvictionRun" value="${jdbc.numTestsPerEvictionRun}"/> <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}"/> </bean> <!-- 第三个数据源dataSourceThree --> <bean id="dataSourceThree" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close"> <property name="driverClassName" value="${jdbc.three.driver}"/> <property name="url" value="${jdbc.three.url}"/> <property name="username" value="${jdbc.three.username}"/> <property name="password" value="${jdbc.three.password}"/> <property name="initialSize" value="${jdbc.initialSize}"/> <property name="minIdle" value="${jdbc.minIdle}"/> <property name="maxIdle" value="${jdbc.maxIdle}"/> <property name="maxActive" value="${jdbc.maxActive}"/> <property name="maxWait" value="${jdbc.maxWait}"/> <property name="defaultAutoCommit" value="${jdbc.defaultAutoCommit}"/> <property name="removeAbandoned" value="${jdbc.removeAbandoned}"/> <property name="removeAbandonedTimeout" value="${jdbc.removeAbandonedTimeout}"/> <property name="testWhileIdle" value="${jdbc.testWhileIdle}"/> <property name="timeBetweenEvictionRunsMillis" value="${jdbc.timeBetweenEvictionRunsMillis}"/> <property name="numTestsPerEvictionRun" value="${jdbc.numTestsPerEvictionRun}"/> <property name="minEvictableIdleTimeMillis" value="${jdbc.minEvictableIdleTimeMillis}"/> </bean> <!-- 使用自己实现的数据源实现类MultipleDataSource,这个类随意放在哪个包下都行 --> <bean id="multipleDataSource" class="com.cnblogs.datasource.MultipleDataSource"> <!-- 设置默认的数据源 --> <property name="defaultTargetDataSource" ref="dataSourceOne"/> <property name="targetDataSources"> <map> <!-- 这个key是对应数据源的别称,通过这个key可以找到对应的数据源,value-ref就是上面数据源的id --> <entry key="dataSourceOneKey" value-ref="dataSourceOne"/> <entry key="dataSourceTwoKey" value-ref="dataSourceTwo"/> <entry key="dataSourceThreeKey" value-ref="dataSourceThree"/> </map> </property> </bean> <!-- 让spring使用我们配置的多数据源 --> <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"> <property name="dataSource" ref="multipleDataSource"/> </bean> <!-- mybatis.spring自动映射 --> <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer"> <property name="basePackage" value="com.cnblogs.mapper"/> </bean> <!-- 自动扫描,多个包以 逗号分隔 --> <context:component-scan base-package="com.cnblogs.**"/> <!-- AOP配置(事务控制) --> <aop:config> <!--pointcut元素定义一个切入点,execution中的第一个星号 用以匹配方法的返回类型, 这里星号表明匹配所有返回类型。 com.abc.service.*.*(..)表明匹配com.abc.service包下的所有类的所有方法 --> <aop:pointcut id="myPointcut" expression="execution(* com.cnblogs.service.*.*(..))" /> <!--将定义好的事务处理策略应用到上述的切入点 --> <aop:advisor advice-ref="txAdvice" pointcut-ref="myPointcut" /> </aop:config> </beans>
MultipleDataSource.java实现:
package com.cnblogs.datasource; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; /** 多数据源java实现 */ public class MultipleDataSource extends AbstractRoutingDataSource { private static final ThreadLocal<String> dataSourceKey = new InheritableThreadLocal<String>(); public static void setDataSourceKey(String dataSource) { dataSourceKey.set(dataSource); } @Override protected Object determineCurrentLookupKey() { return dataSourceKey.get(); } }
下面详解使用SpringAOP来实现数据源切换,并支持事务控制
上面我们配置的AOP是针对service层的,所以在调用service层的任何方法时都会经过AOP,因此我们就在AOP中先于service调用时把数据源切换,看代码
新建类MultipleDataSourceAspectAdvice,把MultipleDataSourceAspectAdvice.java放在spring能自动注入的包中,比如controller包,或者自己把这个类所在的包加入到component-scan配置中去,总之要要让spring能自动加载
package com.cnblogs.controller; import org.apache.log4j.Logger; import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.annotation.Around; import org.aspectj.lang.annotation.Aspect; import org.aspectj.lang.annotation.Pointcut; import org.springframework.core.annotation.Order; import org.springframework.stereotype.Component; import com.cnblogs.datasource.MultipleDataSource; import com.cnblogs.service.DataBaseOne; import com.cnblogs.service.DataBaseTwo; import com.cnblogs.service.DataBaseThree; /** * 数据库链接自动切换AOP处理 * Order优先级设置到最高,因为在所有service方法调用前都必须把数据源确定 * Order数值越小优先级越高 */ @Component @Aspect @Order(1) public class MultipleDataSourceAspectAdvice { private static final Logger LOGGER = Logger.getLogger(MultipleDataSourceAspectAdvice.class); public MultipleDataSourceAspectAdvice() { LOGGER.info("MultipleDataSourceAspectAdvice 加载成功"); } /** * 定义切面 */ @Pointcut("execution(* com.cnblogs.service.*.*(..))") public void pointCut() { } // dataSourceOneKey // dataSourceTwoKey // dataSourceThreeKey @Around("pointCut()") public Object doAround(ProceedingJoinPoint jp) throws Throwable { if (jp.getTarget() instanceof DataBaseOne) { LOGGER.debug("使用数据库链接:dataSourceOneKey"); MultipleDataSource.setDataSourceKey("dataSourceOneKey"); } else if (jp.getTarget() instanceof DataBaseTwo) { LOGGER.debug("使用数据库链接:dataSourceTwoKey"); MultipleDataSource.setDataSourceKey("dataSourceTwoKey"); } else if (jp.getTarget() instanceof DataBaseThree) { LOGGER.debug("使用数据库链接:dataSourceThreeKey"); MultipleDataSource.setDataSourceKey("dataSourceThreeKey"); } else { // 默认是dataSourceOneKey LOGGER.debug("使用数据库链接:dataSourceOneKey"); MultipleDataSource.setDataSourceKey("dataSourceOneKey"); } return jp.proceed(); } }
到这里我们所以的spring多数据源配置已经完毕,那如何在执行service方法时让service切换到正确的数据库呢?上面的类中定义了有3个类DataBaseOne,DataBaseTwo,DataBaseThree,这3个类其实只是一个interface,没有任何实现方法,我们让具体业务的service都继承至这3个类以区分不同的service对应不同的数据源,因为业务的service我是知道他用的哪个数据源的,比如FooService继承至DataBaseOne,则在使用FooService任何方法时AOP就会先把数据源切换到dataSourceOneKey,以此就达到了自动切换数据源的目的,并且支持事务,下面看代码:
package com.cnblogs.service; /** * DataBaseOne数据库占位类 * 详情参见 MultipleDataSourceAspectAdvice 类 */ public interface DataBaseOne { }
DataBaseTwo,DataBaseThree与这完全相同
至此多数据源就配置完毕,可以安心写业务逻辑了
原理部分博友http://www.cnblogs.com/lzrabbit/p/3750803.html讲的十分清楚,再次感谢