解决 springboot 多数据源或动态数据源的事务问题

最新推荐文章于 2024-08-12 08:30:00 发布

低端程序狗

最新推荐文章于 2024-08-12 08:30:00 发布

阅读量4.3k

点赞数 4

分类专栏： spring

本文链接：https://blog.csdn.net/qq_36845919/article/details/117755862

版权

本文介绍了在SpringBoot项目中解决多数据源或动态数据源事务问题的方法。通过自定义事务注解、实现事务上下文、重写Connection的commit和close方法，以及利用AOP进行方法拦截，确保在使用多数据源时能够正确处理事务。

摘要由CSDN通过智能技术生成

需求背景：

动态数据源或者多数据源在项目当中是经常遇到的，但由于spring 开启事务后，为保证整个事务的 connection 不会变化，spring 在通过 DataSourceUtils 获取 connection 的时候会用 DataSource 作为 key 将 connection 保存到 ThreadLocal 中（这段代码是没办法进行重写的，它是静态方法，并在其他地方直接调用），如下所示：
在这里插入图片描述
spring 事务获取 connection 源代码：

public static Connection doGetConnection(DataSource dataSource) throws SQLException {
   
    Assert.notNull(dataSource, "No DataSource specified");
    // 从事务管理器中获取connection，如果有直接使用
    ConnectionHolder conHolder = (ConnectionHolder) TransactionSynchronizationManager.getResource(dataSource);
    if (conHolder != null && (conHolder.hasConnection() || conHolder.isSynchronizedWithTransaction())) {
   
        conHolder.requested();
        if (!conHolder.hasConnection()) {
   
            logger.debug("Fetching resumed JDBC Connection from DataSource");
            conHolder.setConnection(fetchConnection(dataSource));
        }
        return conHolder.getConnection();
    }
    // 事务期间第一次获取connection，如果开启事务，则将连接进行缓存
    Connection con = fetchConnection(dataSource);
    if (TransactionSynchronizationManager.isSynchronizationActive()) {
   
        logger.debug("Registering transaction synchronization for JDBC Connection");
        // Use same Connection for further JDBC actions within the transaction.
        // Thread-bound object will get removed by synchronization at transaction completion.
        ConnectionHolder holderToUse = conHolder;
        if (holderToUse == null) {
   
            holderToUse = new ConnectionHolder(con);
        }
        else {
   
            holderToUse.setConnection(con);
        }
        holderToUse.requested();
        TransactionSynchronizationManager.registerSynchronization(
            new ConnectionSynchronization(holderToUse, dataSource));
        holderToUse.setSynchronizedWithTransaction(true);
        if (holderToUse != conHolder) {
   
            TransactionSynchronizationManager.bindResource(dataSource, holderToUse);
        }
    }
    return con;
}