RocketMQ源碼分析之RocketMQ事務消息實現原下篇(事務提交或回滾)

摘要： 事務消息提交或回滾的實現原理就是根據commitlogOffset找到消息，若是是提交動做，就恢復原消息的主題與隊列，再次存入commitlog文件進而轉到消息消費隊列，供消費者消費，而後將原預處理消息存入一個新的主題RMQ_SYS_TRANS_OP_HALF_TOPIC，表明該消息已被處理；回滾消息與提交事務消息不一樣的是，提交事務消息會將消息恢復原主題與隊列，再次存儲在commitlog文件中。

若您對RocketMQ技術感興趣，請加入 RocketMQ技術羣交流

本文將重點分析RocketMQ Broker如何處理事務消息提交、回滾命令，根據前面的介紹，其入口EndTransactionProcessor#processRequest：

OperationResult result = new OperationResult();
if (MessageSysFlag.TRANSACTION_COMMIT_TYPE == requestHeader.getCommitOrRollback()) {        // @1
result = this.brokerController.getTransactionalMessageService().commitMessage(requestHeader);    // @2
      if (result.getResponseCode() == ResponseCode.SUCCESS) {  // @3
            RemotingCommand res = checkPrepareMessage(result.getPrepareMessage(), requestHeader);    // @4
          if (res.getCode() == ResponseCode.SUCCESS) {
                MessageExtBrokerInner msgInner = endMessageTransaction(result.getPrepareMessage());     // @5
                msgInner.setSysFlag(MessageSysFlag.resetTransactionValue(msgInner.getSysFlag(), requestHeader.getCommitOrRollback()));
                msgInner.setQueueOffset(requestHeader.getTranStateTableOffset());
                msgInner.setPreparedTransactionOffset(requestHeader.getCommitLogOffset());
                msgInner.setStoreTimestamp(result.getPrepareMessage().getStoreTimestamp());    // @6
                RemotingCommand sendResult = sendFinalMessage(msgInner);                              // @7
                if (sendResult.getCode() == ResponseCode.SUCCESS) {             
                    this.brokerController.getTransactionalMessageService().deletePrepareMessage(result.getPrepareMessage());    // @8
                }
                return sendResult;
           }
          return res;
     }
}

代碼@1：若是請求爲提交事務，進入事務消息提交處理流程。
代碼@2：提交消息，別被這名字誤導了，該方法主要是根據commitLogOffset從commitlog文件中查找消息返回OperationResult實例：

• private MessageExt prepareMessage ：消息對象。
• private int responseCode：查找結果。
• private String responseRemark ：錯誤提示。

代碼@3：若是成功查找到消息，則繼續處理，不然返回給客戶端，消息未找到錯誤信息。

代碼@4：驗證消息必要字段。
驗證消息的生產組與請求信息中的生產者組是否一致。
驗證消息的隊列偏移量（queueOffset）與請求信息中的偏移量是否一致。
驗證消息的commitLogOffset與請求信息中的CommitLogOffset是否一致。

代碼@5:調用endMessageTransaction方法，該方法主要的目的就是恢復事務消息的真實的主題、隊列，並設置事務ID。

代碼@6：設置消息的相關屬性，這一步應該直接在endMessageTransaction中實現就好，統一恢復原消息的數量，特別關注的是取消了事務相關的系統標記。

代碼@7：發送最終消息，其實現原理很是簡單，調用MessageStore將消息存儲在commitlog文件中，此時的消息，會被轉發到原消息主題對應的消費隊列，被消費者消費。

代碼@8：刪除預處理消息(prepare)，實際上是將消息存儲在主題爲：RMQ_SYS_TRANS_OP_HALF_TOPIC的主題中，表明這些消息已經被處理（提交或回滾）。

上述就是事務消息提交的流程，事務回滾相似，接下來大概分析一下事務消息回滾的流程。

EndTransactionProcessor#processRequest
 else if (MessageSysFlag.TRANSACTION_ROLLBACK_TYPE == requestHeader.getCommitOrRollback()) {
       result = this.brokerController.getTransactionalMessageService().rollbackMessage(requestHeader);    // @1
       if (result.getResponseCode() == ResponseCode.SUCCESS) {
            RemotingCommand res = checkPrepareMessage(result.getPrepareMessage(), requestHeader);
            if (res.getCode() == ResponseCode.SUCCESS) {
                this.brokerController.getTransactionalMessageService().deletePrepareMessage(result.getPrepareMessage());   // @2
            }
           return res;
       }
}

代碼@1：回滾消息，其實內部就是根據commitlogOffset查找消息。
代碼@2：將消息存儲在RMQ_SYS_TRANS_OP_HALF_TOPIC中，表明該消息已被處理，與提交事務消息不一樣的是，提交事務消息會將消息恢復原主題與隊列，再次存儲在commitlog文件中。

事務消息在Broker服務端的提交回滾流程就介紹到這了。其核心實現就是根據commitlogOffset找到消息，若是是提交動做，就恢復原消息的主題與隊列，再次存入commitlog文件進而轉到消息消費隊列，供消費者消費，而後將原預處理消息存入一個新的主題RMQ_SYS_TRANS_OP_HALF_TOPIC，表明該消息已被處理；回滾消息與提交事務消息不一樣的是，提交事務消息會將消息恢復原主題與隊列，再次存儲在commitlog文件中。

若您對RocketMQ技術感興趣，請加入 RocketMQ技術羣交流