全部產品
Search
文件中心

ApsaraMQ for RocketMQ:收發事務訊息

更新時間:Jul 01, 2024

雲訊息佇列 RocketMQ 版提供類似XA或Open XA的分散式交易功能,通過雲訊息佇列 RocketMQ 版事務訊息,能達到分散式交易的最終一致。本文提供使用HTTP協議下的Python SDK收發事務訊息的範例程式碼。

背景資訊

事務訊息的互動流程如下圖所示。

圖片1.png

更多資訊,請參見事務訊息

前提條件

您已完成以下操作:

  • 安裝Python SDK。更多資訊,請參見準備環境

  • 建立資源。代碼中涉及的資源資訊,例如執行個體、Topic和Group ID等,需要在控制台上提前建立。更多資訊,請參見建立資源

  • 擷取阿里雲存取金鑰AccessKey ID和AccessKey Secret。更多資訊,請參見建立AccessKey

發送事務訊息

發送事務訊息的範例程式碼如下。

#!/usr/bin/env python
# coding=utf8
import sys

from mq_http_sdk.mq_exception import MQExceptionBase
from mq_http_sdk.mq_producer import *
from mq_http_sdk.mq_client import *
import time
import threading
# 初始化client。
mq_client = MQClient(
    # 設定HTTP協議用戶端存取點,進入訊息佇列RocketMQ版控制台執行個體詳情頁面的存取點地區查看。
     "${HTTP_ENDPOINT}",
    # 請確保環境變數ALIBABA_CLOUD_ACCESS_KEY_ID、ALIBABA_CLOUD_ACCESS_KEY_SECRET已設定。
    # AccessKey ID,阿里雲身分識別驗證標識。
    os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID'],
    # AccessKey Secret,阿里雲身分識別驗證密鑰。
    os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET']
    )
# 訊息所屬的Topic,在訊息佇列RocketMQ版控制台建立。
topic_name = "${TOPIC}"
# 您在訊息佇列RocketMQ版控制台建立的Group ID。
group_id = "${GROUP_ID}"
# Topic所屬的執行個體ID,在訊息佇列RocketMQ版控制台建立。
# 若執行個體有命名空間,則執行個體ID必須傳入;若執行個體無命名空間,則執行個體ID傳入Null 字元串。執行個體的命名空間可以在訊息佇列RocketMQ版控制台的執行個體詳情頁面查看。
instance_id = "${INSTANCE_ID}"

# 迴圈發送4條事務訊息。
msg_count = 4
print("%sPublish Transaction Message To %s\nTopicName:%s\nMessageCount:%s\n" \
      % (10 * "=", 10 * "=", topic_name, msg_count))


def process_trans_error(exp):
    print("\nCommit/Roll Transaction Message Fail! Exception:%s" % exp)
    # 如果Commit或Rollback時超過了TransCheckImmunityTime(針對發送事務訊息的控制代碼)或者超過10s(針對consumeHalfMessage的控制代碼),則Commit或Rollback失敗。
    if exp.sub_errors:
        for sub_error in exp.sub_errors:
            print("\tErrorHandle:%s,ErrorCode:%s,ErrorMsg:%s" % \
                  (sub_error["ReceiptHandle"], sub_error["ErrorCode"], sub_error["ErrorMessage"]))


# 用戶端需要有一個線程或者進程來消費沒有確認的事務訊息。
# 啟動一個線程來檢查沒有確認的事務訊息。
class ConsumeHalfMessageThread(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
        self.count = 0
        # 重新New一個Client
        self.mq_client = MQClient(
                # 設定HTTP接入網域名稱,進入訊息佇列RocketMQ版控制台執行個體詳情頁面的HTTP存取點地區查看。
                "${HTTP_ENDPOINT}",
                # AccessKey ID阿里雲身分識別驗證,在阿里雲RAM控制台建立。
                "${ACCESS_KEY}",
                # AccessKey Secret阿里雲身分識別驗證,在阿里雲RAM控制台建立。
                "${SECRET_KEY}"
                )
        self.trans_producer = self.mq_client.get_trans_producer(instance_id, topic_name, group_id)

    def run(self):
        while 1:
            if self.count == 3:
                break;
            try:
                half_msgs = self.trans_producer.consume_half_message(1, 3)
                for half_msg in half_msgs:
                    print("Receive Half Message, MessageId: %s\nMessageBodyMD5: %s \
                                                  \nMessageTag: %s\nConsumedTimes: %s \
                                                  \nPublishTime: %s\nBody: %s \
                                                  \nNextConsumeTime: %s \
                                                  \nReceiptHandle: %s \
                                                  \nProperties: %s" % \
                          (half_msg.message_id, half_msg.message_body_md5,
                           half_msg.message_tag, half_msg.consumed_times,
                           half_msg.publish_time, half_msg.message_body,
                           half_msg.next_consume_time, half_msg.receipt_handle, half_msg.properties))

                a = int(half_msg.get_property("a"))
                try:
                    if a == 1:
                        # 確認提交事務訊息。
                        self.trans_producer.commit(half_msg.receipt_handle)
                        self.count += 1
                        print("------>commit")
                    elif a == 2 and half_msg.consumed_times > 1:
                        # 確認提交事務訊息。
                        self.trans_producer.commit(half_msg.receipt_handle)
                        self.count += 1
                        print("------>commit")
                    elif a == 3:
                        # 確認復原事務訊息。
                        self.trans_producer.rollback(half_msg.receipt_handle)
                        self.count += 1
                        print("------>rollback")
                    else:
                        # 什麼都不做,下次再檢查。
                        print("------>unknown")
                except MQExceptionBase as rec_commit_roll_e:
                    process_trans_error(rec_commit_roll_e)
            except MQExceptionBase as half_e:
                if half_e.type == "MessageNotExist":
                    print("No half message! RequestId: %s" % half_e.req_id)
                    continue

                print("Consume half message Fail! Exception:%s\n" % half_e)
                break


consume_half_thread = ConsumeHalfMessageThread()
consume_half_thread.setDaemon(True)
consume_half_thread.start()

try:
    trans_producer = mq_client.get_trans_producer(instance_id, topic_name, group_id)
    for i in range(msg_count):
        msg = TopicMessage(
            # 訊息內容。
            "I am test message %s." % i,
            # 訊息標籤。
            "tagA"
        )
        # 設定訊息的自訂屬性。
        msg.put_property("a", i)
        # 設定訊息的Key。
        msg.set_message_key("MessageKey")
        # 設定事務第一次回查的時間,為相對時間。單位:秒,範圍:10~300。
        # 第一次事務回查後如果訊息沒有Commit或者Rollback,則之後每隔10s左右會回查一次,共回查24小時。
        msg.set_trans_check_immunity_time(10)
        re_msg = trans_producer.publish_message(msg)
        print("Publish Transaction Message Succeed. MessageID:%s, BodyMD5:%s, Handle:%s" \
              % (re_msg.message_id, re_msg.message_body_md5, re_msg.receipt_handle))
        time.sleep(1)
        if i == 0:
            # 發送完事務訊息後能擷取到半訊息控制代碼,可以直接Commit或Rollback事務訊息。
            try:
                trans_producer.commit(re_msg.receipt_handle)
            except MQExceptionBase as pub_commit_roll_e:
                process_trans_error(pub_commit_roll_e)

except MQExceptionBase as pub_e:
    if pub_e.type == "TopicNotExist":
        print("Topic not exist, please create it.")
        sys.exit(1)
    print("Publish Message Fail. Exception:%s" % pub_e)

while 1:
    if not consume_half_thread.is_alive():
        break
    time.sleep(1)

訂閱事務訊息

訂閱事務訊息的範例程式碼如下。

from mq_http_sdk.mq_exception import MQExceptionBase
from mq_http_sdk.mq_consumer import *
from mq_http_sdk.mq_client import *

# 初始化client。
mq_client = MQClient(
    # 設定HTTP協議用戶端存取點,進入訊息佇列RocketMQ版控制台執行個體詳情頁面的存取點地區查看。
    "${HTTP_ENDPOINT}",
    # 請確保環境變數ALIBABA_CLOUD_ACCESS_KEY_ID、ALIBABA_CLOUD_ACCESS_KEY_SECRET已設定。
    # AccessKey ID,阿里雲身分識別驗證標識。
    os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID'],
    # AccessKey Secret,阿里雲身分識別驗證密鑰。
    os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET']
     )
# 訊息所屬的Topic,在訊息佇列RocketMQ版控制台建立。
topic_name = "${TOPIC}"
# 您在訊息佇列RocketMQ版控制台建立的Group ID。
group_id = "${GROUP_ID}"
# Topic所屬的執行個體ID,在訊息佇列RocketMQ版控制台建立。
# 若執行個體有命名空間,則執行個體ID必須傳入;若執行個體無命名空間,則執行個體ID傳入Null 字元串。執行個體的命名空間可以在訊息佇列RocketMQ版控制台的執行個體詳情頁面查看。
instance_id = "${INSTANCE_ID}"

consumer = mq_client.get_consumer(instance_id, topic_name, group_id)

# 長輪詢表示如果Topic沒有訊息,則用戶端請求會在服務端掛起3秒,3秒內如果有訊息可以消費則立即返迴響應。
# 長輪詢時間3秒(最多可設定為30秒)。
wait_seconds = 3
# 一次最多消費3條(最多可設定為16條)。
batch = 3
print(("%sConsume And Ak Message From Topic%s\nTopicName:%s\nMQConsumer:%s\nWaitSeconds:%s\n" \
        % (10 * "=", 10 * "=", topic_name, group_id, wait_seconds)))
while True:
    try:
        # 長輪詢消費訊息。
        recv_msgs = consumer.consume_message(batch, wait_seconds)
        for msg in recv_msgs:
            print(("Receive, MessageId: %s\nMessageBodyMD5: %s \
                              \nMessageTag: %s\nConsumedTimes: %s \
                              \nPublishTime: %s\nBody: %s \
                              \nNextConsumeTime: %s \
                              \nReceiptHandle: %s \
                              \nProperties: %s\n" % \
                  (msg.message_id, msg.message_body_md5,
                   msg.message_tag, msg.consumed_times,
                   msg.publish_time, msg.message_body,
                   msg.next_consume_time, msg.receipt_handle, msg.properties)))
    except MQExceptionBase as e:
        # Topic中沒有訊息可消費。
        if e.type == "MessageNotExist":
            print(("No new message! RequestId: %s" % e.req_id))
            continue

        print(("Consume Message Fail! Exception:%s\n" % e))
        time.sleep(2)
        continue

    # msg.next_consume_time前若不確認訊息消費成功,則訊息會被重複消費。
    # 訊息控制代碼有時間戳記,同一條訊息每次消費拿到的都不一樣。
    try:
        receipt_handle_list = [msg.receipt_handle for msg in recv_msgs]
        consumer.ack_message(receipt_handle_list)
        print(("Ak %s Message Succeed.\n\n" % len(receipt_handle_list)))
    except MQExceptionBase as e:
        print(("\nAk Message Fail! Exception:%s" % e))
        # 某些訊息的控制代碼可能逾時,會導致訊息消費狀態確認不成功。
        if e.sub_errors:
            for sub_error in e.sub_errors:
                print(("\tErrorHandle:%s,ErrorCode:%s,ErrorMsg:%s" % \
                      (sub_error["ReceiptHandle"], sub_error["ErrorCode"], sub_error["ErrorMessage"])))