动态数据源切换

最新推荐文章于 2025-08-11 21:09:51 发布

原创最新推荐文章于 2025-08-11 21:09:51 发布 · 612 阅读 ·

大模型引用 2 次

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

DSJ_1

关注

标签

#数据库 #postgresql

分类后端开发

在 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 清理。