本文分享自華為雲社區 《實戰指南，SpringBoot + Mybatis 如何對接多數據源》，作者：戰斧。

在我們開發一些具有綜合功能的項目時，往往會碰到一種情況，需要同時連接多個資料庫，這個時候就需要用到多數據源的設計。而Spring與Myabtis其實做了多數據源的適配，只需少許改動即可對接多數據源。本期我們就貼近實戰，以一個單數據源的Demo為例，講述將其改為多數據源項目的過程，希望大家能有所體會。

一、數據源的定義

二、單數據源配置

因為SpringBoot對數據源有著高度的預設配置，只配置一個數據源時，該數據源會被作為預設，所以對接單數據源其實是非常簡單的。如果你的工程採用的yaml格式配置文件，我們僅需做如下配置：

spring:
  #資料庫連接配置
  datasource:
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://127.0.0.1:3306/springtest
    username: root
    password: root

如果是採用properties配置文件的也是一樣的：

spring.datasource.driver-class-name=com.mysql.jdbc.Driver
spring.datasource.url=jdbc:mysql://127.0.0.1:3306/springtest
spring.datasource.username=root
spring.datasource.password=root

三、如何配置多數據源

1. 工程層級調整

我們以曾經搭建的工程為原始模板，進行對接多數據源的操作。沒看過的可以點此查看： 從零開始，手把手教你搭建Spring Boot後臺工程並說明

因為僅變動數據源，所以我們不改動其他層級，僅僅將 mapper 拆為 mapper1 與 mapper2 兩部分

2. Spring項目配置

然後我們需要在 application.properties 或者 application.yml 中定義多個數據源：

spring:
  #資料庫連接配置
  datasource1:
    driver-class-name: com.mysql.cj.jdbc.Driver
    jdbc-url: jdbc:mysql://127.0.0.1:3306/springtest2
    username: root
    password: root
  datasource2:
    driver-class-name: com.mysql.cj.jdbc.Driver
    jdbc-url: jdbc:mysql://127.0.0.1:3306/springtest
    username: root
    password: root

這裡有兩個細節需要註意：

• 因為我們決定使用雙數據源，所以把數據源的連接配置改成了datasource1 和 datasource2。而不再保留datasource，這樣SpringBoot就不再會為我們設定預設資料庫
• 因為我們目前採用的 springBoot2.5.2，預設的連接池為Hikari，該連接池數據源的地址欄位為jdbc-url 而非 url。在只有單個數據源時，SpringBoot走預設數據源邏輯為我們把 url 與 jdbc-url 進行映射，保證我們獲得數據源。此時我們自己設置的數據源沒有進行映射處理，就需要保證欄位符合Hikari的要求。否則會出現 java.lang.IllegalArgumentException: jdbcUrl is required with driverClassName 異常

3. 會話配置

僅有配置文件可不行，接下來，我們需要在代碼中讀取到配置，並建立兩個數據源。如下，每個數據源都有隔離的mapper介面、xml文件、會話工廠及會話模板

第一個數據源 如下（示例）：

@Configuration
@MapperScan(basePackages = "com.zhanfu.springboot.demo.mapper1", sqlSessionFactoryRef = "sqlSessionFactory1")
public class DataSource1Config {

    @Bean
    @ConfigurationProperties(prefix = "spring.datasource1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public SqlSessionFactory sqlSessionFactory1() throws Exception {
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource1());
        String locationPattern = "classpath*:/mapper1/*.xml";
        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        sessionFactoryBean.setMapperLocations(resolver.getResources(locationPattern));
        return sessionFactoryBean.getObject();
    }

    @Bean(name = "sqlSessionTemplate1")
    public SqlSessionTemplate sqlSessionTemplate1(@Qualifier("sqlSessionFactory1") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

配置第二個數據源 DataSource2Config

@Configuration
@MapperScan(basePackages = "com.zhanfu.springboot.demo.mapper2", sqlSessionFactoryRef = "sqlSessionFactory2")
public class DataSource2Config {
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().build();
    }

    @Bean
    public SqlSessionFactory sqlSessionFactory2() throws Exception {
        SqlSessionFactoryBean sessionFactoryBean = new SqlSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource2());
        String locationPattern = "classpath*:/mapper2/*.xml";
        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        sessionFactoryBean.setMapperLocations(resolver.getResources(locationPattern));
        return sessionFactoryBean.getObject();
    }

    @Bean(name = "sqlSessionTemplate2")
    public SqlSessionTemplate sqlSessionTemplate2(@Qualifier("sqlSessionFactory2") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

    @Bean(name = "productMapper")
    public ProductMapper mapper2(@Qualifier("sqlSessionTemplate2") SqlSessionTemplate sqlSessionTemplate) throws Exception {
        return sqlSessionTemplate.getMapper(ProductMapper.class);
    }
}

4. 事務管理器

為兩個數據源分別配置自己的事務管理器，如果你的項目里通篇沒有方法級別的事務（一個SQL就是一個事務），那不設置這個也不影響，否則還是建議加上。

@Configuration
public class TransactionManagerConfig {

    @Autowired
    private DataSource dataSource1;

    @Autowired
    private DataSource dataSource2;

    @Bean
    public PlatformTransactionManager txManager1() {
        return new DataSourceTransactionManager(dataSource1);
    }

    @Bean
    public PlatformTransactionManager txManager2() {
        return new DataSourceTransactionManager(dataSource2);
    }
}

四、驗證

我們把兩張表拆進兩個庫中，以兩個庫模擬兩個數據源，使得程式可以同時連接兩個庫

瀏覽器輸入 http://127.0.0.1:8080/user/findall 查詢介面成功

再在瀏覽器輸入 http://127.0.0.1:8080/product/findall 查詢第二個庫的數據亦成功返回

這樣我們就完成了一個工程同時連接兩個數據源。

總結

經過上述的操作，我們已經成功把項目對接了多數據源。當然，方案肯定不止這一種，後續圍繞該問題，我們還會講解其他方式。但不論是什麼方式，主旨都是加深大家對SpringBoot 和 Mybatis的理解，我們曾經梳理過全流程，但只是蜻蜓點水帶大家看一遍大體輪廓，並不足以讓你精通，後面本專欄將繼續深入講解。

點擊關註，第一時間瞭解華為雲新鮮技術~