最近在學習Spring boot，寫了個讀寫分離。並未照搬網文，而是獨立思考後的成果，寫完以後發現從零開始寫讀寫分離並不難！

我最初的想法是： 讀方法走讀庫，寫方法走寫庫（一般是主庫），保證在Spring提交事務之前確定資料來源.

保證在Spring提交事務之前確定資料來源，這個簡單，利用AOP寫個切換資料來源的切面，讓他的優先順序高於Spring事務切面的優先順序。至於讀，寫方法的區分可以用2個註解。

但是如何切換資料庫呢？ 我完全不知道！多年經驗告訴我

我搜索了一些網文，發現都提到了一個AbstractRoutingDataSource類。檢視原始碼註釋如下

/**

Abstract {@link javax.sql.DataSource} implementation that routes {@link #getConnection()}

* calls to one of various target DataSources based on a lookup key. The latter is usually

* (but not necessarily) determined through some thread-bound transaction context.

*

* @author Juergen Hoeller

* @since 2.0.1

* @see #setTargetDataSources

* @see #setDefaultTargetDataSource

* @see #determineCurrentLookupKey()

*/

AbstractRoutingDataSource就是DataSource的抽象，基於lookup key的方式在多個數據庫中進行切換。重點關注setTargetDataSources，setDefaultTargetDataSource，determineCurrentLookupKey三個方法。那麼AbstractRoutingDataSource就是Spring讀寫分離的關鍵了。

仔細閱讀了三個方法，基本上跟方法名的意思一致。setTargetDataSources設定備選的資料來源集合。 setDefaultTargetDataSource設定預設資料來源，determineCurrentLookupKey決定當前資料來源的對應的key。

但是我很好奇這3個方法都沒有包含切換資料庫的邏輯啊！我仔細閱讀原始碼發現一個方法，determineTargetDataSource方法，其實它才是獲取資料來源的實現。原始碼如下：

簡單說就是，根據determineCurrentLookupKey獲取的key,在resolvedDataSources這個Map中查詢對應的datasource！，注意determineTargetDataSource方法竟然不使用的targetDataSources！

那一定存在resolvedDataSources與targetDataSources的對應關係。我接著翻閱程式碼，發現一個afterPropertiesSet方法（Spring原始碼中InitializingBean介面中的方法），這個方法將targetDataSources的值賦予了resolvedDataSources。原始碼如下：

afterPropertiesSet 方法，熟悉Spring的都知道，它在bean例項已經建立好，且屬性值和依賴的其他bean例項都已經注入以後執行。

也就是說呼叫，targetDataSources，defaultTargetDataSource的賦值一定要在afterPropertiesSet前邊執行。

AbstractRoutingDataSource簡單總結：

1. AbstractRoutingDataSource，內部有一個Map<Object,DataSource>的域resolvedDataSources

2. determineTargetDataSource方法通過determineCurrentLookupKey方法獲得key，進而從map中取得對應的DataSource。

3. setTargetDataSources 設定 targetDataSources

4. setDefaultTargetDataSource 設定 defaultTargetDataSource，

5. targetDataSources和defaultTargetDataSource 在afterPropertiesSet分別轉換為resolvedDataSources和resolvedDefaultDataSource。

6. targetDataSources，defaultTargetDataSource的賦值一定要在afterPropertiesSet前邊執行。

進一步瞭解理論後，讀寫分離的方式則基本上出現在眼前了。（“下列方法不唯一”）

先寫一個類繼承AbstractRoutingDataSource，實現determineCurrentLookupKey方法，和afterPropertiesSet方法。afterPropertiesSet方法中呼叫setDefaultTargetDataSource和setTargetDataSources方法之後呼叫super.afterPropertiesSet。

之後定義一個切面在事務切面之前執行，確定真實資料來源對應的key。但是這又出現了一個問題，如何執行緒安全的情況下傳遞每個執行緒獨立的key呢？沒錯使用ThreadLocal傳遞真實資料來源對應的key。

ThreadLocal，Thread的區域性變數，確保每一個執行緒都維護變數的一個副本

到這裡基本邏輯就想通了，之後就是寫了。

DataSourceContextHolder 使用ThreadLocal儲存真實資料來源對應的key

DataSourceAopAspect 切面切換真實資料來源對應的key，並設定優先順序保證高於事務切面

RoutingDataSouceImpl實現AbstractRoutingDataSource的邏輯

基本邏輯實現完畢了就進行，通用設定,設定資料來源，事務，SqlSessionFactory等

其他程式碼，就不在這裡贅述了，有興趣可以移步完整程式碼。

使用Spring寫讀寫分離，其核心就是AbstractRoutingDataSource，原始碼不難，讀懂之後，寫個讀寫分離就簡單了！。

AbstractRoutingDataSource重點回顧：

1. AbstractRoutingDataSource，內部有一個Map<Object,DataSource>的域resolvedDataSources

2. determineTargetDataSource方法通過determineCurrentLookupKey方法獲得key，進而從map中取得對應的DataSource。

3. setTargetDataSources 設定 targetDataSources

4. setDefaultTargetDataSource 設定 defaultTargetDataSource，

5. targetDataSources和defaultTargetDataSource 在afterPropertiesSet分別轉換為resolvedDataSources和resolvedDefaultDataSource。

6. targetDataSources，defaultTargetDataSource的賦值一定要在afterPropertiesSet前邊執行。

這周確實有點忙，週五花費了些時間不過總算實現了自己的諾言。

完成承諾不容易，喜歡您就點個贊！

推薦一個交流學習群：575745314 裡面會分享一些資深架構師錄製的視訊錄影：有Spring，MyBatis，Netty原始碼分析，高併發、高效能、分散式、微服務架構的原理，JVM效能優化這些成為架構師必備的知識體系。還能領取免費的學習資源，目前受益良多：