Mybatis多个数据源动态切换详解

最新推荐文章于 2024-05-27 15:30:06 发布

北港#不夏

最新推荐文章于 2024-05-27 15:30:06 发布

阅读量6k

点赞数 4

分类专栏：知识学习文章标签： java

本文链接：https://blog.csdn.net/weixin_48065337/article/details/124837801

版权

动态数据源

前言：数据源是动态的，不会写死在项目中，可以存储到数据库中，也可以通过请求切换。本示例是通过请求参数进行切换。

文章目录

动态数据源

1. 引入依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>
<!--数据源的依赖包-->
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid-spring-boot-starter</artifactId>
    <version>1.1.10</version>
</dependency>
<!--yaml-->
<dependency>
    <groupId>org.yaml</groupId>
    <artifactId>snakeyaml</artifactId>
    <version>1.29</version>
</dependency>

2. 在application.yml中记录各类数据库的数据源信息

server:
  port: 8081

spring:
  aop:
    proxy-target-class: true  #true为使用CGLIB代理,AOP动态代理
  datasource:
    driverClassName: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
    username: user
    password: user
    dataType: mysql  # 可选值 mysql oracle sqlserver postgresql

# 配置访问xml的路径
mapper-locations: classpath:mapper/**/*.xml

# 四种数据源的配置
datatype:
  mysql:
    driverClassName: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://{IP}:{port}/{dataBase}?useUnicode=true&characterEncoding=UTF-8&useSSL=false&serverTimezone=Asia/Shanghai
  oracle:
    driverClassName: oracle.jdbc.OracleDriver
    url: jdbc:oracle:thin:@{IP}:{port}:xe
  sqlserver:
    driverClassName: com.microsoft.sqlserver.jdbc.SQLServerDriver
    url: jdbc:sqlserver://{IP}:{port};DatabaseName={dataBase}
  postgresql:
    driverClassName: org.postgresql.Driver
    url: jdbc:postgresql://{IP}:{port}/{dataBase}

3. 接收数据源信息的Bean

@Data
public class DataSourceVo {
   
    private String key; // 给数据源的命名
    private String url;  // IP地址
    private String port; //端口号
    private String dataName; //数据库名称
    private String userName; // 用户名
    private String password; // 密码
    private String dataType; //数据库类型 目前支持取值mysql/oracle/sqlserver/postgresql
}

5. 继承AbstractRoutingDataSource，这个类执行数据库相关操作时，会动态获取DataSource

public class DynamicDataSource extends AbstractRoutingDataSource {
   
    @Autowired
    private DataSourceConfig dataSourceConfig;

    // 线程本地存储，线程私有区域，当前线程只能操作当前线程的局部变量
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
    // 存储数据源的集合
    private Map<Object, Object> dynamicTargetDataSources = new HashMap<>();

    @Override
    protected Object determineCurrentLookupKey() {
   
        if (StringUtils.isEmpty(getDataSource())) {
   
            return "default";
        }
        return getDataSource();
    }

    // 设置默认数据源
    @Override
    public void setDefaultTargetDataSource(Object defaultTargetDataSource) {
   
        super.setDefa