實用：Spring的多租戶數據源管理 AbstractRoutingDataSource！
本文轉載自微信公眾號「小姐姐味道」，作者小姐姐養的狗02號 。轉載本文請聯系小姐姐味道公眾號。
很多情況，我們確實需要在一個服務中訪問多個數據源。雖然它讓整體設計變的不那么優雅，但真實的世界確實需要它。比如，你的業務為兩個比較大的客戶服務，但你希望他們能夠共用一套代碼。
也就是說，你的代碼剛開始沒有考慮設計多租戶這種功能，但后面又有這種蛋疼的需求。但還好不是爆炸式的租戶增長。
除了引入一些分庫分表組件，Spring自身提供了AbstractRoutingDataSource的方式，讓多數數據源的管理成為可能。其實分庫分表組件使用上限制很多，你不得不首先梳理這座屎山，接下來還要忍受中間件對你的SQL的苛刻要求;反而是一些野路子，能夠讓代碼的改動量盡量的減少。
心動不如行動。接下來，就讓我們來看一下它的具體實現吧。
1.基本原理
多數據源能進行動態切換的核心就是spring底層提供了AbstractRoutingDataSource類進行數據源路由。AbstractRoutingDataSource實現了DataSource接口，所以我們可以將其直接注入到DataSource的屬性上。
我們主要繼承這個類，實現里面的方法determineCurrentLookupKey()，而此方法只需要返回一個數據庫的名稱即可。
比如，Controller通過拿到前端業務傳遞的數值，進行業務邏輯分發。它就可以手動設置當前請求的數據庫標識，然后路由到正確的庫表里面。
- @Controller
- public class ARDTestController {
- @GetMapping("test")
- public void chifeng(){
- //db-a 應該是上層傳遞下來的屬性，我們可以把它放在ThreadLocal里
- DataSourceContextHolder.setDbKey("db-a");
- }
- }
那么當sql語句執行的時候，它如何知道自己需要切換到哪個數據源呢?是不是需要把db-a這個屬性一直透傳下去呢?
在Java中，可以使用ThreadLocal綁定這個透傳的屬性。像Spring的嵌套事務等實現的原理，也是基于ThreadLocal去運行的。所以，DataSourceContextHolder.本質上是一個操作ThreadLocal的類。
- public class DataSourceContextHolder {
- private static InheritableThreadLocal<String> dbKey = new InheritableThreadLocal<>();
- public static void setDbKey(String key){
- dbKey.set(key);
- }
- public static String getDbKey(){
- return dbKey.get();
- }
- }
2.配置代碼
首先，我們自定義了配置文件的格式。如下面的代碼，就配置了db-a和db-b兩個數據庫。
- multi:
- dbs:
- db-a:
- driver-class-name: org.h2.Driver
- url: jdbc:h2:mem:dba;MODE=MYSQL;DATABASE_TO_UPPER=false;
- db-b:
- driver-class-name: org.h2.Driver
- url: jdbc:h2:mem:dbb;MODE=MYSQL;DATABASE_TO_UPPER=false;
然后，我們將它解析稱properties。
- @ConfigurationProperties(prefix = "multi")
- @Configuration
- public class DbsProperties {
- private Map<String, Map<String, String>> dbs = new HashMap<>();
- public Map<String, Map<String, String>> getDbs() {
- return dbs;
- }
- public void setDbs(Map<String, Map<String, String>> dbs) {
- this.dbs = dbs;
- }
- }
接下來一步，需要配置整個應用所默認的數據源。如你所見，它的主要邏輯，就是在運行的時候，從ThreadLocal里取出提前設置的這個值。
- public class DynamicDataSource extends AbstractRoutingDataSource {
- @Override
- protected Object determineCurrentLookupKey() {
- return DataSourceContextHolder.getDbKey();
- }
- }
最后一步，設置整個項目中默認的DataSource。注意，我們生成DynamicDataSource之后，還需要提供targetDataSource和defaultTargetDataSource兩個屬性的值，才能夠正常運行。
- @Configuration
- public class DynamicDataSourceConfiguration {
- @Autowired
- DbsProperties properties;
- @Bean
- public DataSource dataSource(){
- DynamicDataSource dataSource = new DynamicDataSource();
- final Map<Object,Object> targetDataSource = getTargetDataSource();
- dataSource.setTargetDataSources(targetDataSource);
- //TODO 默認數據庫需要設置
- dataSource.setDefaultTargetDataSource(targetDataSource.values().iterator().next());
- return dataSource;
- }
- private Map<Object,Object> getTargetDataSource(){
- Map<Object,Object> dataSources = new HashMap<>();
- this.properties.getDbs().entrySet().stream()
- .forEach(e->{
- DriverManagerDataSource dmd = new DriverManagerDataSource();
- dmd.setUrl(e.getValue().get("url"));
- dmd.setDriverClassName(e.getValue().get("driver-class-name"));
- dataSources.put(e.getKey(),dmd);
- });
- return dataSources;
- }
- }
3.問題
通過以上簡單的代碼，就可以實現Spring簡單的多數據源管理。但明顯的，它還存在很多問題。
- 需要產品設計選擇模式，進行業務切換。
- 前端可以采用放在localStroage的方式，保存屬性，可使用攔截器方式將變量每次都傳遞。
- 后端每次請求，都需要帶上目標db，可以采用放在ThreadLocal里的方式。但ThreadLocal有線程透傳的問題，如果任務里開啟了子線程，則變量不能共享。
- 由于表是動態選擇的，所以JPA自動創建和update等模式，將不可用。不方便測試和單元測試，在測試接口的時候，也需要每次強制指定指向的庫。
- 由于是修改數據源的模式，每次增加庫，都需要重新啟動上線才可以。如果要做到動態性，數據源銷毀是個問題。
End
對于一個微服務來說，有很多默認的限制策略，比如，不同域之間的服務是不能共享一個數據庫的。這些基本原則，把微服務整的清清爽爽，是一些基本的原則。
同理的，如果我們在設計開始，就給每一張表加上租戶的字段ID，那么寫代碼的時候就順暢的多。但是世界上沒有這么多如果。
原則為何而存在?當然是為了讓人去打破的。
編程只是工具，反正代碼在自己手里，怎么玩，看需要，也看心情。條條大路通羅馬，曲徑通幽處，風光無限好。
作者簡介：小姐姐味道 (xjjdog)，一個不允許程序員走彎路的公眾號。聚焦基礎架構和Linux。十年架構，日百億流量，與你探討高并發世界，給你不一樣的味道。
