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/db1username: user1password: pass1driver-class-name: com.mysql.cj.jdbc.Driverdb2:url: jdbc:mysql://localhost:3306/db2username: user2password: pass2driver-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 {@Overrideprotected 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;
}