在Spring中,我们要实现事务,一般通过@Transactional注解实现。这在引入RocketMQ之前没有问题,但是在引入了RocketMQ之后,如果消息发送之后的业务逻辑处理发生了异常的话,这时候消息已经发送出去了,就会导致业务的问题。
为了解决这一问题,RocketMQ引入了Transactional Message【事务消息】。
- 生产者向MQServer发送半消息【半消息:会存储进MQ Server,但是被标记为不能投递状态】
- 发送半消息成功,生产者实行本地事务
- 根据本地事务结果向MQ Server发送二次确认请求
- MQ Server根据接受到的消息投递或者丢弃消息
- 若在本地事务执行过程中缺少二次确认消息或生产者处于等待状态,MQ服务器将向同一组中的每个生产者发送检查消息,然后继续3,4的操作
PS:消息三态
- Commit:提交事务信息,消费者可以消费此消息
- Rollback:回滚事务消息,broker会删除这条消息,消费者不能消费
- UNKNOWN:broker需要回查确认消息状态
代码实现
发送半消息
利用rocketMQTemplate类的sendMessageInTransaction实现半消息发送
第一个参数为txProducerGroup:就是group名称,根据业务自定义
第二次信息为destination:topic名称
第三个信息为message:消息体,利用MessageBuilder.withPayload构建
第四个信息为arg:业务对象,用于处理本地业务
代码如下:
// 发送半消息
rocketMQTemplate.sendMessageInTransaction(
"test-transactional",
"test-topic",
MessageBuilder.withPayload(
Demo.builder().demoId(1).remark("哈哈哈").build()
).setHeader(RocketMQHeaders.TRANSACTION_ID, UUID.randomUUID().toString()).build(),
forObject
);
实现RocketMQLocalTransactionListener接口
新建demoTransactionalListener类,继承RocketMQLocalTransactionListener接口,实现了两个方法
executeLocalTransaction:处理本地业务
checkLocalTransaction:MQ Server发送检查信息相应
添加@RocketMQTransactionListener注解,txProducerGroup属性值与半消息的txProducerGroup参数值相同
代码如下:
import org.apache.rocketmq.spring.annotation.RocketMQTransactionListener;
import org.apache.rocketmq.spring.core.RocketMQLocalTransactionListener;
import org.apache.rocketmq.spring.core.RocketMQLocalTransactionState;
import org.apache.rocketmq.spring.support.RocketMQHeaders;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
@RocketMQTransactionListener(txProducerGroup = "test-transactional")
public class demoTransactionalListener implements RocketMQLocalTransactionListener {
/**
* 处理本地事务
*/
@Override
public RocketMQLocalTransactionState executeLocalTransaction(Message message, Object arg) {
// 消息头
MessageHeaders headers = message.getHeaders();
String transactionalId = (String) headers.get(RocketMQHeaders.TRANSACTION_ID);
// arg:sendMessageInTransaction方法的第四个参数,用于处于本地业务
try {
// 本地业务
return RocketMQLocalTransactionState.COMMIT;
} catch (Exception e) {
return RocketMQLocalTransactionState.ROLLBACK;
}
}
/**
* 若在本地事务执行过程中缺少二次确认消息或生产者处于等待状态
* MQ Server将向同一组中的每个生产者发送检查消息
*/
@Override
public RocketMQLocalTransactionState checkLocalTransaction(Message message) {
try {
// 检查
return RocketMQLocalTransactionState.COMMIT;
} catch (Exception e) {
return RocketMQLocalTransactionState.ROLLBACK;
}
}
}
PS:MQ Server发送检查:可以通过新建一个表,记录本地业务成功或者失败,然后检查相应只需要查一下数据就可以了