Spring @Transaction 注解是如何執(zhí)行事務(wù)的？

前言

相信小伙伴一定用過 @Transaction 注解，那 @Transaction 背后的秘密又知道多少呢？

Spring 是如何開啟事務(wù)的？又是如何進(jìn)行提交事務(wù)和關(guān)閉事務(wù)的呢？

1

畫圖猜測

在開始 debug 閱讀源碼之前，小伙伴們應(yīng)該已經(jīng)知道 MySQL 是如何開啟事務(wù)的。

因此可以得出猜測：

那下面跟著源碼一起讀一讀，Spring 的 @Transaction 注解是如何執(zhí)行事務(wù)邏輯的？

2

Spring 事務(wù)執(zhí)行流程

開啟事務(wù)

這里使用的是 Spring Boot + MySQL + Druid

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid-spring-boot-starter</artifactId>
    <version>1.2.6</version>
</dependency>

1. 在創(chuàng)建 Bean 的時候，會對 UserService 基于 AOP 生成代理對象；

AbstractAutowireCapableBeanFactory#initializeBean 

... 

wrapIfNecessary 

AbstractAutoProxyCreator#createProxy CglibAopProxy#getProxy 生成代理對象

是不是看到 doBegin 這個詞突然感覺很熟悉。

跟進(jìn) DataSourceTransactionManager#doBegin 方法，注意看，此時是在 spring-jdbc-5.3.8.jar 包下面的。

因為使用的 druid 連接池，所以這塊 Connection 是 durid 的連接池。

• DruidPooledConnection#setAutoCommit(false) 關(guān)閉自動提交；

這里就是 druid 的邏輯，一頓執(zhí)行然后到 com.alibaba.druid.filter.FilterChainImpl#connection_setAutoCommit。

• ConnectionImpl#setAutoCommit，這個是在 mysql-connector-java-8.0.25.jar 包下的。

這一句才是重點 SET autocommit=0。

SET autocommit=0

開啟事務(wù)了！

總結(jié)一下流程：

執(zhí)行 SQL

在開始事務(wù)之后，會通過回調(diào)執(zhí)行方法的內(nèi)部邏輯。

• 因為這里使用的是 Mybatis，所以還是會被代理，MapperProxy#invoke；

• DruidPooledPreparedStatement#execute；

• ClientPreparedStatement#execute；

執(zhí)行過程相對比較簡單：

提交事務(wù)

在 TransactionAspectSupport#invokeWithinTransaction 最后一行，commitTransactionAfterReturning(txInfo); 就是提交事務(wù)。

• AbstractPlatformTransactionManager#commit 抽象事務(wù)管理器，進(jìn)行提交事務(wù)

• DataSourceTransactionManager#doCommit 數(shù)據(jù)源數(shù)據(jù)管理器，提交事務(wù)

這里肯定是調(diào)用連接池的方法，所以會執(zhí)行到 DruidPooledConnection 中

• DruidPooledConnection commit

• 最終還是執(zhí)行到 mysql-connector-java-8.0.25.jar 包下面的 ConnectionImpl#commit

調(diào)用 commit 提交事務(wù)。

commit

異常回滾

異常在這里 TransactionAspectSupport#invokeWithinTransaction 會被 catch。

AbstractPlatformTransactionManager#rollback 在這里進(jìn)行 rollback

執(zhí)行 DataSourceTransactionManager#doRollback

最終執(zhí)行到  mysql-connector-java-8.0.25.jar 的 ConnectionImpl#rollback() 到 ConnectionImpl#rollbackNoChecks

從而執(zhí)行 rollback 語句

rollback

恢復(fù) autocommit

cleanupTransactionInfo(txInfo);

在 這個方法中會將之前設(shè)置的 autocommit 進(jìn)行恢復(fù)。

3

Java 原生開啟事務(wù)

如果覺得這樣有點繞，那咱們可以看簡單版本的，不帶 Spring。

/**
 * @author liuzhihang
 * @date 2021/6/18 16:51
 */
public class MainTest {


    public static void main(String[] args) throws Exception {

        DruidDataSource dataSource = new DruidDataSource();
        dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
        dataSource.setUrl("jdbc:mysql://localhost:3306/demo");
        dataSource.setUsername("root");
        dataSource.setPassword("root");

        Connection connection = dataSource.getConnection();

        try {
            // 關(guān)閉自動提交
            connection.setAutoCommit(false);

            connection.prepareStatement("update user_info set user_name = 'liuzhihang' where user_id = '1001';").executeUpdate();
            connection.prepareStatement("update user_address set address = 'anhui' where user_id = '1001';").executeUpdate();
            // 提交事務(wù)
            connection.commit();
        } catch (Exception e) {
            // 回滾
            connection.rollback();
        } finally {
            // 開啟自動提交
            connection.setAutoCommit(true);
        }
    }
}

看完 Java 原生提交事務(wù)的方式，是不是感覺簡單明了。

Spring @Transaction 只是創(chuàng)建了 AOP 代理，通過代理調(diào)用原生的開啟關(guān)閉事務(wù)，同樣在執(zhí)行 SQL 那一塊，也是 Mybatis 進(jìn)行了代理，從而提交 SQL。

4

總結(jié)

最后，將圖進(jìn)行合并，總結(jié)流程。

至此，事務(wù)執(zhí)行過程分析完畢。

不過還是有一個疑問？

為什么使用 set autocommit = 0 來開啟事務(wù)，而不是使用 begin 或者 start transaction 來開啟事務(wù)呢？

- <End /> -

• Spring 動態(tài)代理時是如何解決循環(huán)依賴的？為什么要使用三級緩存？
• Spring 是如何解決循環(huán)依賴的？
• Spring 自調(diào)用事務(wù)失效，你是怎么解決的？