1 由于项目需要,需要将不同来源的数据存入不同的数据库,所以需要根据入参的信息动态切换数据源
项目core层采用了spring+mybitas 方式进行架构
1 在spring的配置中定义两个数据源
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 的配置文件中新增配置
<map key-type="java.lang.String ">
<entry value-ref="dataSource1 " key="dataSource1 "></entry>
<entry value-ref="dataSource2 " key="dataSource2 "></entry>
</map>
4 将配置的dataSource 添加到sessionFactory 中
5 将sessionFactory 注入到sqlSession中
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 定义数据切换类
6.3定义切点
<aop:before method="before " pointcut-ref="dataBase " arg-names="inputObject,outputObject " />
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
<map key-type="java.lang.String ">
<entry value-ref="dataSource1 " key="dataSource1 "></entry>
<entry value-ref="dataSource2 " key="dataSource2 "></entry>
</map>
配置中的 标签中的 key 为动态切换数据源的key
defaultTargetDataSource 参数 指的是默认的数据源