MyBatis Plus多数据源配置与应用

在实际应用中，有时候需要访问多个数据库，这时候就需要配置多数据源来实现对不同数据库的读写操作。MyBatis Plus提供了较为便捷的多数据源配置方式。本文将介绍如何在MyBatis Plus中配置多数据源以及在应用中使用多数据源进行数据库操作。

1. 添加依赖

首先，确保项目中引入了MyBatis Plus的依赖，可以在pom.xml中添加如下依赖：

<!-- MyBatis Plus依赖 -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.x.x</version>
</dependency>

2. 配置多数据源

在application.properties或application.yml中配置多数据源：

# 主数据源
spring.datasource.primary.url=jdbc:mysql://localhost:3306/primarydb
spring.datasource.primary.username=root
spring.datasource.primary.password=password
spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver

# 第二个数据源
spring.datasource.secondary.url=jdbc:mysql://localhost:3306/secondarydb
spring.datasource.secondary.username=root
spring.datasource.secondary.password=password
spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver

3. 创建数据源配置类

创建一个配置类来配置多数据源，以及使用@Primary注解标识主数据源：

@Configuration
@MapperScan(basePackages = "com.example.primary.mapper", sqlSessionTemplateRef = "primarySqlSessionTemplate")
public class PrimaryDataSourceConfig {

    @Primary
    @Bean(name = "primaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.primary")
    public DataSource dataSource() {
        return DataSourceBuilder.create().build();
    }

    @Primary
    @Bean(name = "primarySqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("primaryDataSource") DataSource dataSource,
                                               ApplicationContext applicationContext) throws Exception {
        MybatisSqlSessionFactoryBean sqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath*:mapper/primary/*.xml"));
        sqlSessionFactoryBean.setPlugins(new Interceptor[]{new PaginationInterceptor()});
        sqlSessionFactoryBean.setGlobalConfig(globalConfiguration());
        sqlSessionFactoryBean.setApplicationEventPublisher(applicationContext);
        return sqlSessionFactoryBean.getObject();
    }

    @Primary
    @Bean(name = "primaryTransactionManager")
    public DataSourceTransactionManager transactionManager(@Qualifier("primaryDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Primary
    @Bean(name = "primarySqlSessionTemplate")
    public SqlSessionTemplate sqlSessionTemplate(@Qualifier("primarySqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

    private GlobalConfig globalConfiguration() {
        GlobalConfig.DbConfig dbConfig = new GlobalConfig.DbConfig();
        dbConfig.setLogicDeleteValue("1");
        dbConfig.setLogicNotDeleteValue("0");

        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setDbConfig(dbConfig);
        return globalConfig;
    }
}

同样，创建第二个数据源的配置类：

@Configuration
@MapperScan(basePackages = "com.example.secondary.mapper", sqlSessionTemplateRef = "secondarySqlSessionTemplate")
public class SecondaryDataSourceConfig {

    @Bean(name = "secondaryDataSource")
    @ConfigurationProperties(prefix = "spring.datasource.secondary")
    public DataSource dataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "secondarySqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("secondaryDataSource") DataSource dataSource,
                                               ApplicationContext applicationContext) throws Exception {
        MybatisSqlSessionFactoryBean sqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dataSource);
        sqlSessionFactoryBean.setMapperLocations(new PathMatchingResourcePatternResolver()
                .getResources("classpath*:mapper/secondary/*.xml"));
        sqlSessionFactoryBean.setPlugins(new Interceptor[]{new PaginationInterceptor()});
        sqlSessionFactoryBean.setGlobalConfig(globalConfiguration());
        sqlSessionFactoryBean.setApplicationEventPublisher(applicationContext);
        return sqlSessionFactoryBean.getObject();
    }

    @Bean(name = "secondaryTransactionManager")
    public DataSourceTransactionManager transactionManager(@Qualifier("secondaryDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }

    @Bean(name = "secondarySqlSessionTemplate")
    public SqlSessionTemplate sqlSessionTemplate(@Qualifier("secondarySqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }

    private GlobalConfig globalConfiguration() {
        GlobalConfig.DbConfig dbConfig = new GlobalConfig.DbConfig();
        dbConfig.setLogicDeleteValue("1");
        dbConfig.setLogicNotDeleteValue("0");

        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setDbConfig(dbConfig);
        return globalConfig;
    }
}

在上述配置类中，分别配置了两个数据源，分别对应两个数据库。其中使用@MapperScan注解指定了Mapper接口的扫描路径，并且使用@Qualifier注解指定了使用哪个数据源。

4. 创建Mapper和Service

分别创建两个Mapper接口和Service类，分别对应两个数据源的操作。在Mapper接口中使用@Mapper注解指定具体的XML文件位置。

// PrimaryMapper.java
@Mapper
public interface PrimaryMapper {
    List<User> getAllUsers();
    // other methods
}

// SecondaryMapper.java
@Mapper
public interface SecondaryMapper {
    List<Order> getAllOrders();
    // other methods
}

// PrimaryService.java
public interface PrimaryService {
    List<User> getAllUsers();
    // other methods
}

// SecondaryService.java
public interface SecondaryService {
    List<Order> getAllOrders();
    // other methods
}

5. 使用多数据源

在需要使用多数据源的地方，使用@Qualifier注解指定要使用的数据源：

@Service
public class UserServiceImpl implements PrimaryService {

    @Autowired
    @Qualifier("primaryMapper")
    private PrimaryMapper primaryMapper;

    @Override
    public List<User> getAllUsers() {
        return primaryMapper.getAllUsers();
    }
    // other methods
}

@Service
public class OrderServiceImpl implements SecondaryService {

    @Autowired
    @Qualifier("secondaryMapper")
    private SecondaryMapper secondaryMapper;

    @Override
    public List<Order> getAllOrders() {
        return secondaryMapper.getAllOrders();
    }
    // other methods
}

在上述示例中，@Qualifier("primaryMapper")和@Qualifier("secondaryMapper")分别指定了使用主数据源和第二个数据源。

6. 总结

使用MyBatis Plus实现多数据源配置是一项常见的数据库操作任务，它使得应用能够方便地访问不同的数据库。在配置多数据源时，确保使用了正确的数据源配置、Mapper扫描、SQLSessionFactory的设置等。在使用多数据源时，通过@Qualifier注解指定具体使用的数据源，以确保正确的数据源被注入到相应的Service中。

综上所述，MyBatis Plus提供了灵活且便捷的方式来实现多数据源配置，使得应用在访问不同的数据库时更加方便。在实际项目中，根据具体需求和性能要求，选择合适的数据源配置方案，以保障系统的稳定性和性能。希望本文对你理解MyBatis Plus中多数据源的配置和应用提供了帮助。