Java动态数据源案例怎么切换

wen java案例 2026-06-29 28

Java动态数据源切换实战：多数据源架构的最佳实践与案例解析

文章目录

为什么需要动态数据源切换？
核心实现原理：AbstractRoutingDataSource
实战案例：基于Spring Boot的动态数据源切换
- 1 环境准备与依赖配置
- 2 自定义数据源路由类
- 3 动态数据源配置与管理
- 4 注解驱动切换数据源
多数据源事务处理策略
常见问题与问答（FAQ）
架构选型与性能优化建议

为什么需要动态数据源切换？

在现代企业级应用中,单一数据源往往无法满足业务需求，常见场景包括：

Java动态数据源案例怎么切换

读写分离：主库负责写入，从库负责查询，提升系统吞吐量。
多租户系统：每个租户独立数据库，根据登录用户动态切换。
分库分表：不同业务模块对应不同数据库实例。
数据迁移/灰度发布：临时切换至新数据库进行验证。

静态配置多个数据源无法实现灵活的路由决策,而动态数据源切换允许在运行时根据业务规则（如请求参数、当前用户、方法注解等）选择目标数据源，从而提升系统扩展性与维护性。

核心实现原理：AbstractRoutingDataSource

Spring框架的内置类AbstractRoutingDataSource是实现动态数据源切换的基础，其核心机制是：

维护一个Map<Object, DataSource>类型的目标数据源映射。
每个数据库操作调用前,通过determineCurrentLookupKey()方法返回当前要使用的数据源key。
Spring会从目标数据源映射中取出对应的DataSource实例，并绑定到当前线程。

public abstract class AbstractRoutingDataSource extends AbstractDataSource {
    @Override
    public Connection getConnection() {
        return determineTargetDataSource().getConnection();
    }
    protected DataSource determineTargetDataSource() {
        Object lookupKey = determineCurrentLookupKey();
        // 从映射中获取数据源
    }
    protected abstract Object determineCurrentLookupKey();
}

关键点：determineCurrentLookupKey()必须由开发者实现，通常结合ThreadLocal存储当前线程的数据源标识。

实战案例：基于Spring Boot的动态数据源切换

1 环境准备与依赖配置

Spring Boot 2.x
MySQL + Maven
核心依赖：spring-boot-starter-jdbc、mysql-connector-java

配置application.yml中的多个数据源：

datasource:
  master:
    url: jdbc:mysql://localhost:3306/db_master
    username: root
    password: root
  slave:
    url: jdbc:mysql://localhost:3306/db_slave
    username: root
    password: root

2 自定义数据源路由类

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

3 动态数据源配置与管理

DataSourceContextHolder（ThreadLocal存储当前key）：

public class DataSourceContextHolder {
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
    public static void setDataSourceKey(String key) { contextHolder.set(key); }
    public static String getDataSourceKey() { return contextHolder.get(); }
    public static void clear() { contextHolder.remove(); }
}

多数据源配置类：

@Configuration
public class DataSourceConfig {
    @Bean(name = "masterDataSource")
    @ConfigurationProperties(prefix = "datasource.master")
    public DataSource masterDataSource() { return DataSourceBuilder.create().build(); }
    @Bean(name = "slaveDataSource")
    @ConfigurationProperties(prefix = "datasource.slave")
    public DataSource slaveDataSource() { return DataSourceBuilder.create().build(); }
    @Primary
    @Bean(name = "dynamicDataSource")
    public DataSource dynamicDataSource() {
        Map<Object, Object> targetDataSources = new HashMap<>();
        targetDataSources.put("master", masterDataSource());
        targetDataSources.put("slave", slaveDataSource());
        DynamicDataSource dataSource = new DynamicDataSource();
        dataSource.setDefaultTargetDataSource(masterDataSource()); // 默认主库
        dataSource.setTargetDataSources(targetDataSources);
        return dataSource;
    }
}

4 注解驱动切换数据源

定义注解：

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
    String value() default "master";
}

AOP拦截实现：

@Aspect
@Component
public class DataSourceAspect {
    @Around("@annotation(dataSource)")
    public Object around(ProceedingJoinPoint joinPoint, DataSource dataSource) throws Throwable {
        DataSourceContextHolder.setDataSourceKey(dataSource.value());
        try {
            return joinPoint.proceed();
        } finally {
            DataSourceContextHolder.clear(); // 避免内存泄漏
        }
    }
}

使用示例：

@Service
public class UserService {
    @DataSource("slave")
    public List<User> findAll() { // 读操作走从库
        return userMapper.selectAll();
    }
    @DataSource("master")
    public void insertUser(User user) { // 写操作走主库
        userMapper.insert(user);
    }
}

注意：动态数据源切换的核心是AOP代理，因此不要在同一类内部调用带@DataSource的方法，否则代理失效（解决方法：注入自身Bean或使用AopContext.currentProxy()）。

多数据源事务处理策略

动态数据源结合@Transactional时需注意：

事务与数据源绑定：@Transactional默认绑定主库，若在方法内切换数据源，事务无法感知。
方案：
- 使用@DataSource+@Transactional组合，但需确保数据源切换在事务开启前完成（AOP优先级控制）。
- 对于跨数据源事务（如分布式事务），需引入AT（ Seata）、TCC模式或基于MQ的最终一致性方案。

核心原则：一个事务内尽量只操作一个数据源，避免分布式事务复杂度。