多数据源下@transcation事务踩坑
多数据源使用DynamicDatasource,再用@transcation注解,造成数据源会切换的BUG:
- 方法上写了
@DS("slave") - 执行 SQL 时却仍然走了
master
情景浮现:
kotlin
@DS("master")
public class DB1Service{
public list<?> queryDB1()
}
@DS("slaver")
public class DB2Service{
public list<?> queryDB2()
}
public class Main{
@Transactional(rollbackFor = Exception.class)
public void sync(){
// 预期会走slaver库,⚠️ 但实际还是会查询 master DB
var list2 = DB2Service.queryDB2()
// .....
var list1 = DB1Service.queryDB1()
}
}
1 🤔为什么?
先说结论 :
在同一个本地事务 里,第一次拿到的连接会绑定到当前线程。后续再切
@DS,通常只会改路由标识,不会替换已经绑定的连接。
调用 sync() 时,大致发生下面几步:
- 进入
@Transactional代理,事务开始 - 事务管理器通过
DataSourceUtils.getConnection(...)拿连接 - 连接绑定到
TransactionSynchronizationManager(ThreadLocal) - 执行业务代码,进入
db2Service.queryDB2() @DS("slave")把当前线程的数据源 key 改成slave- 但取连接时发现线程里已经有事务连接,于是直接复用
- SQL 继续走原连接(常见是
master)
这里最重要的一点: @DS 决定的是 "下一次新建连接时选哪个数据源",不是"把当前事务里的连接热切换"。
如果连接已经在事务里绑定好了,它不会换连接。
2 扩展:Dynamic-Datasource实现多数据源原理
-
启动阶段装配
• DynamicDataSourceAutoConfiguration 注册 DynamicRoutingDataSource 作为主 DataSource。
• 把你配置的 spring.datasource.dynamic.datasource.* 加载成数据源池,维护在内部 dataSourceMap/groupDataSources 里。
-
AOP 拦截 @DS
• DynamicDataSourceAopConfiguration 装配 DynamicDataSourceAnnotationAdvisor + Interceptor。
• 方法进入时,DynamicDataSourceAnnotationInterceptor 解析 @DS 值(DataSourceClassResolver,支持方法/类、接口、继承等)。
• 解析出的 key(如 master/doris)被 push 到
DynamicDataSourceContextHolder(ThreadLocal,栈结构,支持嵌套调用)。
• 方法退出后 poll,恢复上一个数据源上下文。
-
真正路由发生在"取连接" • JdbcTemplate/MyBatis 最终都会走到当前主数据源取连接。
• 主数据源是 DynamicRoutingDataSource,它在 determineDataSource() 里 peek 当前线程 key。
• 然后 getDataSource(key):
• key 为空 -> 用 primary(默认 master)
• key 是具体库名 -> 直连该库
• key 是组名 -> 按策略(默认负载均衡)选组内某个库
• 未命中时:strict=true 抛异常,strict=false 回退 primary
总结:
• @DS 只是改"路由上下文"
• DynamicRoutingDataSource 在取连接时按上下文选库
• 不是直接"切换已拿到的连接"