Spring boot 使用@DS 配合druid连接池切换数据源及切换数据源失效情况

news2024/11/17 0:04:09

一、导入包

<!-- dynamic-datasource -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
    <version>3.2.1</version>
</dependency>

二、配置yam配置文件

spring:
  datasource:
    druid:
       dynamic:
         datasource:
           master: 
            url: jdbc:mysql://192.168.1.1:3306/master?characterEncoding=UTF-8&useUnicode=true&useSSL=false&tinyInt1isBit=false&allowPublicKeyRetrieval=true&serverTimezone=Asia/Shanghai
             username: root
             password: "123tttt@444tyMAIYUE2022"
             driver-class-name: com.mysql.cj.jdbc.Driver
           slave:
             url: jdbc:mysql://192.168.1.1:3306/slave?characterEncoding=UTF-8&useUnicode=true&useSSL=false&tinyInt1isBit=false&allowPublicKeyRetrieval=true&serverTimezone=Asia/Shanghai
             username: root
             password: "123@444tyMAIYUE2022"
             driver-class-name: com.mysql.cj.jdbc.Driver

三、在类上或者在方法上使用@DS切换数据源

在类或者方法上使用数据源很简单，但是需要注意会有以下几种情况会导致切换数据源失效：

1、在使用@DS的方法在调用的方法中使用了@Transactional(rollbackFor = Exception.class)开启了事务，会导致切换数据源失败

2、在方法A使用slave数据源，方法B使用master数据源，方法A使用this调用方法B，方法B会继续使用slave数据源，造成切换数据源失效

四、问题剖析及解决方案

1、主方法开启了@Transactional导致切换数据源失效问题剖析及解决方案

为什么开启了事务会导致切换数据源失效，首先我们需要从Spring 事务的传播机制说起

PROPATATION_REQUIRE（默认）	如果当前没有事务，就创建一个事务，如果当前有事务，那就加入该事务
PROPATATION_SUPPORTS	如果当前有事务，就加入该事务，没有就以非事务的方式执行
PROPATATION_MANDRTOR	如果有事务就加入事务，没有就抛出异常
PROPATATION_NEW_REQURIRE	新建一个事务，如果当前有事务，就挂起当前事务
PROPATATION_NOT_SUPPORTS	以非事务的方式执行，如果当前存在事务，就挂起当前事务
PROPATATION_NEVER	以非事务的方式执行，如果存在当前事务，就抛出异常
PROPATATION_NESTED	如果当前存在事务，就嵌套在当前事务中执行，没有就新建一个事务

Spring 默认传播机制是PROPATAION_REQUIRE，使用这个注解，会从默认创建事务的数据源中获取数据源，从源码中分析Spring 获取数据源的源码，可以直观看到，如果存在已经开启事务的数据源就使用存在事务的数据源。

解决方案：

1、在使用@DS切换数据源的主调用方法不要使用事务，去掉@Transactional

2、如果一定要使用@Transactional标签开启事务，不要使用默认的事务传播机制，为每个方法开启一个独立的事务

@Transactional(propagation = Propagation.REQUIRES_NEW, rollbackFor = Exception.class)

2、在方法A使用slave数据源，方法B使用master数据源，方法A调用方法B，方法B会继续使用slave数据源，造成切换数据源失效问题剖析及解决方案

DS切换数据源使用了AOP切面，AOP切面使用的动态代理技术，动态代理在执行方法A时会生成一个方法A的代理对象，而调用方法B时使用的是this调用，this的指向的是源对象本身，所以方法B没有代理的对象调用，所以方法B不会进入代理。看DS AOP拦截类可以看到，只有代理的方法执行完毕后清空切换的数据源信息，这两个问题导致了方法B是没有切换数据源失效。