深入探讨：如何在Spring Boot中使用RocketMQ实现事务消息

最新推荐文章于 2024-06-18 08:15:00 发布

SteveCode.

最新推荐文章于 2024-06-18 08:15:00 发布

阅读量688

点赞数 15

文章标签： java-rocketmq spring boot rocketmq

本文链接：https://blog.csdn.net/weixin_48278764/article/details/137201130

版权

生产者端

业务场景：考虑一个在线电商平台的订单支付业务场景，该场景中我们需要处理用户的支付操作并确保订单状态的更新与库存减少这两个操作要么同时成功，要么同时失败，以保证数据的一致性。我们将使用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;
    }
}