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

前言

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

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

1

畫(huà)圖猜測(cè)

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

因此可以得出猜測(cè)：

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

2

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

開(kāi)啟事務(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 的時(shí)候，會(huì)對(duì) UserService 基于 AOP 生成代理對(duì)象；

AbstractAutowireCapableBeanFactory#initializeBean 

... 

wrapIfNecessary 

AbstractAutoProxyCreator#createProxy CglibAopProxy#getProxy 生成代理對(duì)象

是不是看到 doBegin 這個(gè)詞突然感覺(jué)很熟悉。

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

因?yàn)槭褂玫?druid 連接池，所以這塊 Connection 是 durid 的連接池。

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

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

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

這一句才是重點(diǎn) SET autocommit=0。

SET autocommit=0

開(kāi)啟事務(wù)了！

總結(jié)一下流程：

執(zhí)行 SQL

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

• 因?yàn)檫@里使用的是 Mybatis，所以還是會(huì)被代理，MapperProxy#invoke；

• DruidPooledPreparedStatement#execute；

• ClientPreparedStatement#execute；

執(zhí)行過(guò)程相對(duì)比較簡(jiǎn)單：

提交事務(wù)

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

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

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

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

• DruidPooledConnection commit

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

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

commit

異常回滾

異常在這里 TransactionAspectSupport#invokeWithinTransaction 會(huì)被 catch。

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

執(zhí)行 DataSourceTransactionManager#doRollback

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

從而執(zhí)行 rollback 語(yǔ)句

rollback

恢復(fù) autocommit

cleanupTransactionInfo(txInfo);

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

3

Java 原生開(kāi)啟事務(wù)

如果覺(jué)得這樣有點(diǎn)繞，那咱們可以看簡(jiǎn)單版本的，不帶 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)閉自動(dòng)提交
            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 {
            // 開(kāi)啟自動(dòng)提交
            connection.setAutoCommit(true);
        }
    }
}

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

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

4

總結(jié)

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

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

不過(guò)還是有一個(gè)疑問(wèn)？

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

- <End /> -

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