1. 程式人生 > 程式設計 >Springboot +Mybatis實現多資料來源配置

Springboot +Mybatis實現多資料來源配置

前言:

隨著應用使用者數量的增加,相應的併發請求的數量也會跟著不斷增加,慢慢地,單個資料庫已經沒有辦法滿足我們頻繁的資料庫操作請求了,在某些場景下,我們可能會需要配置多個資料來源,使用多個資料來源(例如實現資料庫的讀寫分離)來緩解系統的壓力等,同樣的,Springboot官方提供了相應的實現來幫助開發者們配置多資料來源,一般分為兩種方式(目前我所瞭解到的),分包和AOP,其中利用AOP實現多個資料來源到的動態切換時候會另開一篇文章來寫。考慮到mybatis是java開發者們使用較為頻繁的資料庫框架,所以本篇文章使用Springboot+Mybatis來實現多資料來源的配置。

廢話不多說,走起。

1. 資料庫準備:

既然是配置多資料來源,那麼我們自然就要先把相應的資料來源給準備好,這裡呢,我本地新建了兩個資料庫,如下表:

資料庫 testdatasource1 testdatasource2
資料表 sys_user sys_user2
欄位 user_id(int),user_name(varchar) user_age(int)

並分別插入兩條記錄,為了方便對比,其中testdatasource1為芳年25歲的張三, testdatasource2為芳年30歲的李四。

2.環境準備

首先新建一個Springboot專案,我這裡版本是2.1.7.RELEASE,並在pom檔案中引入相關依賴:關鍵依賴如下:

        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>1.3.2</version>
        </dependency>

        <dependency>
            <groupId>
mysql</groupId> <artifactId>mysql-connector-java</artifactId> </dependency> 複製程式碼

到這裡我們的環境已經基本配置完成了。

3.程式碼部分

1. 多資料來源配置

首先呢,在我們Springboot的配置檔案中配置我們的datasourse,和以往不一樣的是,因為我們有兩個資料來源,所以要指定相關資料庫的名稱,其中主資料來源為primary,次資料來源為secondary如下:

#配置主資料庫
spring.datasource.primary.jdbc-url=jdbc:mysql://localhost:3306/testdatasource1?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC&useSSL=false
spring.datasource.primary.username=root
spring.datasource.primary.password=root
spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver

##配置次資料庫
spring.datasource.secondary.jdbc-url=jdbc:mysql://localhost:3306/testdatasource2?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC&useSSL=false
spring.datasource.secondary.username=root
spring.datasource.secondary.password=root
spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver


spring.http.encoding.charset=UTF-8
spring.http.encoding.enabled=true
spring.http.encoding.force=true
複製程式碼

需要我們注意的是,Springboot2.0 在配置資料庫連線的時候需要使用jdbc-url,如果只使用url的話會報

jdbcUrl is required with driverClassName.錯誤。

新建一個配置類PrimaryDataSourceConfig,用於配置我們的主資料庫相關的bean,程式碼如下:

@Configuration
@MapperScan(basePackages = "com.jdkcb.mybatisstuday.mapper.one",sqlSessionFactoryRef = "PrimarySqlSessionFactory")//basePackages:介面檔案的包路徑
public class PrimaryDataSourceConfig {

    @Bean(name = "PrimaryDataSource")
    // 表示這個資料來源是預設資料來源
    @Primary//這個一定要加,如果兩個資料來源都沒有@Primary會報錯
    @ConfigurationProperties(prefix = "spring.datasource.primary")//我們配置檔案中的字首
    public DataSource getPrimaryDateSource() {
        return DataSourceBuilder.create().build();
    }


    @Bean(name = "PrimarySqlSessionFactory")
    @Primary
    public SqlSessionFactory primarySqlSessionFactory(@Qualifier("PrimaryDataSource") DataSource datasource)
            throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(datasource);
        bean.setMapperLocations( 
                new PathMatchingResourcePatternResolver().getResources("classpath*:mapping/one/*.xml"));
        return bean.getObject();// 設定mybatis的xml所在位置
    }
    
    
    @Bean("PrimarySqlSessionTemplate")
    // 表示這個資料來源是預設資料來源
    @Primary
    public SqlSessionTemplate primarySqlSessionTemplate(
            @Qualifier("PrimarySqlSessionFactory") SqlSessionFactory sessionfactory) {
        return new SqlSessionTemplate(sessionfactory);
    }

}

複製程式碼

註解說明:

@MapperScan :配置mybatis的介面類放的地方

@Primary :表示使用的是預設資料庫,這個一個要加,否則會因為不知道哪個資料庫是預設資料庫而報錯

@ConfigurationProperties:讀取application.properties中的配置引數對映成為一個物件,其中prefix表示引數的字首

大功告成~ ~ 了嗎?並沒有,然後配置我們的第二個資料來源的配置類,程式碼如下:

@Configuration
@MapperScan(basePackages = "com.jdkcb.mybatisstuday.mapper.two",sqlSessionFactoryRef = "SecondarySqlSessionFactory")
public class SecondaryDataSourceConfig {

    @Bean(name = "SecondaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.secondary")
    public DataSource getSecondaryDataSource() {
        return DataSourceBuilder.create().build();
    }
    @Bean(name = "SecondarySqlSessionFactory")
    public SqlSessionFactory secondarySqlSessionFactory(@Qualifier("SecondaryDataSource") DataSource datasource)
            throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(datasource);
        bean.setMapperLocations(
                new PathMatchingResourcePatternResolver().getResources("classpath*:mapping/two/*.xml"));
        return bean.getObject();// 設定mybatis的xml所在位置
    }
    @Bean("SecondarySqlSessionTemplate")
    public SqlSessionTemplate secondarySqlSessionTemplate(
            @Qualifier("SecondarySqlSessionFactory") SqlSessionFactory sessionfactory) {
        return new SqlSessionTemplate(sessionfactory);
    }

複製程式碼

剩下的就是編寫我們相應的xml檔案和介面類了,程式碼如下:

@Component
@Mapper
public interface PrimaryUserMapper {
    List<User> findAll();
}


@Component
@Mapper
public interface SecondaryUserMapper {
    List<User> findAll();
}
複製程式碼

相關的xml檔案如下:

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.jdkcb.mybatisstuday.mapper.one.PrimaryUserMapper">

    <select id="findAll" resultType="com.jdkcb.mybatisstuday.pojo.User">
                select * from sys_user;
    </select>
</mapper>


<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.jdkcb.mybatisstuday.mapper.two.SecondaryUserMapper">

    <select id="findAll" resultType="com.jdkcb.mybatisstuday.pojo.User">
                select * from sys_user2;
    </select>

</mapper>
複製程式碼

注:其中xml檔案在本例項中目錄為:resources/mapping

2.測試

編寫一個Controller用於測試,因為是測試例項且程式碼相對來說較為簡單,所以這裡就不寫Service層了。

程式碼如下:

@RestController
public class UserController {

    @Autowired
    private PrimaryUserMapper primaryUserMapper;
    @Autowired
    private SecondaryUserMapper secondaryUserMapper;
    @RequestMapping("primary")
    public Object primary(){
        List<User> list = primaryUserMapper.findAll();
        return list;
    }
    @RequestMapping("secondary")
    public Object secondary  (){
        List<User> list = secondaryUserMapper.findAll();
        return list;
    }
    

}
複製程式碼

在瀏覽器分別輸入:http://127.0.0.1:8080/primaryhttp://127.0.0.1:8080/secondary

結果如下:

[{"user_id":1,"user_name":"張三","user_age":25}] //primary 
[{"user_id":1,"user_name":"李四","user_age":30}] //secondary
複製程式碼

到此,Springboot結合mybatis配置多資料來源就大功告成啦。

最後的最後,大家好,我是韓數,哼,關注我,有你好果子吃(叉腰)。

記得點個贊再走哦~

等一下:

相關原始碼歡迎去我的github下載(歡迎star):

github.com/hanshuaikan…