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

心動不如行動。接下來，就讓我們來看一下它的具體實(shí)現(xiàn)吧。

1.基本原理

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

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

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

@Controller
public?class?ARDTestController?{
????@GetMapping("test")
????public?void?chifeng(){
????????//db-a?應(yīng)該是上層傳遞下來的屬性，我們可以把它放在ThreadLocal里
????????DataSourceContextHolder.setDbKey("db-a");
????}
}

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

在Java中，可以使用ThreadLocal綁定這個透傳的屬性。像Spring的嵌套事務(wù)等實(shí)現(xiàn)的原理，也是基于ThreadLocal去運(yùn)行的。所以，DataSourceContextHolder.本質(zhì)上是一個操作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兩個數(shù)據(jù)庫。

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;
????}
}

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

public?class?DynamicDataSource?extends?AbstractRoutingDataSource?{
????@Override
????protected?Object?determineCurrentLookupKey()?{
????????return?DataSourceContextHolder.getDbKey();
????}
}

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

@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?默認(rèn)數(shù)據(jù)庫需要設(shè)置
????????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.問題

通過以上簡單的代碼，就可以實(shí)現(xiàn)Spring簡單的多數(shù)據(jù)源管理。但明顯的，它還存在很多問題。

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

總結(jié)

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

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

原則為何而存在？當(dāng)然是為了讓人去打破的。

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

到此這篇關(guān)于Spring多租戶數(shù)據(jù)源管理 AbstractRoutingDataSource的文章就介紹到這了,更多相關(guān)AbstractRoutingDataSource內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: