动态数据源切换
在 Java 开发中,DynamicDataSourceContextHolder.push("slave2")
通常用于 动态数据源切换 场景,是实现多数据源(如主从数据库、分库分表)切换的核心操作之一。
具体作用
DynamicDataSourceContextHolder
是动态数据源框架(如自定义实现或基于 spring-jdbc
扩展的动态数据源组件)中的上下文持有者,其核心功能是 存储当前线程使用的数据源标识(如数据源名称)。
push("slave2")
方法的作用是:
将数据源标识 slave2
存入当前线程的上下文(通常基于 ThreadLocal
实现),告诉动态数据源框架:当前线程接下来的数据库操作,使用名为 slave2
的数据源。
工作原理
动态数据源的实现通常依赖以下组件:
DynamicDataSource
:继承自 Spring 的AbstractRoutingDataSource
,重写determineCurrentLookupKey()
方法,从DynamicDataSourceContextHolder
中获取当前线程的数据源标识,从而路由到对应的数据源。DynamicDataSourceContextHolder
:内部维护一个ThreadLocal<Deque<String>>
(线程本地的双端队列),用于存储数据源标识。push()
:将数据源标识入栈(添加到队列),设置当前线程的数据源。pop()
:将数据源标识出栈,恢复上一个数据源(适用于嵌套切换场景)。get()
:获取当前线程的数据源标识。
典型使用场景
在需要切换数据源的业务代码中,通常会结合 try-finally
确保数据源标识正确清理,避免线程复用导致的数据源混乱:
// 切换到 slave2 数据源
DynamicDataSourceContextHolder.push("slave2");
try {// 执行数据库操作(此时会使用 slave2 数据源)userMapper.selectByPrimaryKey(1);
} finally {// 清除数据源标识,恢复默认数据源DynamicDataSourceContextHolder.pop();
}
常见扩展
- 嵌套切换:如果业务中需要临时切换到另一个数据源(如
slave2
→slave3
),Deque
结构支持嵌套入栈 / 出栈,确保恢复时回到正确的上一级数据源。 - 注解驱动:很多框架会基于此实现注解式切换(如
@DataSource("slave2")
),通过 AOP 在方法执行前push
数据源,方法执行后pop
清理。