RocketMQ 中 使用 Half 消息

RocketMQ 中 使用 Half 消息

在Rocket MQ中，Half消息（半消息）是一种特殊的消息类型，通常用于分布式事务场景。在这种场景下，消息的生产者发送一条消息到消息队列，但这条消息在事务提交之前处于“预备提交”状态，即Half消息状态。如果事务最终提交成功，则这条Half消息会被正式提交；如果事务回滚，则这条Half消息会被删除。

1.生产者发送Half消息

在生产者发送消息时，需要设置消息为Half消息。这通常是通过在发送消息时指定事务属性来实现的。以下是一个简化的示例代码：

// 假设已经创建了producer实例
TransactionMQProducer producer = new TransactionMQProducer("group_name");

// 设置本地事务监听器
producer.setTransactionListener(new TransactionListenerImpl());

// 启动生产者
producer.start();

// 发送Half消息
Message msg = new Message("TopicTest", "TagA", ("Hello RocketMQ ").getBytes());
SendResult sendResult = producer.sendMessageInTransaction(msg, null);

// 检查发送结果
System.out.printf("%s%n", sendResult);

在这个示例中，sendMessageInTransaction方法用于发送Half消息。第一个参数是消息本身，第二个参数是用户自定义的事务上下文，可以为null。

2. 执行本地事务

生产者需要实现TransactionListener接口来处理本地事务和提交/回滚Half消息的逻辑。以下是一个简单的TransactionListener实现示例：

public class TransactionListenerImpl implements TransactionListener {

    @Override
    public LocalTransactionState executeLocalTransaction(Message msg, Object arg) {
        // 执行本地事务逻辑
        boolean result = executeLocalBusinessLogic(msg);
        
        if (result) {
            // 事务执行成功，返回COMMIT_MESSAGE
            return LocalTransactionState.COMMIT_MESSAGE;
        } else {
            // 事务执行失败，返回ROLLBACK_MESSAGE
            return LocalTransactionState.ROLLBACK_MESSAGE;
        }
    }

    @Override
    public LocalTransactionState checkLocalTransaction(MessageExt msg) {
        // 检查本地事务状态
        boolean result = checkLocalBusinessLogic(msg);
        
        if (result) {
            // 事务已提交，返回COMMIT_MESSAGE
            return LocalTransactionState.COMMIT_MESSAGE;
        } else {
            // 事务已回滚或未知状态，返回ROLLBACK_MESSAGE
            return LocalTransactionState.ROLLBACK_MESSAGE;
        }
    }

    private boolean executeLocalBusinessLogic(Message msg) {
        // 实现本地事务逻辑
        // 返回true表示事务成功，返回false表示事务失败
        return true;
    }

    private boolean checkLocalBusinessLogic(MessageExt msg) {
        // 检查本地事务状态
        // 返回true表示事务已提交，返回false表示事务已回滚或未知状态
        return true;
    }
}

在executeLocalTransaction方法中，您需要实现本地事务逻辑，并根据事务执行的结果返回相应的LocalTransactionState。在checkLocalTransaction方法中，您可以检查本地事务的状态，并返回相应的LocalTransactionState。

3. 提交或回滚Half消息

Rocket MQ会根据TransactionListener返回的结果自动提交或回滚Half消息。如果executeLocalTransaction方法返回COMMIT_MESSAGE，则Half消息会被提交；如果返回ROLLBACK_MESSAGE，则Half消息会被回滚。