前言

別誤會(huì)，本篇【并不是】 要用mybatis-plus自身的多租戶方案：在表中加一個(gè)tenant_id字段來(lái)區(qū)分不同的租戶數(shù)據(jù)。并不是的！
而是在假設(shè)業(yè)務(wù)系統(tǒng)已經(jīng)使用mybatis-plus多數(shù)據(jù)源的前提下，如何實(shí)現(xiàn)業(yè)務(wù)數(shù)據(jù)庫(kù)隔開(kāi)的多租戶系統(tǒng)。
這里面有點(diǎn)繞：多數(shù)據(jù)源可以是一個(gè)系統(tǒng)本身的功能需求，假設(shè)當(dāng)前系統(tǒng)算做是個(gè)單租戶，它使用了兩個(gè)數(shù)據(jù)庫(kù): master1和sys1，那么做多租戶改造后，假設(shè)現(xiàn)在有了2個(gè)租戶，那么就要添加2個(gè)數(shù)據(jù)庫(kù)：master2和sys2 , 總共就是四個(gè)數(shù)據(jù)庫(kù)（數(shù)據(jù)源）了…
咱們這里簡(jiǎn)單化處理，假設(shè)一個(gè)業(yè)務(wù)系統(tǒng)只使用一個(gè)數(shù)據(jù)庫(kù)。

大綱

在本篇我們可以

• 看到mybatis-plus底層多數(shù)據(jù)源的實(shí)現(xiàn)原理
• 在不破壞多數(shù)據(jù)源的前提下，實(shí)現(xiàn)多租戶功能
• spring security結(jié)合jwt記錄租戶信息

代碼版本：

springboot: 2.7.0
dynamic-datasource-spring-boot-starter: 4.3.0
io.jsonwebtoken: 0.12.3

回顧mybatis-plus多數(shù)據(jù)源使用

1.yaml配置:

2.serviceImpl:

或者使用切面動(dòng)態(tài)設(shè)置crud對(duì)應(yīng)的數(shù)據(jù)源。

改造需求

• 不要把所有租戶信息都直接放在yaml等配置文件中
• 可動(dòng)態(tài)的添加刪除數(shù)據(jù)源
• 用戶登錄成功后，把租戶信息封裝到j(luò)wt token中，后續(xù)業(yè)務(wù)訪問(wèn)提取中租戶信息，動(dòng)態(tài)切換數(shù)據(jù)源訪問(wèn)

方案

租戶本身的信息可放在resources/tenants目錄下，一個(gè)租戶使用一個(gè)單獨(dú)的配置文件，或者通過(guò)讀取另外的數(shù)據(jù)庫(kù)獲取。本篇先使用前者。
修改某個(gè)租戶的配置文件內(nèi)容/數(shù)據(jù)庫(kù)，重啟服務(wù)/通過(guò)controller接口觸發(fā)數(shù)據(jù)源的變更。

正篇開(kāi)始
yaml配置文件中只保留一個(gè)主數(shù)據(jù)庫(kù)，如上yaml截圖所示。
另外的2個(gè)租戶配置放classpath下tenant目錄，如下所示：

新建多數(shù)據(jù)源配置類(lèi)，內(nèi)容如下：

@Configuration(proxyBeanMethods = false)
@Slf4j
public class MultiDataSourceConfig {@Resourceprivate DruidDataSourceCreator druidDataSourceCreator;@Resourceprivate DynamicRoutingDataSource dataSource;@PostConstructpublic void init() {ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();try {org.springframework.core.io.Resource[] resources = resolver.getResources("classpath:tenants/*.properties");for (org.springframework.core.io.Resource resource : resources) {Properties tenantProperties = new Properties();tenantProperties.load(resource.getInputStream());String tenantId = tenantProperties.getProperty("name");DataSourceProperty dataSourceProperty = new DataSourceProperty();dataSourceProperty.setPoolName(tenantId);dataSourceProperty.setUrl(tenantProperties.getProperty("datasource.url"));dataSourceProperty.setUsername(tenantProperties.getProperty("datasource.username"));dataSourceProperty.setPassword(tenantProperties.getProperty("datasource.password"));dataSourceProperty.setDriverClassName(tenantProperties.getProperty("datasource.driver-class-name"));dataSource.addDataSource(tenantId, druidDataSourceCreator.createDataSource(dataSourceProperty));}} catch (IOException exp) {throw new RuntimeException("Problem in tenant datasource:" + exp);}}
}

上述代碼讀取classpath:tenants/目錄下的所有.properties配置文件內(nèi)容，組裝并添加數(shù)據(jù)源。

登錄

登錄操作時(shí)，查詢此登錄用戶對(duì)應(yīng)的租戶信息，并在生成jwt token時(shí)，把租戶信息也封裝進(jìn)去，通過(guò)http響應(yīng)頭返回給用戶。如下所示：

public static void addToken(HttpServletResponse res, String username, String tenant) {String JwtToken = Jwts.builder().subject(username).audience().add(tenant).and().issuedAt(new Date(System.currentTimeMillis())).expiration(new Date(System.currentTimeMillis() + EXPIRATIONTIME)).signWith(SIGNINGKEY).compact();res.addHeader("Authorization", PREFIX + JwtToken);}

業(yè)務(wù)操作

在一個(gè)攔截器類(lèi)中：token校驗(yàn)，并從中提取出租戶信息，如下所示：

public static String getTenant(HttpServletRequest req) {String token = req.getHeader("Authorization");if (token == null) {return null;}String tenant = Jwts.parser().setSigningKey(SIGNINGKEY).build().parseClaimsJws(token.replace(PREFIX, "").trim()).getBody().getAudience().iterator().next();return tenant;}

動(dòng)態(tài)切換數(shù)據(jù)源

得到一個(gè)請(qǐng)求所屬租戶信息后，要訪問(wèn)數(shù)據(jù)庫(kù)時(shí)切換源：記住這里
記住它：DynamicDataSourceContextHolder

    DynamicDataSourceContextHolder.push(tenant);try {chain.doFilter(request, response);} finally {DynamicDataSourceContextHolder.clear();}

完工！是的，使用層面上就結(jié)束了。接下來(lái)是原理分析。

剖析

1.有個(gè)類(lèi)名叫：AbstractRoutingDataSource，mybatis-plus和spring-jdbc都有叫這個(gè)名的類(lèi)，
并且它們都繼承了AbstractDataSource類(lèi)，但是這個(gè)父類(lèi)也只是同名而已。但是它們的功能都說(shuō)得很清楚：抽象動(dòng)態(tài)獲取數(shù)據(jù)源，它們都有個(gè)抽象方法：抽象獲取連接池，如下所示：

spring-jdbc下的源碼：

然后看看mybatis-plus的抽象方法實(shí)現(xiàn)：


所以回顧我們業(yè)務(wù)代碼的寫(xiě)法：DynamicDataSourceContextHolder.push(tenant);
正是我們把當(dāng)前請(qǐng)求對(duì)應(yīng)的tenant作為數(shù)據(jù)源key 壓棧了，后面切換數(shù)據(jù)源時(shí)依據(jù)它去得到數(shù)據(jù)源。那么還記得這個(gè)租戶key 是在哪里和數(shù)據(jù)源對(duì)應(yīng)上的嗎？
正是在正篇開(kāi)頭的新建多數(shù)據(jù)源配置類(lèi)中：

dataSource.addDataSource(tenantId, druidDataSourceCreator.createDataSource(dataSourceProperty));

    /*** 添加數(shù)據(jù)源** @param ds         數(shù)據(jù)源名稱(chēng)* @param dataSource 數(shù)據(jù)源*/public synchronized void addDataSource(String ds, DataSource dataSource) {DataSource oldDataSource = dataSourceMap.put(ds, dataSource);// 新數(shù)據(jù)源添加到分組this.addGroupDataSource(ds, dataSource);// 關(guān)閉老的數(shù)據(jù)源if (oldDataSource != null) {closeDataSource(ds, oldDataSource, graceDestroy);}log.info("dynamic-datasource - add a datasource named [{}] success", ds);}

如此就打通了流程。如果大家感興趣，可以看到com.baomidou.dynamic.datasource.DynamicRoutingDataSource類(lèi)中有一些方法刪除數(shù)據(jù)源，還有數(shù)據(jù)源分組功能，這可以用于主主(從，如果業(yè)務(wù)場(chǎng)景都是只讀的話)，策略是輪詢和隨機(jī)：