RocketMQ中事务消息的实现机制

关注

原创最新推荐文章于 2026-04-11 23:05:10 发布 1.5k 阅读 AI 写同款· GEO 优化 ›

RocketMQ的事务消息实现原理与Kafka的事务机制差异 RocketMQ和Kafka事务机制对比分析：RocketMQ采用两阶段提交（2PC）实现事务消息，通过半消息和自动回查机制保障原子性，开发简单且隔离性强，适合金融支付等强一致性场景（吞吐10-50万/秒，延迟20-100ms）。Kafka依赖事务协调器和幂等生产者，吞吐更高（100-200万/秒）但需复杂配置，适合流处理等最终一致性场景。RocketMQ优势在于低业务侵入性、灵活的顺序控制和主动回查机制，而Kafka在纯吞吐性能上更优。选型需根据业务场景的可靠性要求与性能需求权衡。阅读详情

在分布式系统中，确保消息与本地事务的一致性是一个关键问题。RocketMQ通过事务消息提供了对这种需求的支持，其核心思想是通过两阶段提交来确保消息和本地事务的原子性。本文将深入探讨RocketMQ事务消息的实现机制，包括基本流程、事务回查机制以及消息状态的处理。

一、事务消息的基本流程

（一）第一阶段：半消息的发送

当生产者发送事务消息时，RocketMQ会将消息存储在一个特殊的队列RMQ_SYS_TRANS_HALF_TOPIC中，并将其标记为“半消息”，此时消费者无法看到该消息。随后，生产者会执行本地事务，并根据事务的结果决定提交或回滚。

- 事务成功：生产者提交事务，通知RocketMQ将半消息转为正式消息。

- 事务失败：生产者回滚事务，RocketMQ会删除该半消息。

（二）第二阶段：事务提交与回滚

- 提交操作：生产者确认事务成功时，发送COMMIT_MESSAGE操作，RocketMQ将半消息转为正式消息，消费者可以消费该消息。

- 回滚操作：如果事务失败，生产者发送ROLLBACK_MESSAGE操作，RocketMQ会丢弃该半消息，消费者无法消费。

二、事务回查机制

在实际场景中，生产者可能因为网络问题、程序崩溃等原因无法及时提交或回滚事务。为了确保消息与本地事务的一致性，RocketMQ引入了事务回查机制。

当RocketMQ检测到某条半消息长时间未提交或回滚时，会主动发起回查请求。生产者需要实现checkLocalTransaction方法，以返回事务的执行状态：

- 事务成功：返回COMMIT_MESSAGE，RocketMQ将半消息提交。

- 事务失败：返回ROLLBACK_MESSAGE，RocketMQ会丢弃消息。

- 状态未知：返回UNKNOWN，RocketMQ会再次回查。

三、消息的三种状态

RocketMQ事务消息有三种状态：

1. COMMIT_MESSAGE：本地事务成功，消息可以投递给消费者。

2. ROLLBACK_MESSAGE：本地事务失败，消息被丢弃。

3. UNKNOWN：事务状态无法确定，RocketMQ会继续回查。

四、源码解析

（一）sendMessageInTransaction方法解析

生产者通过sendMessageInTransaction方法发送事务消息。该方法首先将消息标记为事务消息，并发送到RocketMQ。消息被存储在RMQ_SYS_TRANS_HALF_TOPIC队列中，直到本地事务完成。

public SendResult sendMessageInTransaction(Message msg, LocalTransactionExecuter transactionExecuter, Object arg) throws MQClientException {
    // 设置事务标志
    msg.putUserProperty(MessageConst.PROPERTY_TRANSACTION_PREPARED, "true");
    
    // 发送消息至 Broker
    SendResult sendResult = producer.send(msg);
    
    // 执行本地事务
    transactionExecuter.executeTransaction(msg, arg);
    
    return sendResult;
}

（二）Broker处理事务消息

RocketMQ的Broker会接收到事务消息，并将其存储在特殊队列RMQ_SYS_TRANS_HALF_TOPIC中。此时，消费者无法看到这些消息。Broker会检查消息是否为事务消息，如果是事务消息，则会将其状态设置为“半消息”，并等待生产者提交或回滚事务。

public void sendMessage(SendMessageRequestHeader requestHeader, Message msg) {
    // 检查消息是否为事务消息
    if (msg.getProperties().containsKey(MessageConst.PROPERTY_TRANSACTION_PREPARED)) {
        // 事务消息处理
        messageQueue = new MessageQueue(RMQ_SYS_TRANS_HALF_TOPIC, "0");
        msg.setTopic(RMQ_SYS_TRANS_HALF_TOPIC);
    }
    // 存储消息
    brokerController.getMessageStore().putMessage(msg);
}

（三）事务提交与回滚

当生产者决定提交或回滚事务时，RocketMQ的Broker会通过EndTransactionProcessor处理请求，根据事务状态将消息移出半消息队列，放入正式队列或丢弃。

public void processRequest(ChannelHandlerContext ctx, EndTransactionRequestHeader requestHeader) {
    if (requestHeader.getTransactionStatus() == TransactionStatus.CommitMessage) {
        // 提交事务，移动消息至正式队列
        brokerController.getMessageStore().putMessage(requestHeader.getMessage());
    } else if (requestHeader.getTransactionStatus() == TransactionStatus.RollbackMessage) {
        // 回滚事务，删除半消息
        brokerController.getMessageStore().deleteMessage(requestHeader.getMessage());
    }
}

（四）事务回查

如果半消息未被确认，RocketMQ会定期扫描RMQ_SYS_TRANS_HALF_TOPIC队列，并发起回查请求。生产者通过checkLocalTransaction方法返回事务状态。

public TransactionStatus checkLocalTransactionState(Message msg) {
    // 检查本地事务状态
    boolean success = localTransactionService.checkTransactionStatus(msg);
    if (success) {
        return TransactionStatus.CommitMessage;
    } else {
        return TransactionStatus.RollbackMessage;
    }
}