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

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

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;br/>}
}
代碼@1:如果請求爲提交事務,進入事務消息提交處理流程。
代碼@2:提交消息,別被這名字誤導了,該方法主要是根據commitLogOffset從commitlog文件中查找消息返回OperationResult實例:
RocketMQ源碼分析之RocketMQ事務消息實現原下篇(事務提交或回滾)

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#proce***equest
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;br/>}
}
代碼@1:回滾消息,其實內部就是根據commitlogOffset查找消息。
代碼@2:將消息存儲在RMQ_SYS_TRANS_OP_HALF_TOPIC中,代表該消息已被處理,與提交事務消息不同的是,提交事務消息會將消息恢復原主題與隊列,再次存儲在commitlog文件中。

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章