滚雪球学MyBatis-Plus(11)：多数据源配置

最新推荐文章于 2024-09-18 16:05:59 发布

bug菌¹

最新推荐文章于 2024-09-18 16:05:59 发布

阅读量769

点赞数 26

分类专栏： # 滚雪球学MyBatis-Plus 文章标签：零基础入门教学 mybatis-plus 多数据源

本文链接：https://blog.csdn.net/weixin_43970743/article/details/141028600

版权

滚雪球学MyBatis-Plus 专栏收录该内容

15 篇文章 3 订阅

订阅专栏

前言

在上期内容中，我们详细介绍了如何使用 MyBatis Plus 的代码生成器。通过代码生成器，我们能够根据数据库表结构自动生成实体类、Mapper 接口、服务类、控制器和 XML 映射文件，大大提高了开发效率，并减少了重复劳动。同时，我们还探讨了如何进行代码生成器的自定义配置，使其生成的代码更加符合项目规范。

本期内容将深入探讨 MyBatis Plus 的多数据源配置。多数据源配置在需要连接多个数据库的场景下非常实用，能够显著提高应用程序的灵活性和扩展性。通过本期内容，你将学会如何实现多数据源配置，并通过实例演示如何在项目中应用这些技术。

一、引入依赖

首先，我们需要在 pom.xml 文件中添加多数据源相关的依赖。

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

二、配置多数据源

接下来，我们需要在 application.yml 文件中配置多个数据源。

spring:
  datasource:
    dynamic:
      primary: master # 设置默认数据源
      datasource:
        master:
          url: jdbc:mysql://localhost:3306/mybatis_plus_master?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
          username: root
          password: your_password
          driver-class-name: com.mysql.cj.jdbc.Driver
        slave:
          url: jdbc:mysql://localhost:3306/mybatis_plus_slave?useUnicode=true&characterEncoding=UTF-8&serverTimezone=UTC
          username: root
          password: your_password
          driver-class-name: com.mysql.cj.jdbc.Driver

三、配置数据源管理

在项目中创建一个配置类，用于配置数据源的管理。

package com.example.mybatisplusdemo.config;

import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.provider.YmlDynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceAutoConfiguration;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class DataSourceConfig {

    @Bean
    public DynamicDataSourceProvider dynamicDataSourceProvider(DynamicDataSourceProperties properties) {
        return new YmlDynamicDataSourceProvider(properties);
    }

    @Bean
    public DynamicRoutingDataSource dynamicRoutingDataSource(DynamicDataSourceProvider dynamicDataSourceProvider) {
        DynamicRoutingDataSource dynamicRoutingDataSource = new DynamicRoutingDataSource();
        dynamicRoutingDataSource.setPrimary("master");
        dynamicRoutingDataSource.setProvider(dynamicDataSourceProvider);
        return dynamicRoutingDataSource;
    }
}

四、使用多数据源

在服务类或方法上使用 @DS 注解，指定使用的数据源。

在服务类中使用多数据源：

在 UserService 接口中定义方法：

IPage<User> findAllFromSlave(Page<User> page);

在 UserServiceImpl 类中实现方法，并使用 @DS 注解指定数据源：

@DS("slave")
@Override
public IPage<User> findAllFromSlave(Page<User> page) {
    return baseMapper.selectPage(page, null);
}

在控制器中提供多数据源查询接口：

在 UserController 类中编写接口，接收查询请求并调用服务层方法：

@GetMapping("/listFromSlave")
public IPage<User> listFromSlave(@RequestParam int pageNo, @RequestParam int pageSize) {
    Page<User> page = new Page<>(pageNo, pageSize);
    return userService.findAllFromSlave(page);
}

五、测试多数据源配置

启动 Spring Boot 应用：
- 在 IDE 中运行 MybatisPlusDemoApplication 类，启动项目。
测试多数据源查询：
- 使用浏览器或 Postman 访问 http://localhost:8080/user/listFromSlave?pageNo=1&pageSize=5，查看从从库查询的数据结果。

六、动态切换数据源

MyBatis Plus 还支持在运行时动态切换数据源。我们可以通过编程方式动态切换数据源，以满足更复杂的业务需求。

在服务层中动态切换数据源：

在 UserService 接口中定义方法：

IPage<User> findAllFromDynamicDataSource(Page<User> page, String datasource);

在 UserServiceImpl 类中实现方法，使用编程方式动态切换数据源：

@Override
public IPage<User> findAllFromDynamicDataSource(Page<User> page, String datasource) {
    DynamicRoutingDataSource dataSource = (DynamicRoutingDataSource) DynamicDataSourceContextHolder.peek();
    DynamicDataSourceContextHolder.push(datasource);
    IPage<User> result = baseMapper.selectPage(page, null);
    DynamicDataSourceContextHolder.poll();
    return result;
}

在控制器中提供动态切换数据源的接口：

在 UserController 类中编写接口，接收请求并调用服务层方法：

@GetMapping("/listFromDynamic")
public IPage<User> listFromDynamic(@RequestParam int pageNo, @RequestParam int pageSize, @RequestParam String datasource) {
    Page<User> page = new Page<>(pageNo, pageSize);
    return userService.findAllFromDynamicDataSource(page, datasource);
}