一個注解搞定多數據源切換，你學會了嗎？
我們都知道，再開發的過程中，對于多個數據庫，就會有各種各樣的數據源，比如Oracle，比如 Mysql,并且再開發的過程中，我們很有可能會同時使用到兩個數據庫，這樣就設計到了切換不同的數據源來進行實現，就比如我們在 Oracle 的數據庫中去查詢一些必要的數據，然后讓這些源頭數據進入到 Mysql ,這個時候，如果不通過第三方的工具，比如 dataWorks或者其他的組件的話， 那么就得通過 Java 代碼來進行實現了，今天了不起就來給大家介紹一下這個關于這個一個注解就切換數據源的操作。
數據源
數據源的意思是“數據庫應用程序所使用的數據庫或者數據庫服務器”；數據源也即數據的來源，是提供某種所需要數據的器件或原始媒體，在數據源中存儲了所有建立數據庫連接的信息，通過提供正確的數據源名稱，可以找到相應的數據庫連接。
@DS
我們先來看看這個注解的使用
<!--配置多數據源-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>dynamic-datasource-spring-boot-starter</artifactId>
<version>3.5.0</version>
</dependency>修改YML 的配置如下
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 的數據源，那么你在使用的時候，可以這樣，在我們的方法，或者類上面
@DS("slave_1")這個注解，你可以使用在你的方法上面，你也可以使用在你的類上面，（一般是用在mapper、service），使用過該注解的方法即可操作到對應的數據源。既然我們知道怎么去使用這個注解了，那么就得來看看這個源碼的部分了。
圖片
首先mybatis-plus使用com.baomidou.dynamic.datasource.AbstractRoutingDataSource繼承 AbstractDataSource接管數據源；
具體實現類（ctrl+alt+b可以查看抽象類或者的具體實現類）為com.baomidou.dynamic.datasource.DynamicRoutingDataSource。項目初始化調用public synchronized void addDataSource(String ds, DataSource dataSource)加載數據源，數據源存進dataSourceMap中。
圖片
當我們發送具體的操作請求的時候，進行數據操作時，方法會被com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationInterceptor攔截
圖片
攔截器首先從被攔截的方法或者類（一般@DS注解用于Service，也可用于Mapper和Controller）上尋找@DS注解，獲取到@DS注解的值后將其存入com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder；DynamicDataSourceContextHolder使用ThreadLocal存儲當前線程的數據源名。
進行數據操作時，會調用org.springframework.jdbc.datasource.getConnection()方法；
getConnection()方法最終調用了com.baomidou.dynamic.datasource.AbstractRoutingDataSource的getConnection()方法；
數據操作完成后，方法返回第二步中的攔截器，執行DynamicDataSourceContextHolder.poll();清除掉此次Threadlocal中的數據源，避免影響后續數據操作。
但是還有一個問題了，那就是不可在事務中切換數據庫，保證事務需要方法使用同一連接，使用@DS(dataSourceOne)方法調用@DS(dataSourceTwo)無法切換連接，會導致方法報錯。
