spring多数据源的切换,主要用到的是AbstractRoutingDataSource这个路由类,当我们的自定义的一个路由分发类继承AbstractRoutingDataSource类后,重写determineCurrentLookupKey()这个方法,重写的内容就是我们的分发规则。那么spring在需要选择数据源的时候,就会执行这个方法,然后根据我们的自定义的规则自动进行分发,从而实现多数据源的切换。

  步骤如下:

  1、配置多个数据源

 <!-- 
  <bean id="c3p0DataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource">
    <property name="driverClass" value="${jdbc.driverClass}" />
    <property name="jdbcUrl" value="${jdbc.jdbcUrl}" />
    <property name="user" value="${jdbc.user}" />
    <property name="password" value="${jdbc.password}" />
    <property name="initialPoolSize" value="${c3p0.initialPoolSize}" />
    <property name="minPoolSize" value="${c3p0.minPoolSize}" />
    <property name="maxPoolSize" value="${c3p0.maxPoolSize}" />
    <property name="maxStatements" value="${c3p0.maxStatements}" />
    <property name="checkoutTimeout" value="${c3p0.checkoutTimeout}" />
    <property name="maxIdleTime" value="${c3p0.maxIdleTime}" />
    <property name="idleConnectionTestPeriod" value="${c3p0.idleConnectionTestPeriod}" />
  </bean>
 -->

  <bean id="druidDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close"> <property name="url" value="${jdbc.jdbcUrl}" /> <property name="username" value="${jdbc.user}" /> <property name="password" value="${jdbc.password}" /> <property name="initialSize" value="${druid.initialSize}" /> <property name="minIdle" value="${druid.minIdle}" /> <property name="maxActive" value="${druid.maxActive}" /> <property name="maxWait" value="${druid.maxWait}" /> <property name="timeBetweenEvictionRunsMillis" value="${druid.timeBetweenEvictionRunsMillis}" /> <property name="minEvictableIdleTimeMillis" value="${druid.minEvictableIdleTimeMillis}" /> <property name="validationQuery" value="${druid.validationQuery}" /> <property name="testWhileIdle" value="${druid.testWhileIdle}" /> <property name="testOnBorrow" value="${druid.testOnBorrow}" /> <property name="testOnReturn" value="${druid.testOnReturn}" /> <property name="poolPreparedStatements" value="${druid.poolPreparedStatements}" /> <property name="maxPoolPreparedStatementPerConnectionSize" value="${druid.maxPoolPreparedStatementPerConnectionSize}" /> <property name="filters" value="${druid.filters}" /> </bean>
<bean id="druidDataSource_1" parent="druidDataSource"> <property name="url" value="${jdbc.jdbcUrl1}" /> </bean>

  2、配置路由器,注意这个odd和even,他就是我们数据源的标识,在AbstractRoutingDataSource路由determineCurrentLookupKey()这个方法中返回哪个,就是用其对应的数据源。如返回"odd"就是用druidDataSource数据源。

<bean id="dataSourcreRouter" class="cn.yyh.router.NMRoutingDataSource">
      <property name="targetDataSources">
            <map>
                <entry key="odd" value-ref="druidDataSource" />
                <entry key="even" value-ref="druidDataSource_1" />
            </map>
      </property>
      <property name="defaultTargetDataSource" ref="druidDataSource" />
</bean>
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"> <property name="dataSource" ref="dataSourcreRouter" /> </bean>

  3、编写AbstractRoutingDataSource路由分发器,这个NMRoutingToken是什么叻?这个NMRoutingToken是一个令牌,他是存储在ThreadLocal中的一个对象。而ThreadLocal其实就是当前线程内的一个map对象。

public class NMRoutingDataSource extends AbstractRoutingDataSource {

    protected Object determineCurrentLookupKey() {
         NMRoutingToken token = NMRoutingToken.getCurrentToken();
        if (token != null) {
            String dataSourceName = token.getDataSourceName();
            // 解除令牌
            NMRoutingToken.unbindToken();
            return dataSourceName;
        }
        return null;
    }

}

  4、可以看到,其实NMRoutingToken就是在当前线程中的ThreadLocal存储一个字符串,前面我们就说过determineCurrentLookupKey()返回什么,spring就自动选择其对应的数据源。那么如果我们new NMRoutingToken("odd"),他就会绑定"odd"到ThreadLocal中,然后在determineCurrentLookupKey()取出来返回就可以了。

public class NMRoutingToken {

    private static ThreadLocal<NMRoutingToken> token = new ThreadLocal<NMRoutingToken>();
    private String dataSourceName;
    
    public String getDataSourceName() {
        return dataSourceName;
    }

    public NMRoutingToken(String dataSourceName) {
        super();
        this.dataSourceName = dataSourceName;
        bindToken(this);
    }

    /**
     * 绑定令牌对象到当前线程
     */
    private static void bindToken(NMRoutingToken t) {
        token.set(t);
    }

    /**
     * 解除与当前线程绑定的令牌
     */
    public static void unbindToken() {
        token.remove();
    }

    /**
     * 取得与当前线程绑定的令牌
     */
    public static NMRoutingToken getCurrentToken() {
        return token.get();
    }

}

  5、测试

public class NMDataSourceTest extends BaseTest {
    @Autowired
    private UserMapper userMapper;
    @Autowired
    private IUserService userService;

    @Test
    /**
     * 未开启事务
     */
    public void testMapper() {
        /**
         * 不配置任何数据库,使用默认数据库
         */
        int count1 = userMapper.getCount();
        System.out.println(count1);

        /**
         * 配置任何数据库odd
         */
        NMRoutingToken token = new NMRoutingToken("odd");
        count1 = userMapper.getCount();
        System.out.println(count1);

        /**
         * 配置任何数据库even
         */
        token = new NMRoutingToken("even");
        count1 = userMapper.getCount();
        System.out.println(count1);

        /**
         * 乱配将使用默认数据库
         */
        token = new NMRoutingToken("asdf");
        count1 = userMapper.getCount();
        System.out.println(count1);
    }

    @Test
    /**
     * 开启事务
     * 注意,由于tomcat容器不支持跨库事务,所以在选择数据源,开启事务后就会挂起,
     * 就算再次绑定令牌,但它不会再执行NMRoutingDataSource的determineCurrentLookupKey()方法了,
     * 所以后续选择数据源是无效的,默认还是第一次选择的数据源
     */
    public void testService() {
        userService.getNMCount();
    }
}
@Transactional(readOnly = true)
    public void getNMCount() {
        int count1 = -1;
        /**
         * 配置数据库even
         */
        new NMRoutingToken("even");
        count1 = userMapper.getCount();
        System.out.println(count1);
        
        /**
         * 配置数据库odd无效,默认还是使用even数据源
         */
        new NMRoutingToken("odd");
        count1 = userMapper.getCount();
        System.out.println(count1);
    }

实例下载:示例下载

注:此处的令牌里面存储的是一个字符串,此为最简单的分发规则,我们自己手动选择,如果需要根据对象的某些属性进行自动选择,那么可以在令牌中存储对应的对象,然后在路由中取出对应的属性,如id,然后根据我们的规则,动态返回,这样就不需要我们自己指定了。

注意:tomcat等容器不支持跨库事务,所以只能针对单个service进行事务控制。
不能在同一个开启事务的service中同时操作多个数据库。
如:
@Tranctional
AService(){
    methodA(){
        dataSourceA.insert();
        dataSourceB.update();
    }
}
AService(){
    @Tranctional
    methodA(){
        dataSourceA.insert();
        dataSourceB.update();
    }
}
此类写法(在类或方法上开启事务,同时操作多个数据库)都是不允许的,原理请见src/test/java/ResourceTest.java


只能跟别进行事务控制:
Service(){
    aService.insert();
    bService.update();
}
AService{
    @Tranctional
    insert();
}
BService{
    @Tranctional
    update();
}