生产者端
业务场景:考虑一个在线电商平台的订单支付业务场景,该场景中我们需要处理用户的支付操作并确保订单状态的更新与库存减少这两个操作要么同时成功,要么同时失败,以保证数据的一致性。我们将使用RocketMQ事务消息来实现这个需求。
业务场景描述
- 用户在电商平台上下单购买商品。
- 用户进行支付操作。
- 系统需要同时进行两个操作:
- 更新订单状态为“已支付”。
- 减少对应商品的库存数量。
这两个操作都需要成功执行,才能保证业务数据的一致性。如果在执行这两个操作的过程中,任何一个操作失败,系统都需要进行回滚,保证数据不会出现不一致的情况。
使用RocketMQ事务消息实现
为了确保上述操作的原子性,我们可以使用RocketMQ的事务消息功能来实现。具体步骤如下:
步骤1:用户支付操作
用户在前端进行支付操作后,系统首先发送一个预备事务消息到RocketMQ,该消息暂时不会被消费。
步骤2:执行本地事务
系统接着执行本地事务逻辑,即更新订单状态和减少商品库存。这两个操作需要在同一个数据库事务中执行,以确保它们要么同时成功,要么同时失败。
步骤3:返回事务状态
- 如果本地事务成功,系统会通知RocketMQ提交事务消息,这时消费者可以消费这条消息进行后续处理,例如发送支付成功通知给用户。
- 如果本地事务失败,系统会通知RocketMQ回滚事务消息,这时消息将不会被消费,系统还可以进行一些回滚操作,例如退款给用户。
代码实现(简化示例)
发送事务消息
@Service
public class PaymentService {
@Autowired
private RocketMQTemplate rocketMQTemplate;
@Transactional
public void payOrder(String orderId, BigDecimal paymentAmount) {
// 构建事务消息
String transactionId = UUID.randomUUID().toString();
Message<String> message = MessageBuilder.withPayload(orderId)
.setHeader(RocketMQHeaders.TRANSACTION_ID, transactionId)
.build();
// 发送事务消息
rocketMQTemplate.sendMessageInTransaction("pay-topic", message, null);
}
}
事务监听器实现
@RocketMQTransactionListener(txProducerGroup = "pay-group")
public class PayTransactionListener implements RocketMQLocalTransactionListener {
@Autowired
private OrderService orderService; // 订单服务
@Autowired
private InventoryService inventoryService; // 库存服务
@Override
public RocketMQLocalTransactionState executeLocalTransaction(Message msg, Object arg) {
try {
String orderId = new String((byte[]) msg.getPayload());
// 执行本地事务操作
orderService.updateOrderStatus(orderId, "PAID");
inventoryService.decreaseInventory(orderId);
return RocketMQLocalTransactionState.COMMIT;
} catch (Exception e) {
return RocketMQLocalTransactionState.ROLLBACK;
}
}
@Override
public RocketMQLocalTransactionState checkLocalTransaction(Message msg) {
// 检查本地事务执行状态
String orderId = new String((byte[]) msg.getPayload());
boolean isPaid = orderService.checkOrderPaid(orderId);
return isPaid ? RocketMQLocalTransactionState.COMMIT : RocketMQLocalTransactionState.ROLLBACK;
}
}
在这个简化的示例中,PaymentService
负责发送事务消息,而 PayTransactionListener
则根据本地事务的执行结果来决定是否提交或回滚消息。这样就能够确保订单支付和库存减少这两个操作要么同时成功,要么同时失败,从而保持业务数据的一致性。
消费者端:
消费者是用来消费事务消息并处理支付成功后的逻辑的。在实际应用中,你需要编写消费者代码来监听并处理支付成功的事务消息。
消费者的代码通常会监听一个特定的主题(Topic),当有新的事务消息到达时,消费者会收到通知并执行相应的业务逻辑。在这个例子中,消费者需要监听与支付相关的主题,当收到支付成功的事务消息时,可以执行后续的业务逻辑,比如发送支付成功通知给用户,更新用户账户余额等。
所以,虽然在示例中没有直接包含消费者的代码,但在实际应用中,消费者是必须的,因为他们负责处理事务消息的结果。
以上出自:AI