spring+mybitas 实现多数据源动态切换

1 由于项目需要，需要将不同来源的数据存入不同的数据库，所以需要根据入参的信息动态切换数据源

项目core层采用了spring+mybitas 方式进行架构

1 在spring的配置中定义两个数据源

<bean id="dataSource2" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
 
</bean>
<bean id="dataSource1" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
 
</bean>

2创建一个新的类 MultiDataSource

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

这个类需要继承 AbstractRoutingDataSource
并且重写 determineCurrentLookupKey方法
3 在spring 的配置文件中新增配置

<bean class="com.cmos.nganocecore.util.MultiDataSource" id="dataSource">
   <property name="targetDataSources">
      <map key-type="java.lang.String">
         <entry value-ref="dataSource1" key="dataSource1"></entry>
         <entry value-ref="dataSource2" key="dataSource2"></entry>
      </map>
   </property>
   <!-- 默认使用server的数据源 -->
   <property name="defaultTargetDataSource" ref="dataSource1"></property>
</bean>

4 将配置的dataSource 添加到sessionFactory 中

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
   <property name="configLocation" value="classpath:orm/Configuration.xml"></property>
   <property name="dataSource" ref="dataSource"></property>
</bean>

5 将sessionFactory 注入到sqlSession中

<bean id="sqlSession" class="org.mybatis.spring.SqlSessionTemplate">
   <constructor-arg index="0" ref="sqlSessionFactory" />
</bean>

6 动态切换数据源决定采用spring-aop 添加切点 去进行动态切换

6.1 新建数据切换类

public class DataBaseAop {
  
 

    public void before(InputObject inputObject,OutputObject outputObject) {
      if(条件){
MultiDataSource.setDataSourceKey("dataSource1");
}else{
MultiDataSource.setDataSourceKey("dataSource2");
}    

    }
}

6.2 定义数据切换类

<bean id="databaseAop" class="com.*.*.aop.DataBaseAop">

</bean>

6.3定义切点

<aop:config proxy-target-class="true">
   <aop:pointcut expression="(execution(void com.cmos.nganocecore.service.*.*.insert*(..))
     or execution(void com.cmos.nganocecore.service.*.*.query*(..))
     or execution(void com.cmos.nganocecore.service.*.*.select*(..))
     or execution(void com.cmos.nganocecore.service.*.*.find*(..))
     or execution(void com.cmos.nganocecore.service.*.*.insert*(..))
     or execution(void com.cmos.nganocecore.service.*.*.save*(..))
     or execution(void com.cmos.nganocecore.service.*.*.update*(..))
     or execution(void com.cmos.nganocecore.service.*.*.edit*(..))
     or execution(void com.cmos.nganocecore.service.*.*.modify*(..))
     or execution(void com.cmos.nganocecore.service.*.*.del*(..))
     or execution(void com.cmos.nganocecore.service.*.*.delete*(..)))

     and args(inputObject,outputObject)" id="dataBase"/>
   <aop:aspect ref="databaseAop">
      <aop:before method="before" pointcut-ref="dataBase" arg-names="inputObject,outputObject" />
   </aop:aspect>
 
</aop:config>

proxy-target-class="true" 参数是为了设置 拦截 器可以定义为class

注释
1 AbstractRoutingDataSource 类中 存在跟数据库资源建立连接的方法
public Connection getConnection()

在该方法中 调用了 protected DataSource determineTargetDataSource() 方法

在该方法中 会调用 determineCurrentLookupKey 获取dataBase 的key 并根据 key 从对应的数据源中获取对应的数据库连接资源 如数据库连接池资源
类 MultiDataSource 重写了 determineCurrentLookupKey 方法

@Override
    protected Object determineCurrentLookupKey() {
        return dataSourceKey.get();
    }

在需要切换数据源之前 调用全局方法 MultiDataSource.setDataSourceKey(String key) 去切换数据源

2

<bean class="com.*.*.util.MultiDataSource" id="dataSource">
   <property name="targetDataSources">
      <map key-type="java.lang.String">
         <entry value-ref="dataSource1" key="dataSource1"></entry>
         <entry value-ref="dataSource2" key="dataSource2"></entry>
      </map>
   </property>
   <!-- 默认使用server的数据源 -->
   <property name="defaultTargetDataSource" ref="dataSource1"></property>
</bean>

配置中的 标签中的 key 为动态切换数据源的key
defaultTargetDataSource 参数 指的是默认的数据源