多数据源管理：掌握 @DS 注解的威力

来源｜blog.csdn.net/jdcdev_/article/details/133635988

👉 欢迎加入小哈的星球 ，你将获得: 专属的项目实战/ Java 学习路线 / 一对一提问 / 学习打卡/  赠书福利

全栈前后端分离博客项目 1.0 版本完结啦，2.0 正在更新中..., 演示链接：http://116.62.199.48/ ，全程手摸手，后端 + 前端全栈开发，从 0 到 1 讲解每个功能点开发步骤，1v1 答疑，直到项目上线。目前已更新了170小节，累计27w+字，讲解图：1162张，还在持续爆肝中.. 后续还会上新更多项目，目标是将Java领域典型的项目都整一波，如秒杀系统, 在线商城, IM即时通讯，Spring Cloud Alibaba 等等，戳我加入学习，已有610+小伙伴加入(早鸟价超低)

大家在日常后端开发过程，不可避免的会接触到需要用到配置多个数据源的场景，在这里，小编介绍一种简单方便的，只需要简单的配置和一个 @DS 注解就能实现动态数据源的方式，这种动态数据源底层原理是基于 Mybatis-plus 来实现的。

1、配置方式

首先是 pom.xml

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

随后配上多数据源，我们的项目中分别配置了 MySQL 和 Doris 两种数据库。

spring:  
    datasource:    
        druid:      
            localdb:        
                url: xxx        
                username: xxx     
                driver-class-name: com.mysql.jdbc.Driver        
                type: com.alibaba.druid.pool.DruidDataSource      
            doris:       
                url: xxx        
                username: xxx
                driver-class-name: com.mysql.jdbc.Driver        
                type: com.alibaba.druid.pool.DruidDataSource

最后需要使用哪个数据库，在对应的方法上加上对应的 @DS（“数据库名”）注解就可以了，在类上加 @DS（“数据库名”），那么整个类下的所有方法都会使用到这个数据源，实现方式是不是非常简单。

@Service
@DS("localdb")
public class testServiceImpl extends BaseServiceImpl<testMapper, testBean> implements testService {

    public void save(String arg1) {
       //TODO
    }
    
    public void find(String arg2) {
       //TODO
    }
}

2、底层实现原理

底层实现核心类是 com.baomidou.dynamic.datasource.DynamicRoutingDataSource，继承自 AbstractDataSource。

图片

图片

项目在初始化的时候会调用 DynamicRoutingDataSource 里面的 public synchronized void addDataSource(String ds, DataSource dataSource) 方法加载数据源，数据源存进 dataSourceMap 中。

图片

随后，如果进行数据库操作，以方法为最小粒度，执行对应方法时，会被  DynamicDataSourceAnnotationInterceptor 拦截器拦截。

图片

执行 determineDatasource 方法，扫描加了 @DS 注解的类或者方法，随后调用 DynamicDataSourceContextHolder.poll 方法。

图片

DynamicDataSourceContextHolder.poll 方法将当前线程的数据源名加到对应的 ThreadLocal 线程本地中。

图片

在之后的数据操作中，会调用 org.springframework.jdbc.datasource.getConnection() 方法，ThreadLocal 中获取之前拦截器存进去动态数据源名，如果没有获取到，就默认为配置的 primary 数据源，这完成了对应的动态数据源切换。

图片

3、总结

基于 Mybatis-plus 的 @DS 注解可以实现动态数据源切换，具有实现简单的优点，虽然没有 AOP 实现，但是用到了 AOP 的思想。

👉 欢迎加入小哈的星球 ，你将获得: 专属的项目实战/ Java 学习路线 / 一对一提问 / 学习打卡/  赠书福利

全栈前后端分离博客项目 1.0 版本完结啦，2.0 正在更新中..., 演示链接：http://116.62.199.48/ ，全程手摸手，后端 + 前端全栈开发，从 0 到 1 讲解每个功能点开发步骤，1v1 答疑，直到项目上线。目前已更新了170小节，累计27w+字，讲解图：1162张，还在持续爆肝中.. 后续还会上新更多项目，目标是将Java领域典型的项目都整一波，如秒杀系统, 在线商城, IM即时通讯，Spring Cloud Alibaba 等等，戳我加入学习，已有600+小伙伴加入(早鸟价超低)

1. 我的私密学习小圈子~

2. 优秀的代码都是如何分层的？看了直呼NB！

3. 放弃 ELK！SpringBoot接入轻量级分布式日志框架GrayLog

4. 2 万字详解，吃透 ES！

最近面试BAT，整理一份面试资料《Java面试BATJ通关手册》，覆盖了Java核心技术、JVM、Java并发、SSM、微服务、数据库、数据结构等等。
获取方式：点“在看”，关注公众号并回复 Java 领取，更多内容陆续奉上。

PS：因公众号平台更改了推送规则，如果不想错过内容，记得读完点一下“在看”，加个“星标”，这样每次新文章推送才会第一时间出现在你的订阅列表里。
点“在看”支持小哈呀，谢谢啦