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

數(shù)據(jù)源

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

@DS

我們先來看看這個注解的使用

<!--配置多數(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
        # 配置獲取連接等待超時的時間
        maxWait: 60000
        # 配置間隔多久才進(jìn)行一次檢測，檢測需要關(guān)閉的空閑連接，單位是毫秒
        timeBetweenEvictionRunsMillis: 6000
        # 配置一個連接在池中最小生存的時間，單位是毫秒
        minEvictableIdleTimeMillis: 60000
        # 配置一個連接在池中最大生存的時間，單位是毫秒
        maxEvictableIdleTimeMillis: 900000
        validationQuery: SELECT 1 FROM DUAL
        testWhileIdle: true
        testOnBorrow: false
        testOnReturn: false
        # 打開PSCache，并且指定每個連接上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ù)源時拋異常,false使用默認(rèn)數(shù)據(jù)源
      datasource:
        master:
          driver-class-name: com.mysql.cj.jdbc.Driver # 3.2.0開始支持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開始支持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

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

@DS("slave_1")

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

圖片

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

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

圖片

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

圖片

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

進(jìn)行數(shù)據(jù)操作時，會調(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ù)操作。

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