我們都知道，再開(kāi)發(fā)的過(guò)程中，對(duì)于多個(gè)數(shù)據(jù)庫(kù)，就會(huì)有各種各樣的數(shù)據(jù)源，比如Oracle，比如 Mysql,并且再開(kāi)發(fā)的過(guò)程中，我們很有可能會(huì)同時(shí)使用到兩個(gè)數(shù)據(jù)庫(kù)，這樣就設(shè)計(jì)到了切換不同的數(shù)據(jù)源來(lái)進(jìn)行實(shí)現(xiàn)，就比如我們?cè)?Oracle 的數(shù)據(jù)庫(kù)中去查詢一些必要的數(shù)據(jù)，然后讓這些源頭數(shù)據(jù)進(jìn)入到 Mysql ,這個(gè)時(shí)候，如果不通過(guò)第三方的工具，比如 dataWorks或者其他的組件的話， 那么就得通過(guò) Java 代碼來(lái)進(jìn)行實(shí)現(xiàn)了，今天了不起就來(lái)給大家介紹一下這個(gè)關(guān)于這個(gè)一個(gè)注解就切換數(shù)據(jù)源的操作。

數(shù)據(jù)源

數(shù)據(jù)源的意思是“數(shù)據(jù)庫(kù)應(yīng)用程序所使用的數(shù)據(jù)庫(kù)或者數(shù)據(jù)庫(kù)服務(wù)器”；數(shù)據(jù)源也即數(shù)據(jù)的來(lái)源，是提供某種所需要數(shù)據(jù)的器件或原始媒體，在數(shù)據(jù)源中存儲(chǔ)了所有建立數(shù)據(jù)庫(kù)連接的信息，通過(guò)提供正確的數(shù)據(jù)源名稱，可以找到相應(yīng)的數(shù)據(jù)庫(kù)連接。

@DS

我們先來(lái)看看這個(gè)注解的使用

<!--配置多數(shù)據(jù)源-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
            <version>3.5.0</version>
        </dependency>

修改YML 的配置如下

datasource:
    dynamic:
      druid:
        initial-size: 5
        min-idle: 5
        maxActive: 20
        # 配置獲取連接等待超時(shí)的時(shí)間
        maxWait: 60000
        # 配置間隔多久才進(jìn)行一次檢測(cè)，檢測(cè)需要關(guān)閉的空閑連接，單位是毫秒
        timeBetweenEvictionRunsMillis: 6000
        # 配置一個(gè)連接在池中最小生存的時(shí)間，單位是毫秒
        minEvictableIdleTimeMillis: 60000
        # 配置一個(gè)連接在池中最大生存的時(shí)間，單位是毫秒
        maxEvictableIdleTimeMillis: 900000
        validationQuery: SELECT 1 FROM DUAL
        testWhileIdle: true
        testOnBorrow: false
        testOnReturn: false
        # 打開(kāi)PSCache，并且指定每個(gè)連接上PSCache的大小
        poolPreparedStatements: true
        maxPoolPreparedStatementPerConnectionSize: 20
      primary: master #設(shè)置默認(rèn)的數(shù)據(jù)源或者數(shù)據(jù)源組,默認(rèn)值即為master
      strict: false #嚴(yán)格匹配數(shù)據(jù)源,默認(rèn)false. true未匹配到指定數(shù)據(jù)源時(shí)拋異常,false使用默認(rèn)數(shù)據(jù)源
      datasource:
        master:
          driver-class-name: com.mysql.cj.jdbc.Driver # 3.2.0開(kāi)始支持SPI可省略此配置
          url: jdbc:mysql://localhost:3306/table1?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&allowPublicKeyRetrieval=true
          username: xxxx
          password: xxxxx
        slave_1:
          driver-class-name: com.mysql.cj.jdbc.Driver # 3.2.0開(kāi)始支持SPI可省略此配置
          url: jdbc:mysql://localhost:3306/table2?serverTimezone=GMT%2B8&useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true&allowPublicKeyRetrieval=true
          username: xxxx
          password: xxxxx

注意這個(gè) dynamic 一定是我們引入的 mybatis 當(dāng)中的，這個(gè)時(shí)候，如果我們?nèi)ナ褂眠@個(gè)@DS注解的話，那么就可以去切換數(shù)據(jù)源，比如我們?cè)谂渲梦募校渲昧?master  還有一個(gè) slave_1 的數(shù)據(jù)源，那么你在使用的時(shí)候，可以這樣，在我們的方法，或者類上面

@DS("slave_1")

這個(gè)注解，你可以使用在你的方法上面，你也可以使用在你的類上面，（一般是用在mapper、service），使用過(guò)該注解的方法即可操作到對(duì)應(yīng)的數(shù)據(jù)源。既然我們知道怎么去使用這個(gè)注解了，那么就得來(lái)看看這個(gè)源碼的部分了。

圖片

首先mybatis-plus使用com.baomidou.dynamic.datasource.AbstractRoutingDataSource繼承 AbstractDataSource接管數(shù)據(jù)源；

具體實(shí)現(xiàn)類（ctrl+alt+b可以查看抽象類或者的具體實(shí)現(xiàn)類）為com.baomidou.dynamic.datasource.DynamicRoutingDataSource。項(xiàng)目初始化調(diào)用public synchronized void addDataSource(String ds, DataSource dataSource)加載數(shù)據(jù)源，數(shù)據(jù)源存進(jìn)dataSourceMap中。

圖片

當(dāng)我們發(fā)送具體的操作請(qǐng)求的時(shí)候，進(jìn)行數(shù)據(jù)操作時(shí)，方法會(huì)被com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationInterceptor攔截

圖片

攔截器首先從被攔截的方法或者類（一般@DS注解用于Service，也可用于Mapper和Controller）上尋找@DS注解，獲取到@DS注解的值后將其存入com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder；DynamicDataSourceContextHolder使用ThreadLocal存儲(chǔ)當(dāng)前線程的數(shù)據(jù)源名。

進(jìn)行數(shù)據(jù)操作時(shí)，會(huì)調(diào)用org.springframework.jdbc.datasource.getConnection()方法；

getConnection()方法最終調(diào)用了com.baomidou.dynamic.datasource.AbstractRoutingDataSource的getConnection()方法；

數(shù)據(jù)操作完成后，方法返回第二步中的攔截器，執(zhí)行DynamicDataSourceContextHolder.poll();清除掉此次Threadlocal中的數(shù)據(jù)源，避免影響后續(xù)數(shù)據(jù)操作。

但是還有一個(gè)問(wèn)題了，那就是不可在事務(wù)中切換數(shù)據(jù)庫(kù)，保證事務(wù)需要方法使用同一連接，使用@DS(dataSourceOne)方法調(diào)用@DS(dataSourceTwo)無(wú)法切換連接，會(huì)導(dǎo)致方法報(bào)錯(cuò)。