MyBatis如何手动指定或切换数据库连接？

在MyBatis中指定数据库是一个常见的需求,尤其是在多数据源或读写分离的场景下，MyBatis作为优秀的持久层框架，提供了多种灵活的方式来指定数据库连接，确保数据操作能够精准地指向目标数据源，本文将详细介绍几种主流的实现方式，帮助开发者根据实际项目需求选择最合适的方案。

通过Configuration配置文件指定

MyBatis的核心配置文件mybatis-config.xml是全局配置的入口，可以通过<environments>标签配置多个环境，每个环境对应一个数据库连接，在开发中，可以通过指定不同的环境ID来切换数据库。

<environments default="development">
    <environment id="development">
        <transactionManager type="JDBC"/>
        <dataSource type="POOLED">
            <property name="driver" value="com.mysql.jdbc.Driver"/>
            <property name="url" value="jdbc:mysql://localhost:3306/dev_db"/>
            <property name="username" value="root"/>
            <property name="password" value="password"/>
        </dataSource>
    </environment>
    <environment id="production">
        <dataSource type="POOLED">
            <property name="url" value="jdbc:mysql://prod-host:3306/prod_db"/>
            <property name="username" value="prod_user"/>
            <property name="password" value="prod_pwd"/>
        </dataSource>
    </environment>
</environments>

在代码中通过SqlSessionFactory指定环境ID即可切换数据库：

SqlSessionFactory factory = new SqlSessionFactoryBuilder().build(reader, "production");

使用Spring集成管理数据源

在Spring与MyBatis集成的项目中,通常通过@Qualifier注解或@Primary来指定数据源，定义多个数据源Bean：

@Configuration
public class DataSourceConfig {
    @Bean("db1")
    public DataSource dataSource1() {
        return DataSourceBuilder.create().url("jdbc:mysql://db1:3306/test").build();
    }
    @Bean("db2")
    public DataSource dataSource2() {
        return DataSourceBuilder.create().url("jdbc:mysql://db2:3306/test").build();
    }
}

在Mapper接口中使用@DataSource注解（需自定义）或通过AOP动态切换数据源：

@Mapper
@DataSource("db1")
public interface UserMapper {
    List<User> selectUsers();
}

编程式动态切换数据源

对于需要动态切换数据库的场景（如分库分表），可以通过ThreadLocal存储数据源标识，并结合AOP或拦截器实现。

1. 定义数据源枚举：

   public enum DataSourceType {
    DB1, DB2
   }

2. 使用ThreadLocal存储当前数据源：

   public class DataSourceContextHolder {
    private static final ThreadLocal<DataSourceType> contextHolder = new ThreadLocal<>();
    public static void setDataSourceType(DataSourceType type) {
        contextHolder.set(type);
    }
    public static DataSourceType getDataSourceType() {
        return contextHolder.get();
    }
   }

3. 在拦截器中切换数据源：

   public class DataSourceInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        DataSourceType type = // 从方法参数或注解获取数据源类型
        DataSourceContextHolder.setDataSourceType(type);
        try {
            return invocation.proceed();
        } finally {
            DataSourceContextHolder.clear();
        }
    }
   }

基于注解的指定方式

通过自定义注解结合AOP,可以在业务方法上直接标注目标数据源。

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface TargetDataSource {
    String value();
}

在AOP切面中解析注解并切换数据源：

@Aspect
@Component
public class DataSourceAspect {
    @Before("@annotation(targetDataSource)")
    public void setDataSource(JoinPoint joinPoint, TargetDataSource targetDataSource) {
        DataSourceContextHolder.setDataSourceType(targetDataSource.value());
    }
}

多数据源场景下的路由实现

对于复杂的路由需求（如根据用户ID分库），可以实现AbstractRoutingDataSource（Spring提供）来自动路由数据源，步骤如下：

1. 继承AbstractRoutingDataSource并重写determineCurrentLookupKey方法：

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

2. 配置数据源路由：

   @Bean
   public DataSource routingDataSource(@Qualifier("db1") DataSource db1, 
                                   @Qualifier("db2") DataSource db2) {
    Map<Object, Object> targetDataSources = new HashMap<>();
    targetDataSources.put("DB1", db1);
    targetDataSources.put("DB2", db2);
    DynamicRoutingDataSource routingDataSource = new DynamicRoutingDataSource();
    routingDataSource.setTargetDataSources(targetDataSources);
    routingDataSource.setDefaultTargetDataSource(db1);
    return routingDataSource;
   }

FAQs

Q1: 如何在MyBatis中实现读写分离的数据源切换？
A: 可通过AbstractRoutingDataSource结合AOP实现，主库用于写操作，从库用于读操作，在Service层方法上添加@Read或@Write注解，AOP根据注解标识设置ThreadLocal中的数据源类型，最终路由到对应的数据源，同时需配置数据库中间件（如MyCat）或使用主从复制技术确保数据一致性。

Q2: 动态切换数据源时事务如何保证一致性？
A: 动态切换数据源可能导致事务失效，需确保所有操作在同一数据源上执行，解决方案包括：1）通过编程式事务（TransactionTemplate）控制事务范围；2）在AOP切面中统一管理事务，避免注解事务因数据源切换而失效；3）使用分布式事务框架（如Seata）处理跨数据源事务场景。