http://blog.csdn.net/yichenlian/article/details/37055393



由于系统有根据方法名实现动态切换数据源的需求，所以研究了下spring的动态切换数据源方式，然后和mybatis集成。

主要使用到了c3p0连接池、mybatis、spring（aop切面）、自定义拦截器

      1.首先定义数据源，我的数据源很多，只拿几个做例子；

  <context:property-placeholder ignore-unresolvable="true"          

  location="classpath*:/conf/dbpool.properties" /> 
        <bean id="dataSourceUserCenter_master" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close"> 
            <property name="driverClass"> 
                <value>${driverClassName}</value> 
            </property> 
            <property name="jdbcUrl"> 
                <value>${usercenter_master}</value> 
            </property> 
            <property name="user"> 
                <value>${username}</value> 
            </property> 
            <property name="password"> 
                <value>${password}</value> 
            </property> 
            <!--连接池中保留的最小连接数。Default: 5 --> 
            <property name="minPoolSize"> 
                <value>${minPoolSize}</value> 
            </property> 
            <!--连接池中保留的最大连接数。Default: 15 --> 
            <property name="maxPoolSize"> 
                <value>${maxPoolSize}</value> 
            </property> 
            <!--初始化时获取的连接数，取值应在minPoolSize与maxPoolSize之间。Default: 10 --> 
            <property name="initialPoolSize"> 
                <value>${initialPoolSize}</value> 
            </property> 
            <!--当连接池中的连接耗尽的时候c3p0一次同时获取的连接数。Default: 3 --> 
            <property name="acquireIncrement"> 
                <value>${acquireIncrement}</value> 
            </property> 
            <!--最大空闲时间,60秒内未使用则连接被丢弃。若为0则永不丢弃。Default: 0 --> 
            <property name="maxIdleTime"> 
                <value>${maxIdleTime}</value> 
            </property> 
            <!--JDBC的标准参数，用以控制数据源内加载的PreparedStatements数量。但由于预缓存的statements 
              属于单个connection而不是整个连接池。所以设置这个参数需要考虑到多方面的因素。 
              如果maxStatements与maxStatementsPerConnection均为0，则缓存被关闭。Default: 0--> 
            <property name="maxStatements"> 
                <value>${maxStatements}</value> 
            </property> 
            <!--每60秒检查所有连接池中的空闲连接。Default: 0 --> 
            <property name="idleConnectionTestPeriod"> 
                <value>${idleConnectionTestPeriod}</value> 
            </property> 
            <!--定义在从数据库获取新连接失败后重复尝试的次数。Default: 30 --> 
            <property name="acquireRetryAttempts"> 
                <value>${acquireRetryAttempts}</value> 
            </property> 
            <!--获取连接失败将会引起所有等待连接池来获取连接的线程抛出异常。但是数据源仍有效 
              保留，并在下次调用getConnection()的时候继续尝试获取连接。如果设为true，那么在尝试 
              获取连接失败后该数据源将申明已断开并永久关闭。Default: false--> 
            <property name="breakAfterAcquireFailure"> 
                <value>${breakAfterAcquireFailure}</value> 
            </property> 
            <!--因性能消耗大请只在需要的时候使用它。如果设为true那么在每个connection提交的 
              时候都将校验其有效性。建议使用idleConnectionTestPeriod或automaticTestTable 
              等方法来提升连接测试的性能。Default: false --> 
            <property name="testConnectionOnCheckout"> 
                <value>${testConnectionOnCheckout}</value> 
            </property> 
        </bean> 
         
        <bean id="dataSourceUserCenter_slave" parent="dataSourceUserCenter_master"> 
            <property name="jdbcUrl"> 
                <value>${usercenter_slave}</value> 
            </property> 
            <property name="user"> 
                <value>${username}</value> 
            </property> 
            <property name="password"> 
                <value>${password}</value> 
            </property> 
        </bean> 
         
         <bean id="dataSourceTest_slave" parent="dataSourceUserCenter_master"> 
            <property name="jdbcUrl"> 
                <value>${test_slave}</value> 
            </property> 
            <property name="user"> 
                <value>${username}</value> 
            </property> 
            <property name="password"> 
                <value>${password}</value> 
            </property> 
        </bean> 

2.对数据源进行整合包装，便于统一管理。

 <bean id="multipleDataSource" class="com.test.interceptor.datasource.MultipleDataSource"> 
    <property name="defaultTargetDataSource" ref="dataSourceUserCenter_master"/> 
    <property name="targetDataSources"> 
        <map> 
            <!-- 需要注意这里的key值与实现类里面的常量值要一样 --> 
            <entry value-ref="dataSourceUserCenter_master" key="UserCenter_master"/> 
            <entry value-ref="dataSourceUserCenter_slave" key="UserCenter_slave"/> 
            <entry value-ref="dataSourceTest_slave" key="Test_slave"/> 
        </map> 
    </property> 
</bean>

其中

com.test.interceptor.datasource.MultipleDataSource

实现了spring的AbstractRoutingDataSource类，故需用切换数据源的能力。defaultTargetDataSource是默认数据源、targetDataSources是所有数据源集合，默认数据源也要放置进来。其中每个entry的key要与java类中的常量相同。

3.和mybatis集成

<bean id="sqlSessionFactory"
          class="org.mybatis.spring.SqlSessionFactoryBean"
          p:dataSource-ref="multipleDataSource"
          p:configLocation="classpath:conf/mybatis-config.xml"
          p:mapperLocations="classpath*:mapper/*.xml"/>

4.aop指向进行数据源切换的处理过滤器

<!-- 启用aop注解 -->
 <aop:aspectj-autoproxy></aop:aspectj-autoproxy>

    <!-- 配置一个拦截器对象，处理具体的切换数据源的业务 -->
    <bean id="dataSourceMethodInterceptor" class="com.test.interceptor.DataSourceMethodInterceptor"/>
   
    <!-- 参与动态切换数据源的切入点对象 (切入点对象，确定何时何地调用拦截器) -->
    <bean id="methodCachePointCut" class="org.springframework.aop.support.RegexpMethodPointcutAdvisor">
        <!-- 配置缓存aop切面 -->
        <property name="advice" ref="dataSourceMethodInterceptor" />
        <!-- .*表示前面的前缀(包括包名) 表示print方法-->
        <property name="patterns">
            <list>
            <value>com.test.*.service.impl.*Service*\.*.*</value>
            </list>
        </property>
    </bean>

如上配置，扫描所有Service类中的方法，然后根据一定的规范来决定切换成哪个数据源。

  5.MultipleDataSource.java

public class MultipleDataSource extends AbstractRoutingDataSource{

      @Override
     protected Object determineCurrentLookupKey() {
     return DataSourceContextHandler.getDataSourceContext();
    }

}

   6.DataSourceContextHandler.java

/**
  * 线程局部变量，只为该线程服务
  */
 private final static ThreadLocal<String> contextHandler = new ThreadLocal<String>();
 
 /**
  * 设置该线程数据源id
  * @param dataSourceType
  */
 public static void setDataSourceContext(String dataSourceType){
  contextHandler.set(dataSourceType);
 }
 /**
  * 得到该线程数据源id
  * @return
  */
 public static String getDataSourceContext(){
  return contextHandler.get();
 }
 /**
  * 清除该线程的数据源变量值
  */
 public static void clearDataSourceContext(){
  contextHandler.remove();
 }

 7.DataSourceMethodInterceptor.java

        @Override
 public void afterPropertiesSet() throws Exception {
  logger.info("after properties set ...");
 }

 @Override
 public Object invoke(MethodInvocation invocation) throws Throwable {
  String dbname=....
        if (StringUtils.isNotBlank(dbname)) {
         if(isSlave){
          DataSourceContextHandler.setDataSourceContext(dbname+"_slave");
         }else{
          DataSourceContextHandler.setDataSourceContext(dbname+"_master");
         }
        }else{
         DataSourceContextHandler.clearDataSourceContext();
        }
        Object result = invocation.proceed();
        return result;
 }

在invoke方法中根据自定义的规则完成了数据源的选择。