RocketMQ快速入门：事务消息原理及实现（十）

0. 引言

rocketmq 的一大特性就是支持事务性消息，这在诸多场景中有所应用。在之前的文章中我们已经讲解过事务消息的使用，但事务消息是如何实现的呢？今天我们就来探究其原理

1. 原理

rocketmq事务消息的基本流程分为三步：
（1）发送 half 消息到 broker。
（2）执行本地事务。
（3）根据本地事务执行结果：本地事务执行成功则将half消息标识为COMMIT，将消息进行提交，变成普通消息，可以被消费者消费；本地事务执行失败将half消息标识为ROLLBACK，并将消息从消息日志中删除，消费者则消费不到了。

从流程上可以看出rocketmq的事务消息，和数据库的事务有些类型，都有一个二阶段提交的概念。

rocketmq的half message（半消息）是一种特殊的消息，该种消息不会被消费者消费，但是可以被TransactionListener事务监听器获取到。

而我们就在事务监听器中书写自己的本地事务逻辑，本地事务执行成功后才将消息正常提交，这时提交后的消息才能被消费者消费到，否则就会回滚消息，消息就相当于从来没发送过。

这里需要注意的是，因为我们的本地事务逻辑已经在半消息接收的时候处理掉了，所以如果后续没有消费逻辑了，就不用再在消费者里书写逻辑，但如果还有后续的逻辑，就可以按照正常消费流程进行书写。
在这里插入图片描述

但是还有一个场景我们需要考虑，当本地事务执行完成后，在返回“执行成功（提交）”或“执行失败（回滚）”状态时，因为网络波动或者broker服务挂了，导致broker没有正常收到这个状态，从而无法及时把half message进行提交或回滚

这时就需要有个定时巡查机制，来检查这些没有正常收到提交状态的消息的实际状态到底是什么，这个巡查机制就是消息回查，也称为事务消息的补偿。在事务监听器TransactionListener中就是通过checkLocalTransaction方法来实现，executeLocalTransaction方法返回值一共有3种状态：

COMMIT_MESSAGE

提交状态，事务正常进行，一般是本地事务执行成功后进行设置。告知broker提交该事务消息，然后消费者可以消费该消息，当然此时消费者已经执行完本地事务了，再消费可以根据业务逻辑进行后续的逻辑处理，如果没有相关逻辑了忽略消息即可

ROLLBACK_MESSAGE

回滚状态，事务撤回，broker将删除当前half消息，一般是本地事务执行失败后进行设置

UNKNOW

未知状态，固定时间后Broker端会通过checkLocalTransaction方法进行消息回查，根据回查结果来判断该消息是提交还是回滚

示例代码：

@Overridepublic LocalTransactionState executeLocalTransaction(Message msg, Object arg) {// TODO 执行本地事务（书写你自己的本地事务逻辑）try{String body = new String(msg.getBody());int i = Integer.parseInt(body);// 模拟偶数执行成功，奇数执行失败if(i % 2 == 0){System.out.println("本地事务执行成功："+body);// 执行成功return LocalTransactionState.COMMIT_MESSAGE;}else{System.out.println("本地事务执行失败："+body);// 执行失败return LocalTransactionState.ROLLBACK_MESSAGE;}}catch (Exception e){e.printStackTrace();// 执行失败return LocalTransactionState.ROLLBACK_MESSAGE;}}

因此checkLocalTransaction方法中就要书写检查本地事务状态的方法，比如事务是对订单提交消息的消费，那么就去查询订单状态，如果已经是提交状态那么就返回COMMIT_MESSAGE，否则就返回ROLLBACK_MESSAGE

示例代码

       @Overridepublic LocalTransactionState checkLocalTransaction(MessageExt msg) {// TODO 去缓存或者数据库查询当前消息的实际状态// 模拟查询到状态为1Integer status = 1;// 不同实际状态对应的消息状态if (null != status) {switch (status) {case 1:return LocalTransactionState.COMMIT_MESSAGE;case 2:return LocalTransactionState.ROLLBACK_MESSAGE;default:return LocalTransactionState.COMMIT_MESSAGE;}}return LocalTransactionState.COMMIT_MESSAGE;}