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

基本概述

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

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

多数据源配置

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

    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.2.2.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>

    <dependencies>
        <!--web 模块-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!--mybatis plus-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.1.0</version>
        </dependency>
        <!--mybatis plus 多数据源-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
            <version>3.5.1</version>
        </dependency>
        <!--mysql 驱动-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        <!--postgresql组件-->
        <dependency>
            <groupId>org.postgresql</groupId>
            <artifactId>postgresql</artifactId>
            <version>42.2.20</version>
        </dependency>
        <!-- druid 连接池-->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid-spring-boot-starter</artifactId>
            <version>1.1.21</version>
        </dependency>
        <!--Lombok -->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <scope>provided</scope>
        </dependency>
        <!--fastjson-->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.55</version>
        </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: <password>
        pgsql:
          driver-class-name: org.postgresql.Driver
          url: jdbc:postgresql://localhost:5432/postgres
          username: postgres
          password: <password>
        doris:
          driver-class-name: com.mysql.cj.jdbc.Driver
          url: jdbc:mysql://<doris-server-ip>:9030/example_db?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=UTC&nuLlCatalogMeansCurrent=true
          username: root
          password: <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();

}