rocketmq-producer之发送事物消息_mqproducer 优雅的发送事务消息-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_36569036/article/details/53448142

RocketMQ提供普通、顺序及事务消息支持。事务消息通过发送prepared消息并执行本地事务来协调多个小事务。在消息状态确认过程中，如果消息发送失败，RocketMQ会定期检查并调用TransactionCheckListener进行相应操作。本地事务状态包括COMMIT_MESSAGE、ROLLBACK_MESSAGE和UNKNOW。发送事务消息涉及LocalTransactionState更新、endTransaction方法以及broker的回查线程。发送时必须设置groupname，以便回查时选择producer。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

rocketmq支持普通消息、顺序消息，此外，还支持事物消息。实现方式是将一个大事务拆分成多个小事物异步执行，事物消息在其中起着桥梁作用。

这里写图片描述

rocketmq在发送事物消息时，会先发送一个prepared消息，返回消息所在地址。然后再执行本地事物，根据事物执行结果去更新prepared消息状态。消息接收者只能消费消息集群中消息状态为已提交的消息。

事物消息demo：

TransactionMQProducer producer = new TransactionMQProducer("unique_group_name");
TransactionCheckListener transactionCheckListener = new TransactionCheckListenerImpl();
producer.setTransactionCheckListener(transactionCheckListener);
TransactionExecuterImpl tranExecuter = new TransactionExecuterImpl();
producer.start();
Message msg =new Message(......);
SendResult sendResult = producer.sendMessageInTransaction(msg, tranExecuter, null);

当确认消息发送失败时，rocketmq会定期扫描消息集群中的事物消息，如果发现了Prepared消息，它会向消息发送者确认，如果此时事物执行成功了是回滚还是继续发送确认消息？rocketmq会调用TransactionCheckListener 的实现类来做出相应的操作。

TransactionExecuterImpl –>本地事物的处理逻辑

发送事物消息源码（伪代码）

public TransactionSendResult sendMessageInTransaction(.....)  {
    // 1.发送消息
    SendResult sendResult = this.send(msg);
    LocalTransactionState localTransactionState = LocalTransactionState.UNKNOW;
    switch (sendResult.getSendStatus()) {
    case SEND_OK: {
        // 2.如果消息发送成功，处理与消息关联的本地事务单元
        localTransactionState = tranExecuter.executeLocalTransactionBranch(msg, arg);
        }
    // 3.结束事务
    this.endTransaction(sendResult, localTransactionState, localException);
}

本地事物状态LocalTransactionState：
COMMIT_MESSAGE,
ROLLBACK_MESSAGE,
UNKNOW

 public void endTransaction(SendResult sendResult, LocalTransactionState localTransactionState, ......){

       final String brokerAddr = this.mQClientFactory.findBrokerAddressInPublish(sendResult.getMessageQueue().getBrokerName());
       EndTransactionRequestHeader requestHeader = new EndTransactionRequestHeader();

       switch (localTransactionState) {
            case COMMIT_MESSAGE:
                requestHeader.setCommitOrRollback(MessageSysFlag.TransactionCommitType);
                break;
            case ROLLBACK_MESSAGE:
                requestHeader.setCommitOrRollback(MessageSysFlag.TransactionRollbackType);
                break;
            case UNKNOW:
                requestHeader.setCommitOrRollback(MessageSysFlag.TransactionNotType);
                break;
            default:
                break;
        }

        this.mQClientFactory.getMQClientAPIImpl().endTransactionOneway(brokerAddr, requestHeader, ......);
    }

endTransaction从sendResult中获取消息地址，将请求发送至broker，根据localTransactionState更新事物消息的最终状态。
如果endTransaction方法执行失败，导致数据没有发送到broker，broker会有回查线程定时（默认1分钟）扫描每个存储事务状态的表格文件，如果是已经提交或者回滚的消息直接跳过，如果是TransactionNotType状态则会向Producer发起CheckTransaction请求，Producer会调用DefaultMQProducerImpl.checkTransactionState()方法来处理broker的定时回调请求，而checkTransactionState会调用我们的事务设置的决断方法，最后调用endTransactionOneway让broker来更新消息的最终状态。

发送事物消息时，groupname必须设置，回查时会根据group随机选择一台producer。