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

DynamicRoutingDataSource是什么?

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

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

要完成這個(gè)功能需要

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

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

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

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

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

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

@Ds與@DsTransactional

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

分別是dynamicDatasourceAnnotationAdvisor及dynamicTransactionAdvisor。

@Ds的原理

其切面方法攔截器會(huì)將指定的ds名稱存入DynamicDataSourceContextHolder中的ThreadLocal<Deque<String>>中，因?yàn)橹付薚hreadLocalMap的泛型為Deque，而Deque的作用更準(zhǔn)確的說是棧的作用，所以支持方法嵌套調(diào)用時(shí)使用不同的ds名稱。

@DsTransactional的原理

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

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

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

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

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

我們?cè)賮砜纯此麖埩艘环裁疵婷病?/p>

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

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

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

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

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

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

總結(jié)一下

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

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

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

最新評(píng)論