多數(shù)據(jù)源的真面目DynamicRoutingDataSource

DynamicRoutingDataSource是什么?

該類實現(xiàn)了DataSource接口并在內(nèi)部維護了一個map，其中存放了多個真實的DataSource,而key則是不同數(shù)據(jù)源的名稱，本質(zhì)上就是基于靜態(tài)代理模式代理了多個真實的數(shù)據(jù)源對象。

在用多數(shù)據(jù)源時可以使用@DS來切換不同的數(shù)據(jù)源，這依賴于DynamicRoutingDataSource。

要完成這個功能需要

@DS+DSProcessor+ThreadLocal+AopMethodInterceptor+DynamicRoutingDataSource#getConnection

相互配合才能成功切換數(shù)據(jù)源。

多數(shù)據(jù)源下的事務(wù)會有什么問題？

在沒有多數(shù)據(jù)源時，系統(tǒng)中一般只會存在一個Datasource，所謂的HikariDataSource,DruidDatasource，CommonsDbcp2PoolDataSource都是DataSource的實現(xiàn),但是在系統(tǒng)中總是唯一存在的。因為SqlSessionTemplateFactory中只能存在一個DataSource,SqlSessionTemplate又是通過SqlSessionTemplateFactory生成的,并且SqlSessionTemplate在容器中又只能存在一個，所以事務(wù)時不會出現(xiàn)混亂的。而DynamicRoutingDataSource同樣也是系統(tǒng)中的唯一的Datasource,只不過它內(nèi)部代理了多個DataSource。

再來說說spring的事務(wù)管理器TransactionSynchronizationManager，內(nèi)部使用ThreadLocal保存當前事務(wù)信息。當程序執(zhí)行到@Transactional的AOP攔截器時，會在ThreadLocal中保存當前的事務(wù)信息，其中就包含了與數(shù)據(jù)庫的Connection對象。在程序執(zhí)行sql時，spring的SqlSessionTemplate#getSession方法會從事務(wù)管理器中獲取到與當前線程綁定的connection對象。

綜上所述，當使用@DS切換數(shù)據(jù)源時，沒有事務(wù)的情況下還好，會使用DynamicRoutingDataSource獲取一個新的connection并使用，但是如果是在事務(wù)的情況下，會使用事務(wù)管理器中獲取與當前線程綁定的Connection，而這個connection則是事務(wù)被創(chuàng)建時獲取的connection，就造成了雖然指定了數(shù)據(jù)源，但是還是原本的那個connection。導致切換數(shù)據(jù)源失敗。

@Ds與@DsTransactional

通過觀察DynamicDataSourceAutoConfiguration自動配置類可以發(fā)現(xiàn)，DynamicDataSource默認自動配置了@Ds注解及@DsTransactional注解的切面。

分別是dynamicDatasourceAnnotationAdvisor及dynamicTransactionAdvisor。

@Ds的原理

其切面方法攔截器會將指定的ds名稱存入DynamicDataSourceContextHolder中的ThreadLocal<Deque<String>>中，因為指定了ThreadLocalMap的泛型為Deque，而Deque的作用更準確的說是棧的作用，所以支持方法嵌套調(diào)用時使用不同的ds名稱。

@DsTransactional的原理

通過源碼可以發(fā)現(xiàn)在沒有使用seta分布式事務(wù)控制的情況下，多數(shù)據(jù)源的事務(wù)是通過dynamicDatasourceAnnotationAdvisor管理的。

dynamicDatasourceAnnotationAdvisor的核心切面攔截器就是DynamicLocalTransactionInterceptor

DynamicLocalTransactionInterceptor多數(shù)據(jù)源本地事務(wù)攔截器

內(nèi)部實現(xiàn)原理非常簡單，其實就是借助于ThreadLocal。

看到這里，你可能已經(jīng)猜到了，面紗下面的真面目就是這個ConnectionFactory類了。

我們再來看看他張了一幅什么面貌。

我們已經(jīng)知道了notify方法是@DsTransactional切面環(huán)繞通知結(jié)束時會被調(diào)用的，本著追溯本源的好奇心，你可能開始好奇了，putConnection又是什么時候被調(diào)用的呢？

我們繼續(xù)跟進就來到了AbstractRoutingDataSource。

好!又回到了DynamicRoutingDataSource中， AbstractRoutingDataSource就是DynamicRoutingDataSource父類。

在本地事務(wù)結(jié)束時，TransactionContext會清空本地事物的狀態(tài)標識，然后分別結(jié)束每一個connection的事務(wù)狀態(tài)。

然后清除ConnectionFactory中保存的與本次本地事物有關(guān)的所有connection對象的引用。

至此本地事物的創(chuàng)建和結(jié)束就完成了閉環(huán)。

總結(jié)一下

多數(shù)據(jù)源事務(wù)的控制中，參與的核心職責類有哪些

• DynamicRoutingDataSource: DataSource接口的實現(xiàn)，也是一個DataSource，本質(zhì)是多個DataSource的靜態(tài)代理類。自定義了getConnection方法的實現(xiàn)邏輯，使其與多數(shù)據(jù)源的事務(wù)管理緊密配合。
• TransactionContext：使用ThreadLocal實現(xiàn)。本地事物上下文。負責管理本地事物的狀態(tài)。
• ConnectionFactory: 使用ThreadLocal實現(xiàn)。connection連接的靜態(tài)代理類,也是一個委派者設(shè)計模式。負責管理當前本地事務(wù)中的所有Connection。
• ConnectionProxy: 封裝了真實的Connection。 目的是將ds數(shù)據(jù)源名稱和connection對應起來，這樣可以通過ds數(shù)據(jù)源名稱拿到對應的connection。
• @Ds:用于指定接下來要使用的數(shù)據(jù)源名稱。
• @DsTransactional: 多數(shù)據(jù)源事務(wù)的開啟注解，用法同@Transactional相同。但是不能指定事務(wù)的一些屬性，因為其實現(xiàn)的原理，也不需要任何其他的事務(wù)的配置。當發(fā)生任何Exception時都會執(zhí)行回滾

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