SpringBoot 手动实现动态切换数据源 DynamicSource （中）

大家好，我是此林。

SpringBoot 手动实现动态切换数据源 DynamicSource （上）-CSDN博客

在上一篇博客中，我带大家手动实现了一个简易版的数据源切换实现，方便大家理解数据源切换的原理。今天我们来介绍一个开源的数据源切换框架，是baomidou 旗下和 Mybatis-plus 同级的开源框架：dynamic-datasource-spring-boot-starter。

1. 简介

dynamic-datasource-spring-boot-starter 是一个基于springboot的快速集成多数据源的启动器。

其支持 Jdk 1.7+, SpringBoot 1.5.x 2.x.x 3.x.x。
JPA用户不建议使用，JPA自带事务，无法连续切库。

2. 使用

下面基于SpringBoot2 做演示。

1.pom.xml

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

2.application.yml

spring:datasource:dynamic:primary: master #设置默认的数据源或者数据源组,默认值即为masterstrict: false #严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源datasource:master:url: jdbc:mysql://192.168.183.128:3306/userusername: rootpassword: 123456slave_1:url: jdbc:mysql://192.168.183.128:3307/userusername: rootpassword: 123456slave_2:url: jdbc:mysql://192.168.183.128:3308/userusername: rootpassword: 123456

参数说明：

以上会配置一个默认库master，一个组slave下有两个子库slave_1,slave_2

后续我们会使用AOP注解的方式指定数据源。

例如：@DS("master")、@DS("slave")

1. 配置文件所有以下划线 _ 分割的数据源 首部 即为组的名称，相同组名称的数据源会放在一个组下。
2. 切换数据源可以是组名，也可以是具体数据源名称。组名则切换时采用负载均衡算法切换。

例如：当我们使用@DS("slave")时，指定了组名  “slave”，那么它会采用负载均衡算法自动切换数据源。

因为我们后续希望实现Mysql主从读写分离，所有写操作走master数据库，所有读操作走slave从数据库。

3. 使用 @DS 切换数据源。

@Service
public class UserServiceImpl implements UserService {@Autowiredprivate UserMapper userMapper;@Overridepublic void insert(User user) {userMapper.insert(user);}@Override@DS("slave_1")public List<User> selectAll() {userMapper.selectAll();}@Override@DS("slave")public User selectById(Integer id) {return userMapper.selectById(id);}
}

说明：

insert() 方法没有加@DS，默认使用master数据库。

selectAll() 方法指定使用数据源slave_1

selectById() 使用slave组，自动负载均衡查询。

注：@DS 可以注解在方法上或类上，同时存在就近原则 方法上注解 优先于 类上注解。

若要实现Mysql主从自动复制，还需要配置mysql主从复制，详情参考博客：

ShardingSphere5：Mysql主从集群搭建、读写分离_shardingsphere mysql读写分离-CSDN博客

其中SpringBoot配置 ShardingSphere5 这一块不用管，Mysql主从集群搭建是不变的。

4.其他实现

3. 底层原理

1. SpringBoot自动配置

我们找到 DynamicDataSourceAutoConfiguration.class。 这是框架的自动配置类。

这一段是从我们配置的application.yml 读取并设置相关属性，也就是我们配置的数据源信息。

具体的话，是赋值给 DynamicRoutingDataSource 这个对象。

2. 扫描@DS注解

这是 DynamicDataSourceAnnotationInterceptor 的 invoke 方法，典型的AOP。

在所有加了@DS的方法上，

1. 当执行目标方法的时候，会先获取到@DS("XXX")中数据源的名称，dsKey。

2. 然后再把dsKey 存到ThreadLocal中，也就是图中的DynamicDataSourceContextHolder。

3. 最后再开始执行目标方法。

3. DynamicRoutingDataSource

回到我们之前自动配置里说的 DynamicRoutingDataSource

观察 DynamicRoutingDataSource 的继承关系链，发现它实现了Spring的 DataSource 和 InitializingBean 接口。

这和我们上一篇博客手动实现数据源切换的思路是一样的。

在 DynamicRoutingDataSource 里，determineDataSource 从 ThreadLocal 里 获取到指定数据源名称

最后通过名称返回对应的DataSource数据源。

对比上一篇博客，

SpringBoot 手动实现动态切换数据源 DynamicSource （上）-CSDN博客

其实思路大差不差，只不过 dynamic-datasource-spring-boot-starter 更加完善，增加了AOP注解功能。

4. 无注解切换数据源

1. 自定义切面，结合 DynamicDatasourceNamedInterceptor ，对于所有以 select 方法开头的，设置使用 slave 数据源， 对于add、update、delete 开头的使用 master 数据源。

2. 自定义 Mybatis 拦截器，对于所有的写操作使用 master 数据源，所有读操作使用 slave 数据源。（即：把数据源标识写入 DynamicDataSourceContextHolder 中）。

3. 其他场景：SaaS多租户系统下，每个租户使用独立的数据库，那么可以要求前端携带请求头 “dsKey”。配置自定义拦截器，拦截请求，把dsKey 写入 DynamicDataSourceContextHolder。

今天的分享就到这里。

我是此林，关注我吧，带你看不一样的世界！