一个注解搞定多数据源切换，你学会了吗？

我们都知道，再开发的过程中，对于多个数据库，就会有各种各样的数据源，比如Oracle，比如 Mysql,并且再开发的过程中，我们很有可能会同时使用到两个数据库，这样就设计到了切换不同的数据源来进行实现，就比如我们在 Oracle 的数据库中去查询一些必要的数据，然后让这些源头数据进入到 Mysql ,这个时候，如果不通过第三方的工具，比如 dataWorks或者其他的组件的话， 那么就得通过 Java 代码来进行实现了，今天了不起就来给大家介绍一下这个关于这个一个注解就切换数据源的操作。Vsx28资讯网——每日最新资讯28at.com

数据源

数据源的意思是“数据库应用程序所使用的数据库或者数据库服务器”；数据源也即数据的来源，是提供某种所需要数据的器件或原始媒体，在数据源中存储了所有建立数据库连接的信息，通过提供正确的数据源名称，可以找到相应的数据库连接。Vsx28资讯网——每日最新资讯28at.com

@DS

我们先来看看这个注解的使用Vsx28资讯网——每日最新资讯28at.com

<!--配置多数据源-->        <dependency>            <groupId>com.baomidou</groupId>            <artifactId>dynamic-datasource-spring-boot-starter</artifactId>            <version>3.5.0</version>        </dependency>

修改YML 的配置如下Vsx28资讯网——每日最新资讯28at.com

datasource:    dynamic:      druid:        initial-size: 5        min-idle: 5        maxActive: 20        # 配置获取连接等待超时的时间        maxWait: 60000        # 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒        timeBetweenEvictionRunsMillis: 6000        # 配置一个连接在池中最小生存的时间，单位是毫秒        minEvictableIdleTimeMillis: 60000        # 配置一个连接在池中最大生存的时间，单位是毫秒        maxEvictableIdleTimeMillis: 900000        validationQuery: SELECT 1 FROM DUAL        testWhileIdle: true        testOnBorrow: false        testOnReturn: false        # 打开PSCache，并且指定每个连接上PSCache的大小        poolPreparedStatements: true        maxPoolPreparedStatementPerConnectionSize: 20      primary: master #设置默认的数据源或者数据源组,默认值即为master      strict: false #严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源      datasource:        master:          driver-class-name: com.mysql.cj.jdbc.Driver # 3.2.0开始支持SPI可省略此配置          url: jdbc:mysql://localhost:3306/table1?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&allowPublicKeyRetrieval=true          username: xxxx          password: xxxxx        slave_1:          driver-class-name: com.mysql.cj.jdbc.Driver # 3.2.0开始支持SPI可省略此配置          url: jdbc:mysql://localhost:3306/table2?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&allowPublicKeyRetrieval=true          username: xxxx          password: xxxxx

注意这个 dynamic 一定是我们引入的 mybatis 当中的，这个时候，如果我们去使用这个@DS注解的话，那么就可以去切换数据源，比如我们在配置文件中，配置了 master  还有一个 slave_1 的数据源，那么你在使用的时候，可以这样，在我们的方法，或者类上面Vsx28资讯网——每日最新资讯28at.com

@DS("slave_1")

这个注解，你可以使用在你的方法上面，你也可以使用在你的类上面，（一般是用在mapper、service），使用过该注解的方法即可操作到对应的数据源。既然我们知道怎么去使用这个注解了，那么就得来看看这个源码的部分了。Vsx28资讯网——每日最新资讯28at.com

图片Vsx28资讯网——每日最新资讯28at.com

首先mybatis-plus使用com.baomidou.dynamic.datasource.AbstractRoutingDataSource继承 AbstractDataSource接管数据源；Vsx28资讯网——每日最新资讯28at.com

具体实现类（ctrl+alt+b可以查看抽象类或者的具体实现类）为com.baomidou.dynamic.datasource.DynamicRoutingDataSource。项目初始化调用public synchronized void addDataSource(String ds, DataSource dataSource)加载数据源，数据源存进dataSourceMap中。Vsx28资讯网——每日最新资讯28at.com

图片Vsx28资讯网——每日最新资讯28at.com

当我们发送具体的操作请求的时候，进行数据操作时，方法会被com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationInterceptor拦截Vsx28资讯网——每日最新资讯28at.com

图片Vsx28资讯网——每日最新资讯28at.com

拦截器首先从被拦截的方法或者类（一般@DS注解用于Service，也可用于Mapper和Controller）上寻找@DS注解，获取到@DS注解的值后将其存入com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder；DynamicDataSourceContextHolder使用ThreadLocal存储当前线程的数据源名。Vsx28资讯网——每日最新资讯28at.com

进行数据操作时，会调用org.springframework.jdbc.datasource.getConnection()方法；Vsx28资讯网——每日最新资讯28at.com

getConnection()方法最终调用了com.baomidou.dynamic.datasource.AbstractRoutingDataSource的getConnection()方法；Vsx28资讯网——每日最新资讯28at.com

数据操作完成后，方法返回第二步中的拦截器，执行DynamicDataSourceContextHolder.poll();清除掉此次Threadlocal中的数据源，避免影响后续数据操作。Vsx28资讯网——每日最新资讯28at.com

但是还有一个问题了，那就是不可在事务中切换数据库，保证事务需要方法使用同一连接，使用@DS(dataSourceOne)方法调用@DS(dataSourceTwo)无法切换连接，会导致方法报错。Vsx28资讯网——每日最新资讯28at.com

本文链接：http://www.28at.com/showinfo-26-34656-0.html一个注解搞定多数据源切换，你学会了吗？

声明：本网页内容旨在传播知识，若有侵权等问题请及时与本网联系，我们将在第一时间删除处理。邮件：2376512515@qq.com

上一篇： 不少人面试都挂在这道题了！你挂了吗？

下一篇： Springboot扩展点之BeanDefinitionRegistryPostProcessor，你学会了吗？