利用 Mybatis-Plus 的动态数据源实现多数据源配置

ゝ一纸荒年。 2024-03-22 17:50 80阅读 0赞

**目录**

一、导入依赖

二、Application.yaml配置文件

三、切换数据源

四、其他方法

4.1 配置多个数据源

4.2 定义Datasource和EntityManager

4.3 在需要使用数据源的地方注入不同的EntityManager

--------------------

官网：[https://baomidou.com/pages/a61e1b/\#dynamic-datasource][https_baomidou.com_pages_a61e1b_dynamic-datasource]

## 一、导入依赖 ##

<dependency>
      <groupId>com.baomidou</groupId>
      <artifactId>dynamic-datasource-spring-boot-starter</artifactId>
      <version>${version}</version>
    </dependency>

## 二、Application.yaml配置文件 ##

spring:
      datasource:
        dynamic:
          primary: master #设置默认的数据源或者数据源组,默认值即为master
          strict: false #严格匹配数据源,默认false. true未匹配到指定数据源时抛异常,false使用默认数据源
          datasource:
            master:
              url: jdbc:mysql://xx.xx.xx.xx:3306/dynamic
              username: root
              password: 123456
              driver-class-name: com.mysql.jdbc.Driver # 3.2.0开始支持SPI可省略此配置
            slave_1:
              url: jdbc:mysql://xx.xx.xx.xx:3307/dynamic
              username: root
              password: 123456
              driver-class-name: com.mysql.jdbc.Driver
            slave_2:
              url: ENC(xxxxx) # 内置加密,使用请查看详细文档
              username: ENC(xxxxx)
              password: ENC(xxxxx)
              driver-class-name: com.mysql.jdbc.Driver
           #......省略
           #以上会配置一个默认库master，一个组slave下有两个子库slave_1,slave_2
           #从库的名称可以自己随便去，但是主库必须为master
           #primary中设置的数据库就是默认访问的数据库
           #当然了，并不一定是主从复制集群才可以这样用，如果业务分成不同的数据库，单纯切换不同数据库也是可以的

# 多主多从                      纯粹多库（记得设置primary）                   混合配置
    spring:                               spring:                               spring:
      datasource:                           datasource:                           datasource:
        dynamic:                              dynamic:                              dynamic:
          datasource:                           datasource:                           datasource:
            master_1:                             mysql:                                master:
            master_2:                             oracle:                               slave_1:
            slave_1:                              sqlserver:                            slave_2:
            slave_2:                              postgresql:                           oracle_1:
            slave_3:                              h2:

## 三、切换数据源 ##

默认是使用配置文件中master参数设置的数据库。

但如果我们想使用其他数据源，则可以通过**@DS**注解来切换数据源。

**@DS** 可以注解在方法上或类上，**同时存在就近原则 方法上注解 优先于 类上注解**。

![90cd6f87a7884b48a8124b56c4c8752b.png][]

@Service
    @DS("slave")
    public class UserServiceImpl implements UserService {
    
      @Autowired
      private JdbcTemplate jdbcTemplate;
    
      public List selectAll() {
        return  jdbcTemplate.queryForList("select * from user");
      }
      
      @Override
      @DS("slave_1")
      public List selectByCondition() {
        return  jdbcTemplate.queryForList("select * from user where age >10");
      }
    }

## 四、其他方法 ##

除了使用MybatisPlus的动态数据源，还有很多其他方法，比如shardingsphere等等。

考虑到如果只是简单切换数据源进行crud。其实没必要使用shardingsphere，其使用成本稍微有点高。那么使用mybatisplus的动态数据源，是一个不错的选择！

当然了，这里还提供另一种方法，就是自己来编写数据源配置。

### 4.1 配置多个数据源 ###

可以通过application.yml或者application.properties文件配置不同的数据源连接信息，例如：

# 数据源1
    spring.datasource.primary.url=jdbc:mysql://localhost:3306/primary
    spring.datasource.primary.username=root
    spring.datasource.primary.password=123456
    
    # 数据源2
    spring.datasource.secondary.url=jdbc:mysql://localhost:3306/secondary
    spring.datasource.secondary.username=root
    spring.datasource.secondary.password=123456

### 4.2 定义Datasource和EntityManager ###

@Configuration
    public class DataSourceConfig {
        @Bean(name = "primaryDataSource")
        @Qualifier("primaryDataSource")
        @ConfigurationProperties(prefix="spring.datasource.primary")
        public DataSource primaryDataSource() {
            return DataSourceBuilder.create().build();
        }
    
        @Bean(name = "secondaryDataSource")
        @Qualifier("secondaryDataSource")
        @ConfigurationProperties(prefix="spring.datasource.secondary")
        public DataSource secondaryDataSource() {
            return DataSourceBuilder.create().build();
        }
    
        @Bean(name = "primaryEntityManager")
        public EntityManager primaryEntityManager(EntityManagerFactoryBuilder builder, @Qualifier("primaryDataSource") DataSource dataSource) {
            return primaryEntityManagerFactory(builder, dataSource).getObject().createEntityManager();
        }
    
        @Bean(name = "primaryEntityManagerFactory")
        public LocalContainerEntityManagerFactoryBean primaryEntityManagerFactory(EntityManagerFactoryBuilder builder, @Qualifier("primaryDataSource") DataSource dataSource) {
            return builder.dataSource(dataSource).packages("com.example.primary.entity").persistenceUnit("primaryPersistenceUnit").build();
        }
    
        @Bean(name = "secondaryEntityManager")
        public EntityManager secondaryEntityManager(EntityManagerFactoryBuilder builder, @Qualifier("secondaryDataSource") DataSource dataSource) {
            return secondaryEntityManagerFactory(builder, dataSource).getObject().createEntityManager();
        }
    
        @Bean(name = "secondaryEntityManagerFactory")
        public LocalContainerEntityManagerFactoryBean secondaryEntityManagerFactory(EntityManagerFactoryBuilder builder, @Qualifier("secondaryDataSource") DataSource dataSource) {
            return builder.dataSource(dataSource).packages("com.example.secondary.entity").persistenceUnit("secondaryPersistenceUnit").build();
        }
    }

### 4.3 在需要使用数据源的地方注入不同的EntityManager ###

@Repository
    public class PrimaryRepository {
        @PersistenceContext(unitName = "primaryPersistenceUnit")
        private EntityManager entityManager;
        
        // ...
    }
    
    @Repository

总之，怎么方便怎么习惯怎么来，用什么都可以。

[https_baomidou.com_pages_a61e1b_dynamic-datasource]: https://baomidou.com/pages/a61e1b/#dynamic-datasource
[90cd6f87a7884b48a8124b56c4c8752b.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/18/f0394bbd68f74e0a973018f83abf2d54.png