下面我将为你详细讲解 Spring Boot + Mybatis-Plus 实现多数据源的方法的完整攻略。

1. 概述

在实际开发中，可能会遇到需要同时连接多个数据源的情况。比如，我们需要从数据库A中获取数据，然后存储到数据库B中。或者，我们需要从两个不同的数据库中获取数据，进行一些关联查询和操作。这时候，就需要用到多数据源的技术。

在 Spring Boot 中，使用 Mybatis-Plus 可以很方便地实现多数据源的配置和使用。本文将通过两个示例演示具体的操作步骤。

2. 示例1：基于配置文件的多数据源配置

2.1 步骤

2.1.1 导入依赖

在 pom.xml 文件中添加 Mybatis-Plus 和数据库驱动依赖：

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>{mybatis-plus-version}</version>
</dependency>

<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>{mysql-version}</version>
</dependency>

其中， {mybatis-plus-version} 和 {mysql-version} 分别是 Mybatis-Plus 和 MySQL 的版本号。需要根据实际情况进行修改。

2.1.2 配置数据源

在 application.yml 文件中配置多个数据源信息：

spring:
  datasource:
    master:
      url: jdbc:mysql://localhost:3306/masterdb?characterEncoding=utf-8&useSSL=true&serverTimezone=Asia/Shanghai
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver
    slave1:
      url: jdbc:mysql://localhost:3306/slavedb1?characterEncoding=utf-8&useSSL=true&serverTimezone=Asia/Shanghai
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver
    slave2:
      url: jdbc:mysql://localhost:3306/slavedb2?characterEncoding=utf-8&useSSL=true&serverTimezone=Asia/Shanghai
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver

其中，master 是主数据源，slave1 和 slave2 是两个从数据源。

2.1.3 配置 Mybatis-Plus

在 MybatisPlusConfig.java 中配置 Mybatis-Plus 相关内容：

@Configuration
@MapperScan(basePackages = "com.example.demo.mapper")
public class MybatisPlusConfig {

    @Autowired
    private DataSource dataSource;

    @Bean(name = "masterSqlSessionFactory")
    public SqlSessionFactory masterSqlSessionFactory() throws Exception {
        MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        bean.setMapperLocations(resolver.getResources("classpath*:mapper/master/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "slave1SqlSessionFactory")
    public SqlSessionFactory slave1SqlSessionFactory() throws Exception {
        MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        bean.setMapperLocations(resolver.getResources("classpath*:mapper/slave1/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "slave2SqlSessionFactory")
    public SqlSessionFactory slave2SqlSessionFactory() throws Exception {
        MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        bean.setMapperLocations(resolver.getResources("classpath*:mapper/slave2/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "masterSqlSessionTemplate")
    public SqlSessionTemplate masterSqlSessionTemplate() throws Exception {
        SqlSessionTemplate template = new SqlSessionTemplate(masterSqlSessionFactory()); 
        return template; 
    }

    @Bean(name = "slave1SqlSessionTemplate")
    public SqlSessionTemplate slave1SqlSessionTemplate() throws Exception {
        SqlSessionTemplate template = new SqlSessionTemplate(slave1SqlSessionFactory()); 
        return template; 
    }

    @Bean(name = "slave2SqlSessionTemplate")
    public SqlSessionTemplate slave2SqlSessionTemplate() throws Exception {
        SqlSessionTemplate template = new SqlSessionTemplate(slave2SqlSessionFactory()); 
        return template; 
    }
}

在代码中，我们配置了三个 SqlSessionFactory 和三个 SqlSessionTemplate。其中，SqlSessionFactory 是 Mybatis 的核心对象，用于创建 SqlSession 实例。而 SqlSession 是 Mybatis 的会话对象，可以执行数据库操作。

2.1.4 测试

在测试代码中，通过 @Autowired 注入多个 SqlSessionTemplate 实例，然后使用 selectOne 方法查询数据。具体实现方法如下：

@Autowired
@Qualifier("masterSqlSessionTemplate")
private SqlSessionTemplate masterSqlSessionTemplate;

@Autowired
@Qualifier("slave1SqlSessionTemplate")
private SqlSessionTemplate slave1SqlSessionTemplate;

@Autowired
@Qualifier("slave2SqlSessionTemplate")
private SqlSessionTemplate slave2SqlSessionTemplate;

public void test() {
    User user1 = masterSqlSessionTemplate.selectOne("com.example.demo.mapper.master.UserMapper.selectById", 1L);
    User user2 = slave1SqlSessionTemplate.selectOne("com.example.demo.mapper.slave1.UserMapper.selectById", 1L);
    User user3 = slave2SqlSessionTemplate.selectOne("com.example.demo.mapper.slave2.UserMapper.selectById", 2L);
    System.out.println(user1);
    System.out.println(user2);
    System.out.println(user3);
}

2.2 示例2：基于注解的多数据源配置

2.2.1 导入依赖

同样，在 pom.xml 文件中添加 Mybatis-Plus 和数据库驱动依赖：

<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>{mybatis-plus-version}</version>
</dependency>

<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <version>{mysql-version}</version>
</dependency>

2.2.2 配置数据源

在 application.yml 文件中配置多个数据源信息：

spring:
  datasource:
    master:
      url: jdbc:mysql://localhost:3306/masterdb?characterEncoding=utf-8&useSSL=true&serverTimezone=Asia/Shanghai
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver
    slave:
      url: jdbc:mysql://localhost:3306/slavedb?characterEncoding=utf-8&useSSL=true&serverTimezone=Asia/Shanghai
      username: root
      password: root
      driver-class-name: com.mysql.cj.jdbc.Driver

2.2.3 配置 Mybatis-Plus

在 MybatisPlusConfig.java 中配置 Mybatis-Plus 相关内容：

@Configuration
@MapperScan(basePackages = "com.example.demo.mapper")
public class MybatisPlusConfig {

    @Autowired
    private DataSource dataSource;

    @Bean
    @Primary
    public SqlSessionFactory masterSqlSessionFactory() throws Exception {
        MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/master/*.xml"));
        return bean.getObject();
    }

    @Bean(name = "slaveSqlSessionFactory")
    public SqlSessionFactory slaveSqlSessionFactory() throws Exception {
        MybatisSqlSessionFactoryBean bean = new MybatisSqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/slave/*.xml"));
        return bean.getObject();
    }

    @Bean
    public DataSourceTransactionManager transactionManager() {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean
    public SqlSessionTemplate sqlSessionTemplate() throws Exception {
        return new SqlSessionTemplate(masterSqlSessionFactory()); 
    }

    @Bean(name = "slaveSqlSessionTemplate")
    public SqlSessionTemplate slaveSqlSessionTemplate() throws Exception {
        return new SqlSessionTemplate(slaveSqlSessionFactory()); 
    }
}

其中，使用了 @Primary 注解标注了主数据源的 SqlSessionFactory 实例。

2.2.4 配置数据源切换

在 DynamicDataSourceAspect.java 中配置数据源切换：

@Aspect
public class DynamicDataSourceAspect {

    @Pointcut("@annotation(com.example.demo.annotation.SlaveDataSource)")
    public void slaveDataSourcePointcut() {

    }

    @Before("slaveDataSourcePointcut()")
    public void switchToSlaveDataSource() {
        DynamicDataSourceContextHolder.setDataSourceKey("slave");
    }

    @After("slaveDataSourcePointcut()")
    public void restoreDataSource() {
        DynamicDataSourceContextHolder.clearDataSourceKey();
    }
}

在代码中，我们通过 AOP 技术，在方法执行前后切换数据源。

2.2.5 测试

在测试代码中，通过 @Autowired 注入 SqlSessionTemplate 实例，然后在方法上使用 @SlaveDataSource 注解，切换数据源，进行查询操作。具体实现方法如下：

@Autowired
private MybatisPlusConfig mybatisPlusConfig;

@Test
@SlaveDataSource
public void test() {
    User user = mybatisPlusConfig.sqlSessionTemplate().selectOne("com.example.demo.mapper.master.UserMapper.selectById", 1L);
    System.out.println(user);
}

通过在方法上使用 @SlaveDataSource 注解，我们可以很方便地切换到从数据源进行操作。

3. 总结

本文主要介绍了 Spring Boot + Mybatis-Plus 实现多数据源的方法，包括基于配置文件和基于注解的两种方法。对于实现多数据源的应用场景，我们应该根据实际情况选择适合的方式进行配置和使用。希望本文对你有所帮助。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：Spring Boot + Mybatis-Plus实现多数据源的方法 - Python技术站

Spring Boot + Mybatis-Plus实现多数据源的方法