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 参数 指的是默认的数据源