SpringBoot - 多個數據源的輕鬆支持

前面文章裏介紹了 SpringBoot - 自定義註解完成數據庫切庫，今天接着這個高併發的話題，繼續說一下項目裏多個數據源的支持。

如何理解支持多個數據源呢？簡單的說，就是一個項目裏，同時可以訪問多個不同的數據庫。

實現的原理先交待一下：單個數據源在配置時會綁定一套mybatis配置，多個數據源時，不同的數據源綁定不同的mybatis配置就可以了，簡單的思路就是讓不同的數據源掃描不同的包，讓不同的包下的mapper對應連接不同的數據源去處理邏輯。

場景假設：項目底層有正常業務庫和日誌庫，希望解決的是將項目中的一些日誌單獨記錄到一個庫裏，比如用戶操作記錄、產品更新記錄等。

說一下爲什麼會有這個需求：用戶操作記錄和產品更新記錄可能很多，而實際中使用的又很少，就只是在某些頁面單獨展示一下操作或更新記錄，絕大部分時間都在不停的做着插入操作，這時就可以把這種記錄放到業務核心庫外面。

自己還遇到一個場景，就是底層產品、訂單什麼的是存在不同的庫裏的，但是代碼重構還沒做到產品相關的一個項目、訂單相關的一個項目這一步，這時候也可以考慮在一個項目裏同時支持多個數據源，訂單相關的類操作訂單庫，產品相關的類操作產品庫，前期做好配置就可以了，儘量別去跨庫join表，基本什麼都不影響。

不多說了，代碼走起來 : （依舊使用springboot進行實現）

第一步、定義多個數據源的mybatis配置

application.properties 

mybatis.mapper-locations=mappers/*.xml
mybatisLog.mapper-locations=mappersLog/*.xml

## datasource master #
spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.driver-class-name=com.mysql.jdbc.Driver
spring.datasource.url=jdbc:mysql://localhost:3306/test1?characterEncoding=UTF-8
spring.datasource.username=root
spring.datasource.password=466420182

## datasource log #
spring.datasourceLog.type=com.alibaba.druid.pool.DruidDataSource
spring.datasourceLog.driver-class-name=com.mysql.jdbc.Driver
spring.datasourceLog.url=jdbc:mysql://localhost:3306/log?characterEncoding=UTF-8
spring.datasourceLog.username=root
spring.datasourceLog.password=466420182

第二步、定義多個數據源

@Configuration
public class DatasourceConfig {

    @Bean(destroyMethod =  "close", name = DataSources.MASTER_DB)
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource dataSource() {
        return DataSourceBuilder.create().type(DruidDataSource.class).build();
    }

    @Bean(destroyMethod =  "close", name = DataSources.LOG_DB)
    @ConfigurationProperties(prefix = "spring.datasourceLog")
    public DataSource dataSourceLog() {
        return DataSourceBuilder.create().type(DruidDataSource.class).build();
    }
}

 第三步、分別配置多個數據源

@Configuration
@MapperScan(basePackages = {"com.mmall.practice.dao"})
public class MybatisConfig {

    @Autowired
    @Qualifier(DataSources.MASTER_DB)
    private DataSource masterDB;

    @Bean
    @Primary
    @ConfigurationProperties(prefix = "mybatis")
    public SqlSessionFactoryBean sqlSessionFactoryBean() {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(masterDB);
        return sqlSessionFactoryBean;
    }
}

@Configuration
@MapperScan(basePackages = {"com.mmall.practice.daoLog"}, sqlSessionFactoryRef = "logSqlSessionFactory")
public class MybatisLogConfig {

    @Autowired
    @Qualifier("logDB")
    private DataSource logDB;

    @Bean(name = "logSqlSessionFactory")
    @ConfigurationProperties(prefix = "mybatisLog")
    public SqlSessionFactoryBean sqlSessionFactoryBean() {
        SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(logDB);
        return sqlSessionFactoryBean;
    }
}

這裏需要注意兩個數據源配置的差別，也是支持多數據源的關鍵

1）Configuration 掃描不同的前綴，取不同包下的sql對應的xml文件

2）SqlSessionFactoryBean 實例化時，默認的額外添加了 @Primary註解

3）MapperScan 掃描的不同的包，如果掃描相同的包也能做，但是還需要做額外的配置，可以自己嘗試

4）不同的數據源使用不同的SqlSessionFactoryBean實例

至此，不同包下面的 Mapper.java 文件就可以連接不同的數據源了。這裏就不說如何去使用了，和之前正常一樣去使用 Mapper.java 就可以了，只是操作的是不同的數據庫。

多數據源支持是不是特別簡單，趕緊在你的項目裏用起來吧~