七、seata客户端二阶段分支事务的提交和回滚

所有文章

https://www.cnblogs.com/lay2017/p/12485081.html

正文

在阅读seata自动配置相关的内容的时候，我们说过。客户端会初始化一个RMClient的RPC客户端，且同时会添加一个监听器RmMessageListener，监听器将监听来自seata的server发送的RPC消息。我们再回顾一下这段代码

public static void init(String applicationId, String transactionServiceGroup) {
    RmRpcClient rmRpcClient = RmRpcClient.getInstance(applicationId, transactionServiceGroup);
    rmRpcClient.setResourceManager(DefaultResourceManager.get());
    rmRpcClient.setClientMessageListener(new RmMessageListener(DefaultRMHandler.get()));
    rmRpcClient.init();
}

RmMessageListener监听到的消息将交付给TransactionMessageHandler处理

这里的DefaultRMHandler.get()将返回一个DefaultRMHandler的单例对象，DefaultRMHandler简介实现TransactionMessageHandler，并且组合了三种handler

1）RMHandlerAT

2) RMHandlerTCC

3) RMHandlerSaga

分别对应不同的事务模式，三种处理器和DefaultRMHandler一样都继承了AbstractRMHandler。而AbstractRMHandler包含了两个核心方法，doBranchCommit和doBranchRollback，分别用于二阶段分支事务的提交和分支事务的回滚。

分支事务提交doBranchCommit

跟进AbstractRMHandler的doBranchCommit方法

protected void doBranchCommit(BranchCommitRequest request, BranchCommitResponse response) throws TransactionException {
    String xid = request.getXid();
    long branchId = request.getBranchId();
    String resourceId = request.getResourceId();
    String applicationData = request.getApplicationData();
    // 选择对应的ResourceManager，调用commit
    BranchStatus status = getResourceManager().branchCommit(request.getBranchType(), xid, branchId, resourceId, applicationData);
    response.setXid(xid);
    response.setBranchId(branchId);
    response.setBranchStatus(status);
}

doBranchCommit方法将会获取到对应的ResourceManager，我们以AT模式为例。AT模式将会获取到DataSourceManager这个ResourceManager。

我们跟进DataSourceManager的branchCommit方法

private ResourceManagerInbound asyncWorker;
@Override
public BranchStatus branchCommit(BranchType branchType, String xid, long branchId, String resourceId, String applicationData) throws TransactionException {
    return asyncWorker.branchCommit(branchType, xid, branchId, resourceId, applicationData);
}

branchCommit方法中交付给了一个异步线程处理，我们跟进AsyncWorker的branchCommit

@Override
public BranchStatus branchCommit(BranchType branchType, String xid, long branchId, String resourceId,
                                 String applicationData) throws TransactionException {
    if (!ASYNC_COMMIT_BUFFER.offer(new Phase2Context(branchType, xid, branchId, resourceId, applicationData))) {
        LOGGER.warn("Async commit buffer is FULL. Rejected branch [" + branchId + "/" + xid
            + "] will be handled by housekeeping later.");
    }
    return BranchStatus.PhaseTwo_Committed;
}

异步线程中使用了一个BlockQueue来排队处理，将会有一个Scheduler定时从BlockQueue中获取poll出来，然后进行undoLog的批量删除

UndoLogManagerFactory.getUndoLogManager(dataSourceProxy.getDbType()).batchDeleteUndoLog(xids, branchIds, conn);

小结：所谓分支事务的二阶段提交，其实就是异步删除undoLog。因为一阶段的时候已经提交了本地事务，所以二阶段就非常地快速。

分支事务回滚doBranchRollback

和doBranchCommit，先跟进AbstractRMHandler的doBranchRollback方法

protected void doBranchRollback(BranchRollbackRequest request, BranchRollbackResponse response) throws TransactionException {
    String xid = request.getXid();
    long branchId = request.getBranchId();
    String resourceId = request.getResourceId();
    String applicationData = request.getApplicationData();
    // 选择ResourceManager，调用rollback
    BranchStatus status = getResourceManager().branchRollback(request.getBranchType(), xid, branchId, resourceId, applicationData);
    response.setXid(xid);
    response.setBranchId(branchId);
    response.setBranchStatus(status);
}

可以看到，一样是调用ResourceManager的方法。我们同样以DataSourceManager为例，跟进DataSourceManager的branchRollback方法

@Override
public BranchStatus branchRollback(BranchType branchType, String xid, long branchId, String resourceId, String applicationData) throws TransactionException {
    // 选择Resource
    DataSourceProxy dataSourceProxy = get(resourceId);

    try {
        // undo补偿
        UndoLogManagerFactory.getUndoLogManager(dataSourceProxy.getDbType()).undo(dataSourceProxy, xid, branchId);
    } catch (TransactionException te) {
        if (te.getCode() == TransactionExceptionCode.BranchRollbackFailed_Unretriable) {
            return BranchStatus.PhaseTwo_RollbackFailed_Unretryable;
        } else {
            return BranchStatus.PhaseTwo_RollbackFailed_Retryable;
        }
    }
    return BranchStatus.PhaseTwo_Rollbacked;
}

在阅读DataSourceProxy的时候说过，DataSourceProxy将会作为一个Resource注册到ResourceManager当中。

而在分支事务回滚的时候，将会获取到该Resource，也就是DataSourceProxy。并且执行对应数据源的undo补偿操作。

我们跟进undo方法，看看补偿操作，方法较长，这里缩减掉一些内容

@Override
public void undo(DataSourceProxy dataSourceProxy, String xid, long branchId) throws TransactionException {
    // ...
    for (; ; ) {
        try {
            //...

            // 查找undoLog
            selectPST = conn.prepareStatement(SELECT_UNDO_LOG_SQL);
            selectPST.setLong(1, branchId);
            selectPST.setString(2, xid);
            rs = selectPST.executeQuery();

            boolean exists = false;
            while (rs.next()) {
                   // ...
                   // 反编码
                UndoLogParser parser = serializer == null ? UndoLogParserFactory.getInstance() : UndoLogParserFactory.getInstance(serializer);
                BranchUndoLog branchUndoLog = parser.decode(rollbackInfo);

                try {
                    List<SQLUndoLog> sqlUndoLogs = branchUndoLog.getSqlUndoLogs();
                    
                    for (SQLUndoLog sqlUndoLog : sqlUndoLogs) {
                        //...
                        // 执行器处理undo
                        AbstractUndoExecutor undoExecutor = UndoExecutorFactory.getUndoExecutor(dataSourceProxy.getDbType(), sqlUndoLog);
                        undoExecutor.executeOn(conn);
                    }
                }
            }

            if (exists) {
                // 删除undoLog
                deleteUndoLog(xid, branchId, conn);
                conn.commit();
            } else {
                //...
            }

            return;
        }
        // ...
    }
}

我们看到，首先是查询出分支事务的undoLog。然后反序列化出undoLog数据对象，且丢给执行器去执行。执行完毕删除undoLog，且提交事务。

由此可见，rollback就是把StatementProxy准备地undoLog拿出来，然后进行反向地补偿操作。

总结

本文我们简单地阅读了一下seata客户端在二阶段分支事务的commit和rollback操作做了啥。commit主要就是把undoLog删除，rollback则是获取了undoLog然后对数据进行反向生成。

到这里，seata的客户端代码阅读部分就结束了。我们从自动配置 -> 切面 -> 数据源代理 -> 监听器这么一个流程阅读下来可以发现AT模式的核心要点就是在于数据源代理，由undoLog做反向补偿操作。

后续，将开始server端的代码阅读...