大家在日常后端開發(fā)過程，不可避免的會接觸到需要用到配置多個數(shù)據(jù)源的場景，在這里，小編介紹一種簡單方便的，只需要簡單的配置和一個@DS注解就能實(shí)現(xiàn)動態(tài)數(shù)據(jù)源的方式，這種動態(tài)數(shù)據(jù)源底層原理是基于Mybatis-plus來實(shí)現(xiàn)的。

1、配置方式

首先是pom.xml

<dependency>    
    <groupId>com.baomidou</groupId>    
    <artifactId>dynamic-datasource-spring-boot-starter</artifactId>    
    <version>3.4.1</version>
</dependency>

隨后配上多數(shù)據(jù)源，我們的項目中分別配置了Mysql和Doris兩種數(shù)據(jù)庫

spring:  
    datasource:    
        druid:      
            localdb:        
                url: xxx        
                username: xxx     
                driver-class-name: com.mysql.jdbc.Driver        
                type: com.alibaba.druid.pool.DruidDataSource      
            doris:       
                url: xxx        
                username: xxx
                driver-class-name: com.mysql.jdbc.Driver        
                type: com.alibaba.druid.pool.DruidDataSource

最后需要使用哪個數(shù)據(jù)庫，在對應(yīng)的方法上加上對應(yīng)的@DS（“數(shù)據(jù)庫名”）注解就可以了，在類上加@DS（“數(shù)據(jù)庫名”），那么整個類下的所有方法都會使用到這個數(shù)據(jù)源，實(shí)現(xiàn)方式是不是非常簡單。

@Service
@DS("localdb")
public class testServiceImpl extends BaseServiceImpl<testMapper, testBean> implements testService {

    public void save(String arg1) {
       //TODO
    }
    
    public void find(String arg2) {
       //TODO
    }
}

2、底層實(shí)現(xiàn)原理

底層實(shí)現(xiàn)核心類是
com.baomidou.dynamic.datasource.DynamicRoutingDataSource，繼承自AbstractDataSource

項目在初始化的時候會調(diào)用DynamicRoutingDataSource里面的public synchronized void addDataSource(String ds, DataSource dataSource)方法加載數(shù)據(jù)源，數(shù)據(jù)源存進(jìn)dataSourceMap中。

隨后，如果進(jìn)行數(shù)據(jù)庫操作，以方法為最小粒度，執(zhí)行對應(yīng)方法時，會被
DynamicDataSourceAnnotationInterceptor攔截器攔截

執(zhí)行determineDatasource方法，掃描加了@DS注解的類或者方法，隨后調(diào)用DynamicDataSourceContextHolder.poll方法。

DynamicDataSourceContextHolder.poll方法將當(dāng)前線程的數(shù)據(jù)源名加到對應(yīng)的ThreadLocal線程本地中

在之后的數(shù)據(jù)操作中，會調(diào)用org.springframework.jdbc.datasource.getConnection()方法，ThreadLocal中獲取之前攔截器存進(jìn)去動態(tài)數(shù)據(jù)源名，如果沒有獲取到，就默認(rèn)為配置的primary數(shù)據(jù)源，這完成了對應(yīng)的動態(tài)數(shù)據(jù)源切換。

3、總結(jié)

基于Mybatis-plus的@DS注解可以實(shí)現(xiàn)動態(tài)數(shù)據(jù)源切換，具有實(shí)現(xiàn)簡單的優(yōu)點(diǎn)，雖然沒有AOP實(shí)現(xiàn)，但是用到了AOP的思想，后續(xù)小編將寫一篇基于AOP的實(shí)現(xiàn)方式，歡迎大家關(guān)注。

作者：京東保險 郭盼

來源：京東云開發(fā)者社區(qū) 轉(zhuǎn)載請注明來源