基于RocketMQ实现分布式事务

2024-03-12 08:57 由 Y00 发表于 #软件设计

背景

在一个微服务架构的项目中，一个业务操作可能涉及到多个服务，这些服务往往是独立部署，构成一个个独立的系统。这种分布式的系统架构往往面临着分布式事务的问题。为了保证系统数据的一致性，我们需要确保这些服务中的操作要么全部成功，要么全部失败。通过使用RocketMQ实现分布式事务，我们可以协调这些服务的操作，保证数据的一致性。

功能原理

RocketMQ的分布式事务消息功能，在普通消息基础上，支持二阶段的提交。将二阶段提交和本地事务绑定，实现全局提交结果的一致性。

整个事务消息的详细交互流程如下图所示：

1、生产者将消息发送至RocketMQ服务端。

2、RocketMQ服务端将消息持久化成功之后，向生产者返回Ack确认消息已经发送成功，此时消息被标记为"暂不能投递"，这种状态下的消息即为半事务消息。

3、生产者开始执行本地事务逻辑。

4、生产者根据本地事务执行结果向服务端提交二次确认结果（Commit或是Rollback），服务端收到确认结果后处理逻辑如下：

二次确认结果为Commit：服务端将半事务消息标记为可投递，并投递给消费者。
二次确认结果为Rollback：服务端将回滚事务，不会将半事务消息投递给消费者。

5、在断网或者是生产者应用重启的特殊情况下，若服务端未收到生产者提交的二次确认结果，或服务端收到的二次确认结果为Unknown未知状态，经过固定时间后，服务端将对消息生产者集群中任一生产者实例发起消息回查。

6、生产者收到消息回查后，需要检查对应消息的本地事务执行的最终结果。

7、生产者根据检查到的本地事务的最终状态再次提交二次确认，服务端仍按照步骤4对半事务消息进行处理。

注意问题

消息类型
事务消息仅支持在MessageType为Transaction的主题使用，即事务消息只能发送至类型为事务消息的主题中。

消息消费
RocketMQ事务消息保证生产者本地事务和下游消息发送事务的一致性，但不保证消息消费结果和上游事务的一致性。因此需要下游业务自行保证消息正确处理，建议消费端做好消费重试。

中间状态
RocketMQ事务消息一致性为最终一致性，即在消息提交到下游消费端处理完成之前，下游和上游事务之间的状态会不一致。因此，事务消息仅适合能接受异步执行的场景。

事务超时
RocketMQ事务消息的生命周期存在超时机制，即半事务消息被生产者发送服务端后，如果在指定时间内服务端无法确认提交或者回滚状态，则消息默认会被回滚。

示例代码

以下为RocketMQ 4.x版本事务消息示例代码，

import org.apache.rocketmq.client.producer.LocalTransactionState;
import org.apache.rocketmq.client.producer.TransactionListener;
import org.apache.rocketmq.client.producer.TransactionMQProducer;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageExt;

import java.util.concurrent.*;

public class RocketMqTransactionDemo {
	public static void main(String[] args) throws Exception {
		// 创建事务消息生产者
		TransactionMQProducer producer = new TransactionMQProducer("transaction_producer");
		producer.setNamesrvAddr("127.0.0.1:9876");

		// 设置事务监听器
		TransactionListener transactionListener = new MyTransactionListener();
		producer.setTransactionListener(transactionListener);

		// 设置事务回查的线程池，可以不必设置，如果不设置也会默认生成一个
		ExecutorService executorService = new ThreadPoolExecutor(2, 5, 100, TimeUnit.SECONDS, new ArrayBlockingQueue <Runnable> (2000), new ThreadFactory() {
			@Override
			public Thread newThread(Runnable r) {
				Thread thread = new Thread(r);
				thread.setName("client-transaction-msg-check-thread");
				return thread;
			}
		});
		producer.setExecutorService(executorService);

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

		// 发送事务消息
		Message message = new Message("transaction_topic", "test_tag", "test_key", "Hello RocketMQ".getBytes());
		producer.sendMessageInTransaction(message, null);

		// 关闭生产者
		producer.shutdown();
	}
}

/**
 * 事务监听器
 */
class MyTransactionListener implements TransactionListener {
	@Override
	public LocalTransactionState executeLocalTransaction(Message msg, Object arg) {
		// 执行本地事务操作
		System.out.println("执行本地事务操作，消息内容：" + new String(msg.getBody()));
		return LocalTransactionState.COMMIT_MESSAGE; // 提交事务，允许消费者消费该消息
		// return LocalTransactionState.ROLLBACK_MESSAGE;// 回滚事务，消息将被丢弃不允许消费。
		// return LocalTransactionState.UNKNOW;// 暂时无法判断状态，等待固定时间以后Broker端根据回查规则向生产者进行消息回查。
	}

	@Override
	public LocalTransactionState checkLocalTransaction(MessageExt msg) {
		// 检查本地事务状态
		System.out.println("检查本地事务状态，消息内容：" + new String(msg.getBody()));
		return LocalTransactionState.COMMIT_MESSAGE;
	}
}

代码解释：
1、事务消息的生产者使用TransactionMQProducer创建。
2、MyTransactionListener作为事务监听器，实现了接口TransactionListener，该接口有两个方法，分别是：

executeLocalTransaction：
半事务消息发送成功后，执行本地事务的方法，具体执行完本地事务后，可以在该方法中返回以下三种状态：
LocalTransactionState.COMMIT_MESSAGE： 提交事务，允许消费者消费该消息。
LocalTransactionState.ROLLBACK_MESSAGE： 回滚事务，消息将被丢弃不允许消费。
LocalTransactionState.UNKNOW： 暂时无法判断状态，等待固定时间以后RocketMQ服务端根据回查规则向生产者进行消息回查。
checkLocalTransaction：
二次确认消息没有收到，RocketMQ服务端回查生产者端事务结果的方法。回查规则：本地事务执行完成后，若RocketMQ服务端收到的本地事务返回状态为LocalTransactionState.UNKNOW，或生产者应用退出导致本地事务未提交任何状态。则RocketMQ服务端会向消息生产者发起事务回查，第一次回查后仍未获取到事务状态，则之后每隔一段时间会再次回查。