Seata?AT模式前后鏡像是如何生成詳解

更新時(shí)間：2022年11月29日 11:40:52 作者：夢(mèng)想實(shí)現(xiàn)家_Z

這篇文章主要為大家介紹了Seata?AT模式前后鏡像是如何生成的方法詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

在Seata 官網(wǎng)中，我們可以知道AT模式一階段的處理流程如下：

1.解析 SQL：得到 SQL 的類(lèi)型（UPDATE），表（product），條件（where name = 'TXC'）等相關(guān)的信息。

2.查詢(xún)前鏡像：根據(jù)解析得到的條件信息，生成查詢(xún)語(yǔ)句，定位數(shù)據(jù)。

3.執(zhí)行業(yè)務(wù) SQL。

4.查詢(xún)后鏡像：根據(jù)前鏡像的結(jié)果，通過(guò) 主鍵定位數(shù)據(jù)。

5.插入回滾日志：把前后鏡像數(shù)據(jù)以及業(yè)務(wù) SQL 相關(guān)的信息組成一條回滾日志記錄，插入到 UNDO_LOG 表中

......

前鏡像的作用是保證在分布式事務(wù)失敗時(shí)能夠成功回滾的重要依據(jù)，后鏡像是在回滾前校驗(yàn)是否臟寫(xiě)的數(shù)據(jù)依據(jù)，那么我們一階段的前后鏡像在真實(shí)的代碼實(shí)現(xiàn)中是如何生成的呢？

前后鏡像的生成

為了能夠探尋到前后鏡像的生成原理，我們需要看一看seata源碼。最終我們把入口定位到AbstractDMLBaseExecutor.executeAutoCommitFalse()方法：

    protected T executeAutoCommitFalse(Object[] args) throws Exception {
        // 獲取前鏡像
        TableRecords beforeImage = beforeImage();
        // 執(zhí)行業(yè)務(wù)SQL
        T result = statementCallback.execute(statementProxy.getTargetStatement(), args);
        // 獲取后鏡像
        TableRecords afterImage = afterImage(beforeImage);
        // 存儲(chǔ)前后鏡像
        prepareUndoLog(beforeImage, afterImage);
        return result;
    }

前鏡像

繼續(xù)深入探究一下到底是怎么獲取beforeImage的，根據(jù)源碼來(lái)看，我們發(fā)現(xiàn)beforeImage()方法有很多實(shí)現(xiàn)：

也就是說(shuō)，Seata會(huì)根據(jù)不同的業(yè)務(wù)SQL來(lái)生成beforeImage，有點(diǎn)經(jīng)驗(yàn)的小伙伴能夠看出，這里其實(shí)使用到了模版模式加上策略模式，我們挑一個(gè)DeleteExecutor來(lái)看一下：

@Override
    protected TableRecords beforeImage() throws SQLException {
        SQLDeleteRecognizer visitor = (SQLDeleteRecognizer) sqlRecognizer;
        // 根據(jù)表名解析出元數(shù)據(jù)
        TableMeta tmeta = getTableMeta(visitor.getTableName());
        ArrayList<List<Object>> paramAppenderList = new ArrayList<>();
        // 生成查詢(xún)beforeImage的SQL語(yǔ)句
        // SELECT [列名,] FROM [表名] (別名) (WHERE) (ORDER BY) (LIMIT) FOR UPDATE
        String selectSQL = buildBeforeImageSQL(visitor, tmeta, paramAppenderList);
        // 執(zhí)行SQL查詢(xún)beforeImage
        return buildTableRecords(tmeta, selectSQL, paramAppenderList);
    }

1.關(guān)鍵原理就是根據(jù)業(yè)務(wù)SQL反向查詢(xún)出被影響的數(shù)據(jù)；

2.為了保證查詢(xún)到的數(shù)據(jù)不是快照數(shù)據(jù)，一定要記得加上FOR UPDATE；

另外的話，我們發(fā)現(xiàn)其實(shí)UpdateExecutor的前鏡像生成方式和DeleteExecutor也差不多，像普通的insert這種SQL的前鏡像就更簡(jiǎn)單了：

    @Override
    protected TableRecords beforeImage() throws SQLException {
        return TableRecords.empty(getTableMeta());
    }

因?yàn)槠胀?code>insert語(yǔ)句不存在任何前鏡像，所以直接返回空記錄；

后鏡像

我們?cè)賮?lái)看一下后鏡像是如何生成的，這次我們看一下UpdateExecutor的后鏡像生成方法afterImage():

@Override
    protected TableRecords afterImage(TableRecords beforeImage) throws SQLException {
        // 獲取元數(shù)據(jù)
        TableMeta tmeta = getTableMeta();
        // 沒(méi)有前鏡像，也不存在后鏡像，說(shuō)明沒(méi)有數(shù)據(jù)被修改
        if (beforeImage == null || beforeImage.size() == 0) {
            return TableRecords.empty(getTableMeta());
        }
        // 生成查詢(xún)后鏡像SQL
        // SELECT [列名,] FROM [表名] (別名) WHERE 主鍵 in (前鏡像的主鍵值)
        // 這里面有一個(gè)配置項(xiàng)[client.undo.onlyCareUpdateColumns]，是否只關(guān)心被修改的列名，默認(rèn)是true
        String selectSQL = buildAfterImageSQL(tmeta, beforeImage);
        ResultSet rs = null;
        try (PreparedStatement pst = statementProxy.getConnection().prepareStatement(selectSQL)) {
            SqlGenerateUtils.setParamForPk(beforeImage.pkRows(), getTableMeta().getPrimaryKeyOnlyName(), pst);
            // 執(zhí)行查詢(xún)后鏡像
            rs = pst.executeQuery();
            // 包裝查詢(xún)結(jié)果
            return TableRecords.buildRecords(tmeta, rs);
        } finally {
            IOUtil.close(rs);
        }
    }

后鏡像的生成原理與前鏡像的生成原理差不多，不過(guò)還是有一些小小的區(qū)別的：

1.后鏡像的查詢(xún)條件使用的是前鏡像對(duì)應(yīng)的主鍵值，就沒(méi)有用業(yè)務(wù)SQL的查詢(xún)條件；不同的Executor處理方式不同，需要根據(jù)具體的業(yè)務(wù)SQL來(lái)區(qū)分；

2.查詢(xún)后鏡像的SQL沒(méi)有使用FOR UPDATE加鎖，直接拿的快照數(shù)據(jù)；