MyBatis（26）MyBatis 有哪些方式可以实现多数据源管理

在企业级应用开发中，有时需要同时操作多个数据库，这就涉及到多数据源管理的问题。MyBatis作为一个流行的持久层框架，本身并没有直接提供多数据源管理的功能，但是可以通过与Spring等框架结合，或者通过自定义方式来实现多数据源管理。

1. 结合Spring框架实现多数据源管理

Spring框架提供了强大的数据源管理能力，通过Spring的配置可以很容易地实现MyBatis的多数据源管理。以下是一个使用Spring Boot和Spring Data来配置和使用多数据源的基本步骤：

步骤1：配置数据源

在application.properties或application.yml中配置多个数据源。例如：

spring:
  datasource:
    db1:
      url: jdbc:mysql://localhost:3306/db1
      username: user1
      password: pass1
      driver-class-name: com.mysql.cj.jdbc.Driver
    db2:
      url: jdbc:mysql://localhost:3306/db2
      username: user2
      password: pass2
      driver-class-name: com.mysql.cj.jdbc.Driver

步骤2：创建数据源配置类

为每个数据源创建一个配置类，并使用@Configuration和@MapperScan注解指定对应的Mapper接口应该使用哪个数据源。例如：

@Configuration
@MapperScan(basePackages = "com.example.mapper.db1", sqlSessionTemplateRef  = "db1SqlSessionTemplate")
public class DataSource1Config {

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

    @Bean(name = "db1SqlSessionFactory")
    public SqlSessionFactory sqlSessionFactory(@Qualifier("db1DataSource") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean bean = new SqlSessionFactoryBean();
        bean.setDataSource(dataSource);
        return bean.getObject();
    }

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

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

为第二个数据源创建类似的配置类，改变相应的Bean名和配置属性即可。

步骤3：使用指定数据源的Mapper

在业务代码中，你可以直接通过Spring注入对应数据源的Mapper接口，然后使用该接口操作对应的数据库。

2. 程序动态路由数据源

如果你的应用需要在运行时动态选择数据源（例如，基于用户的不同选择连接不同的数据库），你可以实现一个动态数据源路由。这通常涉及创建一个继承自AbstractRoutingDataSource的类，在该类中根据一定的逻辑（如ThreadLocal中保存的当前用户信息）来确定当前应该使用哪个数据源。

实现AbstractRoutingDataSource

public class DynamicDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceContextHolder.getDataSource();
    }
}

在DataSourceContextHolder类中，你可以使用ThreadLocal来保存和获取当前线程应该使用的数据源标识。

配置DynamicDataSource

在Spring配置中，你需要将DynamicDataSource设置为默认的数据源，并将之前配置的各个静态数据源作为目标数据源配置给DynamicDataSource。

@Bean
public DataSource dynamicDataSource() {
    DynamicDataSource dynamicDataSource = new DynamicDataSource();
    dynamicDataSource.setDefaultTargetDataSource(defaultDataSource); // 设定默认数据源
    Map<Object, Object> targetDataSources = new HashMap<>();
    targetDataSources.put("db1", db1DataSource());
    targetDataSources.put("db2", db2DataSource());
    dynamicDataSource.setTargetDataSources(targetDataSources);
    return dynamicDataSource;
}

总结

通过上述两种方法，我们可以在MyBatis应用中实现多数据源管理。第一种方法适用于应用启动时就已经确定了数据源的场景，例如，不同业务模块使用不同的数据库；第二种方法适合于需要根据运行时情况动态选择数据源的场景，例如，根据用户的不同选择连接不同的数据库。

实现多数据源管理是一个复杂且灵活的过程，需要根据具体的业务需求来定制。在实现过程中还需要考虑事务管理、数据源的健康检查和连接池管理等问题，以确保应用的稳定性和性能。