很多情況，我們確實需要在一個服務中訪問多個數據源。雖然它讓整體設計變的不那么優(yōu)雅，但真實的世界確實需要它。比如，你的業(yè)務為兩個比較大的客戶服務，但你希望他們能夠共用一套代碼。

也就是說，你的代碼剛開始沒有考慮設計多租戶這種功能，但后面又有這種蛋疼的需求。但還好不是爆炸式的租戶增長。

除了引入一些分庫分表組件，Spring自身提供了AbstractRoutingDataSource的方式，讓多數數據源的管理成為可能。其實分庫分表組件使用上限制很多，你不得不首先梳理這座屎山，接下來還要忍受中間件對你的SQL的苛刻要求;反而是一些野路子，能夠讓代碼的改動量盡量的減少。

心動不如行動。接下來，就讓我們來看一下它的具體實現吧。

1.基本原理

多數據源能進行動態(tài)切換的核心就是spring底層提供了AbstractRoutingDataSource類進行數據源路由。AbstractRoutingDataSource實現了DataSource接口，所以我們可以將其直接注入到DataSource的屬性上。

我們主要繼承這個類，實現里面的方法determineCurrentLookupKey()，而此方法只需要返回一個數據庫的名稱即可。

比如，Controller通過拿到前端業(yè)務傳遞的數值，進行業(yè)務邏輯分發(fā)。它就可以手動設置當前請求的數據庫標識，然后路由到正確的庫表里面。

1. @Controller 
2. public class ARDTestController { 
3.     @GetMapping("test") 
4.     public void chifeng(){ 
5.         //db-a 應該是上層傳遞下來的屬性，我們可以把它放在ThreadLocal里 
6.         DataSourceContextHolder.setDbKey("db-a"); 
7.     } 
8. } 

那么當sql語句執(zhí)行的時候，它如何知道自己需要切換到哪個數據源呢?是不是需要把db-a這個屬性一直透傳下去呢?

在Java中，可以使用ThreadLocal綁定這個透傳的屬性。像Spring的嵌套事務等實現的原理，也是基于ThreadLocal去運行的。所以，DataSourceContextHolder.本質上是一個操作ThreadLocal的類。

1. public class DataSourceContextHolder { 
2.     private static InheritableThreadLocal<String> dbKey = new InheritableThreadLocal<>(); 
3.  
4.     public static void setDbKey(String key){ 
5.         dbKey.set(key); 
6.     } 
7.  
8.     public static String getDbKey(){ 
9.         return dbKey.get(); 
10.     } 
11. } 

2.配置代碼

首先，我們自定義了配置文件的格式。如下面的代碼，就配置了db-a和db-b兩個數據庫。

1. multi: 
2.   dbs: 
3.     db-a: 
4.       driver-class-name: org.h2.Driver 
5.       url: jdbc:h2:mem:dba;MODE=MYSQL;DATABASE_TO_UPPER=false; 
6.     db-b: 
7.       driver-class-name: org.h2.Driver 
8.       url: jdbc:h2:mem:dbb;MODE=MYSQL;DATABASE_TO_UPPER=false; 

然后，我們將它解析稱properties。

1. @ConfigurationProperties(prefix = "multi") 
2. @Configuration 
3. public class DbsProperties { 
4.     private Map<String, Map<String, String>> dbs = new HashMap<>(); 
5.  
6.     public Map<String, Map<String, String>> getDbs() { 
7.         return dbs; 
8.     } 
9.  
10.     public void setDbs(Map<String, Map<String, String>> dbs) { 
11.         this.dbs = dbs; 
12.     } 
13. } 

接下來一步，需要配置整個應用所默認的數據源。如你所見，它的主要邏輯，就是在運行的時候，從ThreadLocal里取出提前設置的這個值。

1. public class DynamicDataSource extends AbstractRoutingDataSource { 
2.     @Override 
3.     protected Object determineCurrentLookupKey() { 
4.         return DataSourceContextHolder.getDbKey(); 
5.     } 
6. } 

最后一步，設置整個項目中默認的DataSource。注意，我們生成DynamicDataSource之后，還需要提供targetDataSource和defaultTargetDataSource兩個屬性的值，才能夠正常運行。

1. @Configuration 
2. public class DynamicDataSourceConfiguration { 
3.     @Autowired 
4.     DbsProperties properties; 
5.  
6.     @Bean 
7.     public DataSource dataSource(){ 
8.         DynamicDataSource dataSource = new DynamicDataSource(); 
9.         final Map<Object,Object> targetDataSource  = getTargetDataSource(); 
10.         dataSource.setTargetDataSources(targetDataSource); 
11.         //TODO 默認數據庫需要設置 
12.         dataSource.setDefaultTargetDataSource(targetDataSource.values().iterator().next()); 
13.         return dataSource; 
14.     } 
15.  
16.     private Map<Object,Object> getTargetDataSource(){ 
17.         Map<Object,Object> dataSources = new HashMap<>(); 
18.         this.properties.getDbs().entrySet().stream() 
19.                 .forEach(e->{ 
20.                     DriverManagerDataSource dmd = new DriverManagerDataSource(); 
21.                     dmd.setUrl(e.getValue().get("url")); 
22.                     dmd.setDriverClassName(e.getValue().get("driver-class-name")); 
23.                     dataSources.put(e.getKey(),dmd); 
24.                 }); 
25.         return  dataSources; 
26.     } 
27. } 

3.問題

通過以上簡單的代碼，就可以實現Spring簡單的多數據源管理。但明顯的，它還存在很多問題。

• 需要產品設計選擇模式，進行業(yè)務切換。
• 前端可以采用放在localStroage的方式，保存屬性，可使用攔截器方式將變量每次都傳遞。
• 后端每次請求，都需要帶上目標db，可以采用放在ThreadLocal里的方式。但ThreadLocal有線程透傳的問題，如果任務里開啟了子線程，則變量不能共享。
• 由于表是動態(tài)選擇的，所以JPA自動創(chuàng)建和update等模式，將不可用。不方便測試和單元測試，在測試接口的時候，也需要每次強制指定指向的庫。
• 由于是修改數據源的模式，每次增加庫，都需要重新啟動上線才可以。如果要做到動態(tài)性，數據源銷毀是個問題。

End

對于一個微服務來說，有很多默認的限制策略，比如，不同域之間的服務是不能共享一個數據庫的。這些基本原則，把微服務整的清清爽爽，是一些基本的原則。

同理的，如果我們在設計開始，就給每一張表加上租戶的字段ID，那么寫代碼的時候就順暢的多。但是世界上沒有這么多如果。

原則為何而存在?當然是為了讓人去打破的。

編程只是工具，反正代碼在自己手里，怎么玩，看需要，也看心情。條條大路通羅馬，曲徑通幽處，風光無限好。

作者簡介：小姐姐味道 (xjjdog)，一個不允許程序員走彎路的公眾號。聚焦基礎架構和Linux。十年架構，日百億流量，與你探討高并發(fā)世界，給你不一樣的味道。

原文鏈接：https://mp.weixin.qq.com/s/KS8GJEqFIRTiSiWsmRTKNg