Java 中注解怎么实现动态数据源的切换控制

陌伟君_5046

陌伟君_5046

2026-08-27

580人浏览

原创

java注解本身不实现动态数据源切换,而是作为声明式标记,配合aop与abstractroutingdatasource协作完成路由:定义@targetdatasource注解标记方法、通过aop将值存入threadlocal、routingdatasource读取该值决定实际数据源。

java 中注解怎么实现动态数据源的切换控制

Java 中注解本身不直接“实现”动态数据源切换,而是作为**声明式标记**,配合 AOP(如 Spring AOP)或代理机制,在方法/类执行前解析注解、决定使用哪个数据源,并通过 AbstractRoutingDataSource 完成实际路由。核心在于“注解 + 切面 + 路由数据源”三者协作。

定义自定义注解标记数据源

创建一个运行时保留的注解,用于标注需要切换数据源的方法或类:

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface TargetDataSource {
    String value() default "master"; // 默认主库
}

例如在 Service 方法上使用:

@Service
public class UserService {
    @TargetDataSource("slave1")
    public List<user> findAllUsers() { ... }
    
    @TargetDataSource("master")
    public void updateUser(User user) { ... }
}
</user>

基于 AbstractRoutingDataSource 实现动态路由

Spring 提供的 AbstractRoutingDataSource 是关键组件,它通过重写 determineCurrentLookupKey() 方法返回当前数据源标识(如 "master"),容器据此从配置的 targetDataSources 中查找对应数据源:

javascript-pro
javascript-pro

专注现代 ECMAScript、异步编程、性能优化和全栈的 JavaScript 专家,适用于现代开发

下载
@Configuration
public class DataSourceConfig {

    @Bean
    @Primary
    public DataSource routingDataSource(
            @Qualifier("masterDataSource") DataSource master,
            @Qualifier("slave1DataSource") DataSource slave1,
            @Qualifier("slave2DataSource") DataSource slave2) {
        
        AbstractRoutingDataSource routing = new AbstractRoutingDataSource();
        Map<object object> targets = new HashMap();
        targets.put("master", master);
        targets.put("slave1", slave1);
        targets.put("slave2", slave2);
        routing.setTargetDataSources(targets);
        routing.setDefaultTargetDataSource(master); // 默认走 master
        return routing;
    }
}</object>

注意:必须确保该 Bean 是事务管理器(DataSourceTransactionManager)所管理的唯一数据源。

用 AOP 解析注解并设置路由键

通过环绕通知(@Around)拦截带 @TargetDataSource 的方法,提取注解值,并将其存入线程上下文(如 ThreadLocal),供 AbstractRoutingDataSource 读取:

@Aspect
@Component
public class DataSourceAspect {

    private final static ThreadLocal<string> dataSourceKey = new ThreadLocal();

    @Around("@annotation(targetDataSource)")
    public Object routeDataSource(ProceedingJoinPoint pjp, TargetDataSource targetDataSource) throws Throwable {
        String key = targetDataSource.value();
        dataSourceKey.set(key);
        try {
            return pjp.proceed();
        } finally {
            dataSourceKey.remove(); // 防止内存泄漏和脏数据
        }
    }

    public static String getDataSourceKey() {
        return dataSourceKey.get();
    }
}</string>

然后在 AbstractRoutingDataSource 子类中重写方法:

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

注意事项与常见问题

  • 事务方法需谨慎:若方法加了 @Transactional,且事务管理器已绑定某个数据源,则注解切换可能失效;建议将 @Transactional 和 @TargetDataSource 分离到不同方法,或确保事务传播行为兼容(如 REQUIRES_NEW)
  • 切面优先级:若同时存在多个切面(如日志、权限),需用 @Order 明确 DataSourceAspect 在事务切面之前执行
  • 多数据源配置:所有真实数据源(master/slave)需独立配置(如 HikariCP),不能共用同一连接池
  • 读写分离场景:可结合方法名前缀(如 select* 自动走 slave)+ 注解覆盖,提升灵活性

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

9457

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

6602

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

5872

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1024

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

868

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1236

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2489

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

19831

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1115

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习