【应用】SpringBoot -- 基于 MyBatis-Plus 实现多数据源配置

基本概述

日常开发中，随着数据量的增加以及各种不同类型的数据的出现，单一的数据库已经不能满足开发的需要，因此，一个项目当中往往需要对多个数据库进行存储以及查询的操作。

基于上述原因，本 demo 中学习并记录了如何使用 MyBatis-Plus 实现多数据源的配置和使用。本文配置了三个数据源，分别为 MySQL 数据库、PostgreSQL 数据库以及 Doris 数据库。

多数据源配置

创建 maven 项目，在 pom 文件中引入依赖，多数据源配置的核心依赖为dynamic-datasource-spring-boot-starter

    <parent>
        <groupId>org.springframework.bootgroupId>
        <artifactId>spring-boot-starter-parentartifactId>
        <version>2.2.2.RELEASEversion>
        <relativePath/> 
    parent>

    <dependencies>
        
        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-webartifactId>
        dependency>
        
        <dependency>
            <groupId>com.baomidougroupId>
            <artifactId>mybatis-plus-boot-starterartifactId>
            <version>3.1.0version>
        dependency>
        
        <dependency>
            <groupId>com.baomidougroupId>
            <artifactId>dynamic-datasource-spring-boot-starterartifactId>
            <version>3.5.1version>
        dependency>
        
        <dependency>
            <groupId>mysqlgroupId>
            <artifactId>mysql-connector-javaartifactId>
        dependency>
        
        <dependency>
            <groupId>org.postgresqlgroupId>
            <artifactId>postgresqlartifactId>
            <version>42.2.20version>
        dependency>
        
        <dependency>
            <groupId>com.alibabagroupId>
            <artifactId>druid-spring-boot-starterartifactId>
            <version>1.1.21version>
        dependency>
        
        <dependency>
            <groupId>org.projectlombokgroupId>
            <artifactId>lombokartifactId>
            <scope>providedscope>
        dependency>
        
        <dependency>
            <groupId>com.alibabagroupId>
            <artifactId>fastjsonartifactId>
            <version>1.2.55version>
        dependency>
    dependencies>

在 application.yml 文件中配置多数据源，datasource 属性的下一级配置的是数据源的名称

server:
  port: 8080

spring:
  datasource:
    dynamic:
      # 指定默认数据源
      primary: mysql
      # true：找不到数据源报错
      strict: false
      datasource:
        mysql:
          driver-class-name: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=UTC&nuLlCatalogMeansCurrent=true
          username: root
          password: >
        pgsql:
          driver-class-name: org.postgresql.Driver
          url: jdbc:postgresql://localhost:5432/postgres
          username: postgres
          password: >
        doris:
          driver-class-name: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://-server-ip>:9030/example_db?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=UTC&nuLlCatalogMeansCurrent=true
          username: root
          password: >

mybatis-plus:
  mapper-locations: classpath*:mapper/*.xml

创建自定义注解，用于指定 mapper 文件的数据源，@DS()括号内的值为配置文件中所配置的数据源名称。

MySQL 数据库自定义注解：

@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@DS("mysql")
public @interface MySQL {
}

PostgreSQL 数据库自定义注解：

@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@DS("pgsql")
public @interface PostgreSQL {
}

Doris 数据库自定义注解

@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@DS("doris")
public @interface Doris {
}

在不同的 mapper 文件中使用自定义注解指定数据源，即可实现多数据源的使用

MySQLMapper

@Mapper
@MySQL
public interface MySQLMapper extends BaseMapper<User> {

    User selectByUsername(@Param("username") String username);

}

PostgreSQLMapper

@Mapper
@PostgreSQL
public interface PostgreSQLMapper extends BaseMapper<User> {

    User selectByUsername(@Param("username") String username);

}

DorisMapper

@Mapper
@Doris
public interface DorisMapper extends BaseMapper<DataInfo> {

    List<DataInfo> selectAll();

}