Mybatis-Plus中使用@DS注解動(dòng)態(tài)選擇數(shù)據(jù)源的源碼解讀

更新時(shí)間：2023年07月31日 10:29:13 作者：labulaka24

這篇文章主要介紹了Mybatis-Plus中使用@DS注解動(dòng)態(tài)選擇數(shù)據(jù)源的源碼解讀,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

一、使用場(chǎng)景

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

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

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

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

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

2、首先mybatis-plus使用

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

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

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

AbstracRoutingDataSource

DynamicRoutingDataSource

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

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

DynamicDataSourceAnnotationInterceptor

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

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

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

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

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

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

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