一、使用場景

1、首先在yml文件中配置多個數(shù)據(jù)源

2、根據(jù)注解的Target可以知道

可以在我們對應(yīng)的類上、方法上使用@DS注解（一般是用在mapper、service），使用過該注解的方法即可操作到對應(yīng)的數(shù)據(jù)源

二、原理剖析（為什么使用過該注解的方法即可操作到對應(yīng)的數(shù)據(jù)源）

1、對應(yīng)源碼所在包的位置

2、首先mybatis-plus使用

com.baomidou.dynamic.datasource.AbstractRoutingDataSource繼承 AbstractDataSource接管數(shù)據(jù)源；

具體實現(xiàn)類（ctrl+alt+b可以查看抽象類或者的具體實現(xiàn)類）為com.baomidou.dynamic.datasource.DynamicRoutingDataSource。

項目初始化調(diào)用public synchronized void addDataSource(String ds, DataSource dataSource)加載數(shù)據(jù)源，數(shù)據(jù)源存進(jìn)dataSourceMap中。

AbstracRoutingDataSource

DynamicRoutingDataSource

3、當(dāng)我們發(fā)送具體的操作請求的時候

進(jìn)行數(shù)據(jù)操作時，方法會被com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationInterceptor攔截

 DynamicDataSourceAnnotationInterceptor

 攔截器首先從被攔截的方法或者類（一般@DS注解用于Service，也可用于Mapper和Controller）上尋找@DS注解，獲取到@DS注解的值后將其存入

com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder；

DynamicDataSourceContextHolder使用ThreadLocal存儲當(dāng)前線程的數(shù)據(jù)源名。

 3、進(jìn)行數(shù)據(jù)操作時

會調(diào)用org.springframework.jdbc.datasource.getConnection()方法；

getConnection()方法最終調(diào)用了com.baomidou.dynamic.datasource.AbstractRoutingDataSource的getConnection()方法；

 跟進(jìn)方法我們可以看到determineDataSource()由子類com.baomidou.dynamic.datasource.DynamicRoutingDataSource實現(xiàn)，可以看到DynamicRoutingDataSource從DynamicDataSourceContextHolder獲取數(shù)據(jù)源名稱，這個在之前攔截器處理存進(jìn)ThreadLocal中，如果有數(shù)據(jù)源名稱則從dataSourceMap中獲取，沒有則獲取默認(rèn)的primary數(shù)據(jù)源。

 此時的datasource已經(jīng)切換成了我們需要的數(shù)據(jù)源

 4、數(shù)據(jù)操作完成后

方法返回第二步中的攔截器，執(zhí)行DynamicDataSourceContextHolder.poll();清除掉此次Threadlocal中的數(shù)據(jù)源，避免影響后續(xù)數(shù)據(jù)操作。

特別注意：

不可在事務(wù)中切換數(shù)據(jù)庫，保證事務(wù)需要方法使用同一連接，使用@DS(dataSourceOne)方法調(diào)用@DS(dataSourceTwo)無法切換連接，會導(dǎo)致方法報錯。

總結(jié)

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。 

最新評論