事务消息是RocketMQ提供的非常重要的一个特性,在4.x版本之后开源,可以利用事务消息轻松地实现分布式事务。本文对RocketMQ的事务消息进行详细介绍,并给出了代码示例。
一. 相关概念
RocketMQ在其消息定义的基础上,对事务消息扩展了两个相关的概念:
- Half(Prepare) Message——半消息(预处理消息)
半消息是一种特殊的消息类型,该状态的消息暂时不能被Consumer消费。当一条事务消息被成功投递到Broker上,但是Broker并没有接收到Producer发出的二次确认时,该事务消息就处于"暂时不可被消费"状态,该状态的事务消息被称为半消息。
- Message Status Check——消息状态回查
由于网络抖动、Producer重启等原因,可能导致Producer向Broker发送的二次确认消息没有成功送达。如果Broker检测到某条事务消息长时间处于半消息状态,则会主动向Producer端发起回查操作,查询该事务消息在Producer端的事务状态(Commit 或 Rollback)。可以看出,Message Status Check主要用来解决分布式事务中的超时问题。
二. 执行流程
上面是官网提供的事务消息执行流程图,下面对具体流程进行分析:
- Step1:Producer向Broker端发送Half Message;
- Step2:Broker ACK,Half Message发送成功;
- Step3:Producer执行本地事务;
- Step4:本地事务完毕,根据事务的状态,Producer向Broker发送二次确认消息,确认该Half Message的Commit或者Rollback状态。Broker收到二次确认消息后,对于Commit状态,则直接发送到Consumer端执行消费逻辑,而对于Rollback则直接标记为失败,一段时间后清除,并不会发给Consumer。正常情况下,到此分布式事务已经完成,剩下要处理的就是超时问题,即一段时间后Broker仍没有收到Producer的二次确认消息;
- Step5:针对超时状态,Broker主动向Producer发起消息回查;
- Step6:Producer处理回查消息,返回对应的本地事务的执行结果;
- Step7:Broker针对回查消息的结果,执行Commit或Rollback操作,同Step4。
三. 代码实例
本节通过一个简单的场景模拟RocketMQ的事务消息:存在2个微服务,分别是订单服务和商品服务。订单服务进行下单处理,并发送消息给商品服务,对于下单成功的商品进行减库存。
首先是订单服务:
/**
* @Auther: ZhangShenao
* @Date: 2019/3/27 16:44
* @Description:使用RocketMQ事务消息——订单服务发送事务消息,然后进行本地下单,并通知商品服务减库存
*/
public class OrderService {
public static void main(String[] args) throws Exception {
TransactionMQProducer producer = new TransactionMQProducer();
producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
producer.setProducerGroup(RocketMQConstants.TRANSACTION_PRODUCER_GROUP);
//自定义线程池,执行事务操作
ThreadPoolExecutor executor = new ThreadPoolExecutor(10, 50, 10L, TimeUnit.SECONDS, new ArrayBlockingQueue<>(20), (Runnable r) -> new Thread("Order Transaction Massage Thread"));
producer.setExecutorService(executor);
//设置事务消息监听器
producer.setTransactionListener(new OrderTransactionListener());
producer.start();
System.err.println("OrderService Start");
for (int i = 0;i < 10;i++){
String orderId = UUID.randomUUID().toString();
String payload = "下单,orderId: " + orderId;
String tags = "Tag";
Message message = new Message(RocketMQConstants.TRANSACTION_TOPIC_NAME, tags, orderId, payload.getBytes(RemotingHelper.DEFAULT_CHARSET));
//发送事务消息
TransactionSendResult result = producer.sendMessageInTransaction(message, orderId);
System.err.println("发送事务消息,发送结果: " + result);
}
}
}
事务消息需要一个TransactionListener,主要进行本地事务的执行和事务回查,代码如下:
/**
* @Auther: ZhangShenao
* @Date: 2019/3/27 16:50
* @Description:订单事务消息监听器
*/
public class OrderTransactionListener implements TransactionListener {
private static final Map<String, Boolean> results = new ConcurrentHashMap<>();
@Override
public LocalTransactionState executeLocalTransaction(Message msg, Object arg) {
String orderId = (String) arg;
//记录本地事务执行结果
boolean success = persistTransactionResult(orderId);
System.err.println("订单服务执行本地事务下单,orderId: " + orderId + ", result: " + success);
return success ? LocalTransactionState.COMMIT_MESSAGE : LocalTransactionState.ROLLBACK_MESSAGE;
}
@Override
public LocalTransactionState checkLocalTransaction(MessageExt msg) {
String orderId = msg.getKeys();
System.err.println("执行事务消息回查,orderId: " + orderId);
return Boolean.TRUE.equals(results.get(orderId)) ? LocalTransactionState.COMMIT_MESSAGE : LocalTransactionState.ROLLBACK_MESSAGE;
}
private boolean persistTransactionResult(String orderId) {
boolean success = Math.abs(Objects.hash(orderId)) % 2 == 0;
results.put(orderId, success);
return success;
}
}
下面是商品服务及监听器:
/**
* @Auther: ZhangShenao
* @Date: 2019/3/27 17:09
* @Description:使用RocketMQ事务消息——商品服务接收下单的事务消息,如果消息成功commit则本地减库存
*/
public class ProductService {
public static void main(String[] args) throws Exception {
DefaultMQPushConsumer consumer = new DefaultMQPushConsumer();
consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
consumer.setConsumerGroup(RocketMQConstants.TRANSACTION_CONSUMER_GROUP);
consumer.subscribe(RocketMQConstants.TRANSACTION_TOPIC_NAME, "*");
consumer.registerMessageListener(new ProductListener());
consumer.start();
System.err.println("ProductService Start");
}
}
/**
* @Auther: ZhangShenao
* @Date: 2019/3/27 17:14
* @Description:
*/
public class ProductListener implements MessageListenerConcurrently {
@Override
public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> msgs, ConsumeConcurrentlyContext context) {
Optional.ofNullable(msgs).orElse(Collections.emptyList()).forEach(m -> {
String orderId = m.getKeys();
System.err.println("监听到下单消息,orderId: " + orderId + ", 商品服务减库存");
});
return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
}
}
分别运行OrderService和ProductService,可以看出只有事务执行成功的订单才会通知商品服务进行减库存。
监听到下单消息,orderId: f25a7127-307e-45ce-8f83-6e0a922ebb94, 商品服务减库存
监听到下单消息,orderId: d960171d-97c0-4e13-aa4a-c2b96102de4b, 商品服务减库存
监听到下单消息,orderId: 63aedaa2-ce74-4cb7-bf58-fb6a73082a73, 商品服务减库存
监听到下单消息,orderId: 25764461-70b2-44db-8296-960211179e6e, 商品服务减库存
监听到下单消息,orderId: fb319fe7-c8be-4edf-ae4e-6108898068ca, 商品服务减库存
监听到下单消息,orderId: 4f61a61a-7254-458a-bc10-9d4006a9f581, 商品服务减库存