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

最新推荐文章于 2021-02-08 05:53:07 发布

转载最新推荐文章于 2021-02-08 05:53:07 发布 · 131 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://yq.aliyun.com/articles/593278

文章标签：

#数据库 #java

本文介绍如何使用Spring和MyBatis实现动态数据源切换，通过自定义MultiDataSource类和AOP切面，在运行时根据业务需求选择不同的数据库。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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) 去切换数据源

  <map key-type="java.lang.String">
     <entry value-ref="dataSource1" key="dataSource1"></entry>
     <entry value-ref="dataSource2" key="dataSource2"></entry>
  </map>

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