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

259 阅读2分钟
原文链接: yq.aliyun.com

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