深度剖析 RocketMQ 事务消息！

猿 java

于 2024-08-04 21:47:12 发布

阅读量997

点赞数 9

文章标签： java-rocketmq rocketmq 数据库

本文链接：https://blog.csdn.net/m0_54369189/article/details/140912608

版权

你好，我是猿java。

这篇文章，我们将深入探讨 RocketMQ 的事务消息原理，并从源码角度进行分析，以及事务消息适合什么场景，使用事务消息需要注意哪些事项。

什么是事务消息

事务消息是为了保证分布式系统中消息的一致性而引入的一种消息类型。事务消息允许消息发送方在发送消息后，进行本地事务操作，并根据本地事务的执行结果来决定消息的最终状态（提交或回滚）。

RocketMQ 事务消息的基本流程

RocketMQ 的事务消息流程大致可以分为以下几个阶段：

发送half消息：生产者首先发送一条 “half消息” 到 RocketMQ Broker。half消息是指消息已经发送到 Broker，但此时消息状态不确定，该消息对消费者不可见。
执行本地事务：生产者在发送half消息之后，立即执行本地事务操作。例如，更新数据库、调用外部服务等。
提交或回滚事务消息：本地事务操作完成后，生产者根据本地事务的执行结果，向 Broker 发送 “提交” 或 “回滚” 请求。如果本地事务执行成功，则发送 “提交” 请求，使得half消息变为可消费的正式消息；如果本地事务失败，则发送 “回滚” 请求，Broker 将删除该half消息。
事务状态回查：如果在规定时间内 Broker 没有收到提交或回滚请求，Broker 会主动向消息发送方发起事务状态回查，以确认该消息的最终状态。
下面给出一个完整事务消息发送示例：

public class TransactionProducer {
    public static void main(String[] args) throws Exception {
        // 创建事务消息生产者
        TransactionMQProducer producer = new TransactionMQProducer("TransactionProducerGroup");
        producer.setNamesrvAddr("localhost:9876");

        // 设置事务状态回查监听器
        producer.setTransactionCheckListener(new TransactionCheckListener() {
            @Override
            public LocalTransactionState checkLocalTransactionState(MessageExt msg) {
                // 处理事务状态回查逻辑
                System.out.println("Checking transaction state for message: " + new String(msg.getBody()));
                return LocalTransactionState.COMMIT_MESSAGE;
            }
        });

        // 启动生产者
        producer.start();

        // 发送事务消息
        Message msg = new Message("TransactionTopic", "TagA", "Transaction Message".getBytes());
        SendResult sendResult = producer.sendMessageInTransaction(msg, new LocalTransactionExecuter() {
            @Override
            public LocalTransactionState executeLocalTransactionBranch(Message msg, Object arg) {
                // 执行本地事务逻辑
                System.out.println("Executing local transaction for message: " + new String(msg.getBody()));
                // 假设本地事务执行成功，返回 COMMIT_MESSAGE
                // 如果本地事务失败，返回 ROLLBACK_MESSAGE
                return LocalTransactionState.COMMIT_MESSAGE;
            }
        }, null);

        System.out.println("Send result: " + sendResult);

        // 阻塞主线程，防止退出
        System.in.read();

        // 关闭生产者
        producer.shutdown();
    }
}

RocketMQ 事务消息的源码分析

发送`half`消息

发送half消息的核心代码在 TransactionMQProducer 类中，通过 sendMessageInTransaction 方法实现：

public TransactionSendResult sendMessageInTransaction(Message msg, LocalTransactionExecuter tranExecuter, Object arg) {
    // 1. 发送`half`消息
    SendResult sendResult = this.defaultMQProducerImpl.send(msg);

    // 2. 执行本地事务
    LocalTransactionState localTransactionState = tranExecuter.executeLocalTransactionBranch(msg, arg);
    
    // 3. 根据本地事务状态提交或回滚消息
    this.endTransaction(msg, localTransactionState);
    
    return new TransactionSendResult(sendResult, localTransactionState);
}

在 sendMessageInTransaction 方法中，首先调用 send 方法发送half消息，然后执行本地事务，并根据本地事务的结果调用 endTransaction 方法提交或回滚消息。

执行本地事务

本地事务的执行由 LocalTransactionExecuter 接口的实现类来完成。在实际使用中，用户需要实现该接口，并在 executeLocalTransactionBranch 方法中定义具体的本地事务逻辑。

public interface LocalTransactionExecuter {
    LocalTransactionState executeLocalTransactionBranch(final Message msg, final Object arg);
}

提交或回滚事务消息

提交或回滚事务消息的实现也在 TransactionMQProducer 类中，通过 endTransaction 方法完成：

private void endTransaction(Message msg, LocalTransactionState localTransactionState) {
    // 构建事务结束请求
    EndTransactionRequestHeader requestHeader = new EndTransactionRequestHeader();
    requestHeader.setCommitOrRollback(localTransactionState == LocalTransactionState.COMMIT_MESSAGE ? 0 : 1);
    requestHeader.setTranStateTableOffset(msg.getQueueOffset());
    requestHeader.setCommitLogOffset(msg.getCommitLogOffset());

    // 发送事务结束请求到 Broker
    this.defaultMQProducerImpl.endTransaction(msg, requestHeader);
}

在 endTransaction 方法中，根据本地事务的执行结果构建事务结束请求，并调用 endTransaction 方法将请求发送到 Broker。

事务状态回查

事务状态回查是由 Broker 发起的。当 Broker 在规定时间内没有收到提交或回滚请求时，会主动向消息发送方发起事务状态回查。回查的实现主要在 TransactionCheckListener 接口中：

public interface TransactionCheckListener {
    LocalTransactionState checkLocalTransactionState(final MessageExt msg);
}

消息发送方需要实现 TransactionCheckListener 接口，并在 checkLocalTransactionState 方法中定义如何检查本地事务的状态。

Broker 端的事务消息处理

Broker 端的事务消息处理主要在 TransactionalMessageServiceImpl 类中实现。Broker 负责接收half消息、提交或回滚请求，并在必要时发起事务状态回查。

接收`half`消息

Broker 接收half消息的逻辑在 TransactionalMessageServiceImpl 类的 prepareMessage 方法中：

public PutMessageResult prepareMessage(MessageExtBrokerInner msgInner) {
    // 存储`half`消息
    return this.store.putMessage(msgInner);
}

提交或回滚消息

Broker 处理提交或回滚请求的逻辑在 TransactionalMessageServiceImpl 类的 commitMessage 和 rollbackMessage 方法中：

public boolean commitMessage(MessageExt msgExt) {
    // 提交消息
    return this.store.commitTransaction(msgExt);
}

public boolean rollbackMessage(MessageExt msgExt) {
    // 回滚消息
    return this.store.rollbackTransaction(msgExt);
}

事务状态回查

Broker 发起事务状态回查的逻辑在 TransactionalMessageServiceImpl 类的 check 方法中：

public void check(long transactionTimeout, int transactionCheckMax, String topic) {
    // 遍历`half`消息队列，发起事务状态回查
    List<MessageExt> halfMessages = this.store.getHalfMessages(topic);
    for (MessageExt msg : halfMessages) {
        // 发起回查请求
        this.brokerController.getBroker2Client().checkProducerTransactionState(msg);
    }
}