spring dataSourceRouter自动切换数据源

　　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();
}