1. 程式人生 > >Rocket MQ 4.3.0分散式事務訊息初析

Rocket MQ 4.3.0分散式事務訊息初析

前言

從4.3.0版本開始支援事務訊息,這是一個令人振奮的訊息,線上目前4.2.0,在正式投產使用之前先進性簡單分析。幫助使用者實現類似 X/Open XA 的分佈事務功能,通過 MQ 事務訊息能達到分散式事務的最終一致。

基礎概念

  • 事務訊息:MQ 提供類似 X/Open XA 的分佈事務功能,通過 MQ 事務訊息能達到分散式事務的最終一致。
  • 半訊息:暫不能投遞的訊息,傳送方已經將訊息成功傳送到了 MQ 服務端,但是服務端未收到生產者對該訊息的二次確認,此時該訊息被標記成“暫不能投遞”狀態,處於該種狀態下的訊息即半訊息。
  • 訊息回查:由於網路閃斷、生產者應用重啟等原因,導致某條事務訊息的二次確認丟失,MQ 服務端通過掃描發現某條訊息長期處於“半訊息”時,需要主動向訊息生產者詢問該訊息的最終狀態(Commit 或是 Rollback),該過程即訊息回查。

實現原理

mq_trans

其中:

  1. 傳送方向 MQ 服務端傳送訊息。
  2. MQ Server 將訊息持久化成功之後,向傳送方 ACK 確認訊息已經發送成功,此時訊息為半訊息。
  3. 傳送方開始執行本地事務邏輯。
  4. 傳送方根據本地事務執行結果向 MQ Server 提交二次確認(Commit 或是 Rollback),MQ Server 收到 Commit 狀態則將半訊息標記為可投遞,訂閱方最終將收到該訊息;MQ Server 收到 Rollback 狀態則刪除半訊息,訂閱方將不會接受該訊息。
  5. 在斷網或者是應用重啟的特殊情況下,上述步驟4提交的二次確認最終未到達 MQ Server,經過固定時間後 MQ Server 將對該訊息發起訊息回查。
  6. 傳送方收到訊息回查後,需要檢查對應訊息的本地事務執行的最終結果。
  7. 傳送方根據檢查得到的本地事務的最終狀態再次提交二次確認,MQ Server 仍按照步驟4對半訊息進行操作。

事務訊息傳送對應步驟1、2、3、4,事務訊息回查對應步驟5、6、7。

code實戰

傳送事務訊息

  • 傳送半訊息及執行本地事務。
public class TransactionProducerClient {
 private final static Logger log = ClientLogger.getLog(); // 您需要設定自己的日誌,便於排查問題
 public static void main(String[] args) throws InterruptedException {
     final BusinessService businessService = new BusinessService(); // 本地業務 Service
     Properties properties = new Properties();
     // 您在控制檯建立的 Producer ID 注意:事務訊息的 Producer ID 不能與其他型別訊息的 Producer ID 共用
     properties.put(PropertyKeyConst.ProducerId, "");
     // 阿里雲身份驗證,在阿里雲伺服器管理控制檯建立
     properties.put(PropertyKeyConst.AccessKey, "");
     // 阿里雲身份驗證,在阿里雲伺服器管理控制檯建立
     properties.put(PropertyKeyConst.SecretKey, "");
     // 設定 TCP 接入域名(此處以公共雲生產環境為例)
     properties.put(PropertyKeyConst.ONSAddr,
       "http://onsaddr-internal.aliyun.com:8080/rocketmq/nsaddr4client-internal");
     TransactionProducer producer = ONSFactory.createTransactionProducer(properties,
             new LocalTransactionCheckerImpl());
     producer.start();
     Message msg = new Message("Topic", "TagA", "Hello MQ transaction===".getBytes());
     try {
             SendResult sendResult = producer.send(msg, new LocalTransactionExecuter() {
                 @Override
                 public TransactionStatus execute(Message msg, Object arg) {
                     // 訊息 ID(有可能訊息體一樣,但訊息 ID 不一樣,當前訊息 ID 在控制檯無法查詢)
                     String msgId = msg.getMsgID();
                     // 訊息體內容進行 crc32,也可以使用其它的如 MD5
                     long crc32Id = HashUtil.crc32Code(msg.getBody());
                     // 訊息 ID 和 crc32id 主要是用來防止訊息重複
                     // 如果業務本身是冪等的,可以忽略,否則需要利用 msgId 或 crc32Id 來做冪等
                     // 如果要求訊息絕對不重複,推薦做法是對訊息體 body 使用 crc32或 md5來防止重複訊息
                     Object businessServiceArgs = new Object();
                     TransactionStatus transactionStatus = TransactionStatus.Unknow;
                     try {
                         boolean isCommit =
                             businessService.execbusinessService(businessServiceArgs);
                         if (isCommit) {
                             // 本地事務成功則提交訊息
                             transactionStatus = TransactionStatus.CommitTransaction;
                         } else {
                             // 本地事務失敗則回滾訊息
                             transactionStatus = TransactionStatus.RollbackTransaction;
                         }
                     } catch (Exception e) {
                         log.error("Message Id:{}", msgId, e);
                     }
                     System.out.println(msg.getMsgID());
                     log.warn("Message Id:{}transactionStatus:{}", msgId, transactionStatus.name());
                     return transactionStatus;
                 }
             }, null);
         }
         catch (Exception e) {
             // 訊息傳送失敗,需要進行重試處理,可重新發送這條訊息或持久化這條資料進行補償處理
             System.out.println(new Date() + " Send mq message failed. Topic is:" + msg.getTopic());
             e.printStackTrace();
         }
     // demo example 防止程序退出(實際使用不需要這樣)
     TimeUnit.MILLISECONDS.sleep(Integer.MAX_VALUE);
 }
}
  • 提交事務訊息狀態

當本地事務執行完成(執行成功或執行失敗),需要通知伺服器當前訊息的事務狀態。 通知方式有以下兩種:

執行本地事務完成後提交
執行本地事務一直沒提交狀態,等待伺服器回查訊息的事務狀態


事務狀態有以下三種:

TransactionStatus.CommitTransaction 提交事務,允許訂閱方消費該訊息。
TransactionStatus.RollbackTransaction 回滾事務,訊息將被丟棄不允許消費。
TransactionStatus.Unknow 無法判斷狀態,期待 MQ Broker 向傳送方再次詢問該訊息對應的本地事務的狀態。

public class LocalTransactionCheckerImpl implements LocalTransactionChecker {
    private final static Logger log = ClientLogger.getLog();
    final  BusinessService businessService = new BusinessService();
    @Override
    public TransactionStatus check(Message msg) {
        //訊息 ID(有可能訊息體一樣,但訊息 ID 不一樣,當前訊息屬於 Half 訊息,所以訊息 ID 在控制檯無法查詢)
        String msgId = msg.getMsgID();
        //訊息體內容進行 crc32,也可以使用其它的方法如 MD5
        long crc32Id = HashUtil.crc32Code(msg.getBody());
        //訊息 ID、訊息本 crc32Id 主要是用來防止訊息重複
        //如果業務本身是冪等的,可以忽略,否則需要利用 msgId 或 crc32Id 來做冪等
        //如果要求訊息絕對不重複,推薦做法是對訊息體使用 crc32 或  md5 來防止重複訊息
        //業務自己的引數物件,這裡只是一個示例,需要您根據實際情況來處理
        Object businessServiceArgs = new Object();
        TransactionStatus transactionStatus = TransactionStatus.Unknow;
        try {
            boolean isCommit = businessService.checkbusinessService(businessServiceArgs);
            if (isCommit) {
                //本地事務已成功則提交訊息
                transactionStatus = TransactionStatus.CommitTransaction;
            } else {
                //本地事務已失敗則回滾訊息
                transactionStatus = TransactionStatus.RollbackTransaction;
            }
        } catch (Exception e) {
            log.error("Message Id:{}", msgId, e);
        }
        log.warn("Message Id:{}transactionStatus:{}", msgId, transactionStatus.name());
        return transactionStatus;
    }
 }
import java.util.zip.CRC32;
public class HashUtil {
    public static long crc32Code(byte[] bytes) {
        CRC32 crc32 = new CRC32();
        crc32.update(bytes);
        return crc32.getValue();
    }
}

事務回查機制說明

  • 傳送事務訊息為什麼必須要實現回查 Check 機制?

    當步驟(1)中 Half 訊息傳送完成,但本地事務返回狀態為 TransactionStatus.Unknow,或者應用退出導致本地事務未提交任何狀態時,從 MQ Broker 的角度看,這條 Half 狀態的訊息的狀態是未知的。 因此 MQ Broker 會定期要求傳送方能 Check 該 Half 狀態訊息,並上報其最終狀態。

  • Check 被回撥時,業務邏輯都需要做些什麼?

    MQ 事務訊息的 check 方法裡面,應該寫一些檢查事務一致性的邏輯。 MQ 傳送事務訊息時需要實現 LocalTransactionChecker 介面,用來處理 MQ Broker 主動發起的本地事務狀態回查請求;因此在事務訊息的 Check 方法中,需要完成兩件事情:

    (1) 檢查該 Half 訊息對應的本地事務的狀態(commited or rollback);

    (2) 向 MQ Broker 提交該 Half 訊息本地事務的狀態。