您现在的位置是:首页 >学无止境 >分布式消息中间件RocketMQ的应用网站首页学无止境

分布式消息中间件RocketMQ的应用

ruozhuliufeng 2024-06-17 11:19:17
简介分布式消息中间件RocketMQ的应用

RocketMQ 应用

所有代码同步至GitCode:https://gitcode.net/ruozhuliufeng/test-rocketmq.git

普通消息

消息发送分类

​ Producer对于消息的发送方式也有多种选择,不同的方式会产生不同的系统效果。

同步发送消息

​ 同步发送消息是指,Producer发出一条消息后,会在收到MQ返回的ACK之后才发下一条消息。该方式的消息可靠性最高,但消息发送效率太低。

image-20230509140625652

异步发送消息

​ 异步发送消息是指,Producer发出消息后无需等待MQ返回ACK,直接发送下一条消息。该方式的消息可靠性可以得到保障,消息发送效率也可以。

image-20230509140737210

单向发送消息

​ 单向发送消息是指,Producer仅负责发送消息,不等待、不处理MQ的ACK。该发送方式时MQ也不返回ACK。该方式的消息发送效率最高,但消息可靠性较差。

image-20230509140752873

示例代码

创建工程

​ 创建一个Maven工程 test-rocketmq

导入依赖

​ 导入rocketmq的client依赖

        <dependency>
            <groupId>org.apache.rocketmq</groupId>
            <artifactId>rocketmq-client</artifactId>
            <version>5.1.0</version>
        </dependency>
定义同步消息发送生产者
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.client.producer.SendStatus;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

/**
 * 同步消息发送生产者
 */
public class SyncProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置当发送失败时重试发送的次数,默认为2次
        producer.setRetryTimesWhenSendFailed(3);
        // 设置发送超时时限为10s,默认3s
        producer.setSendMsgTimeout(20000);
        // 开启生产者
        producer.start();
        // 生产并发送100条消息
        for (int i = 0; i < 100; i++) {
            byte[] body = ("Hi,RocketMQ Test "+i).getBytes();
            Message msg = new Message("testTopic","testTag",body);
            // 为消息指定key
            msg.setKeys("key-"+i);
            // 发送消息
            SendResult sendResult = producer.send(msg);
            System.out.println(sendResult);
        }
        // 关闭 生产者
        producer.shutdown();
    }
}

// 消息发送的状态
public enum SendStatus {
    SEND_OK, 			// 发送成功
    FLUSH_DISK_TIMEOUT, // 刷盘超时,当Broker设置的刷盘策略为同步刷盘时才可能出现这种异常状态,异步刷盘不会出现
    FLUSH_SLAVE_TIMEOUT,// slave同步超时。当Broker集群设置的Master-Slave的复制方式为同步复制时才可能出现这种异常,异步复制不会出现
    SLAVE_NOT_AVAILABLE;// 没有可用的Slave。当Broker集群设置的Master-Slave的复制方式为同步复制时才可能出现这种异常,异步复制不会出现
}

​ 发送结果:

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-OTRlcrbN-1684484588191)(C:/Users/ruozhuliufeng/AppData/Roaming/Typora/typora-user-images/image-20230517201912961.png)]

定义异步消息发送生产者
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendCallback;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.util.concurrent.TimeUnit;

/**
 * 异步消息发送生产者
 */
public class AsyncProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定异步发送失败后不进行重试发送
        producer.setRetryTimesWhenSendAsyncFailed(0);
        // 指定新创建的Topic的Queue数量是2,默认为4
        producer.setDefaultTopicQueueNums(2);
        // 设置发送超时时限为10s,默认3s
        producer.setSendMsgTimeout(20000);
        // 开启生产者
        producer.start();
        // 生产并发送100条消息
        for (int i = 0; i < 100; i++) {
            byte[] body = ("Hi,RocketMQ Async Test "+i).getBytes();
            try {
                Message msg = new Message("testTopic", "testAsyncTag", body);
                // 为消息指定key
                msg.setKeys("key-" + i);
                // 异步发送,指定回调
                producer.send(msg, new SendCallback() {
                    // 当producer接收到MQ发送来的ACK就会出发该回调方法的执行
                    @Override
                    public void onSuccess(SendResult sendResult) {
                        System.out.println("发送结果:"+sendResult);
                    }

                    @Override
                    public void onException(Throwable throwable) {
                        throwable.printStackTrace();
                    }
                });
            }catch (Exception e){
                e.printStackTrace();
            }
        }
        /// sleep 一会
        // 由于采用的是异步发送,所以若这里不sleep
        // 则消息还未发送就会被producer给关闭,报错
        TimeUnit.SECONDS.sleep(Long.MAX_VALUE);
        // 关闭 生产者
        producer.shutdown();
    }
}

定义单向消息发送生产者
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

/**
 * 单向消息发送生产者
 */
public class OnewayProducer {
    public static void main(String[] args) throws Exception {
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        producer.setSendMsgTimeout(20000);
        producer.start();
        for (int i = 0; i < 100; i++) {
            byte[] body = ("Hi,RocketMQ Oneway Test "+i).getBytes();
            Message msg = new Message("single","testTag",body);
            // 单向发送
            producer.sendOneway(msg);
        }
        // 关闭 生产者
        producer.shutdown();
        System.out.println("producer shutdown");
    }
}

定义消息消费者
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyContext;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.util.List;

/**
 * 消息消费者
 */
public class NormalConsumer {
    public static void main(String[] args) throws MQClientException {
        // 定义一个pull消费者
//        DefaultLitePullConsumer consumer = new DefaultLitePullConsumer("cg");
        // 定义一个push消费者
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
        // 指定NameServer
        consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定从第一条消息开始消费
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        // 指定消费Topic与Tag
        consumer.subscribe("testTopic","*");
        // 指定采用“广播模式”进行消费,默认为集群模式
//        consumer.setMessageModel(MessageModel.BROADCASTING);
        // 注册消息监听器
        consumer.registerMessageListener(new MessageListenerConcurrently() {
            // 一旦Broker中有其订阅的消息就会触发该方法的执行,
            // 其返回值为当前consumer消费的状态
            @Override
            public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list, ConsumeConcurrentlyContext consumeConcurrentlyContext) {
                // 逐条消费消息
                for (MessageExt msg:list){
                    System.out.println(msg);
                }
                // 返回消费状态:消费成功
                return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
            }
        });
        // 开启消费者消费
        consumer.start();
        System.out.println("consumer started");
    }
}

顺序消息

什么是顺序消息

​ 顺序消息指的是,严格按照消息的发送顺序进行消费的消息(FIFO)

​ 默认情况下生产者会把消息以Round Robin轮询方式发送到不同的Queue分区队列;而消费消息时会从多个Queue中拉取消息,这种情况下的发送和消费是不能保证顺序的。如果将消息仅发送到同一个Queue中,消费时也只从这个Queue上拉取消息,就严格保证了消息的顺序性。

为什么需要顺序消息

​ 例如,现在Topic ORDER_STATUS(订单状态),其下有4个Queue队列,该Topic中的不同消息用于描述当前订单的不同状态。假设订单有状态:未支付、已支付、发货中、发货成功、发货失败

​ 根据以上订单状态,生产者从时序上可以生成如下几个消息:

订单T00000001:未支付 —> 订单T00000001:已支付 —> 订单T00000001:发货中—> 订单T00000001:发货失败

​ 消息发送到MQ中之后,Queue的选择如果采用轮询策略,消息在MQ的存储可能如下:

image-20230512195826700

​ 这种情况下,我们希望Consumer消费消息的顺序和我们发送是一致的,然而上述MQ的投递和消费方式,我们无法保证顺序是正确的。对于顺序异常的消息,Consumer即使设置有一定的状态容错,也不能完全处理好这么多种随机出现组合情况。

image-20230512201745376

​ 基于上述的情况,可以设计如下方案:对于相同订单号的消息,通过一定的策略,将其放置在一个Queue中,然后消费者再采用一定的策略(例如,一个线程独立处理一个Queue,保证处理消息的顺序性),能够保证消费的顺序性。

有序性分类

​ 根据有序范围的不同,RocketMQ可以严格地保证两种消息的有序性:分区有序全局有序

全局有序

image-20230514162004159

​ 当发送和消费参与的Queue只有一个时锁保证的有序是整个Topic中消息的顺序,称为全局有序

在创建Topic时指定Queue的数量,有三种指定方式:

  • 1)在代码中创建Producer时,可以指定其自动创建的Topic的Queue数量
  • 2)在RocketMQ可视化控制台中手动创建Topic时指定Queue数量
  • 3)使用mqadmin命令手动创建Topic时指定Queue数量
分区有序

image-20230514162230715

​ 如果有多个Queue参与,其仅可保证在该Queue分区队列上的消息顺序,则称为分区有序

如何实现Queue的选择?在定义Producer时我们可以指定消息队列选择器,而这个选择器是我们自己实现了MessageQueueSelector接口定义的。

在定义选择器的选择算法时,一般需要使用选择KEY。这个选择KEY可以是消息的KEY,也可以是其他数据。但无论谁做选择KEY,都不能重复,都是唯一的。

一般性的选择算法是,让选择KEY(或者其Hash值)与该Topic所包含的Queue的数量取模,其结果即为选择出的Queue的QueueId。

取模算法存在一个问题:不同选择key与Queue数量取模结果可能会是相同的,即不同选择key的消息可能会出现相同的Queue,即同一个Consumer可能会消费到不同选择KEY的消息。这个问题如何解决?一般性的做法是,从消息中获取到选择KEY,对其进行判断。若是当前Consumer需要消费的消息,则直接消费,否则什么也不做。这种做法要求选择KEY要能够随着消息一起被Consumer获取到。此时使用消息KEY作为选择KEY是比较好的做法。

以上做法会不会出现新的问题呢?不属于那个Consumer的消息被拉取走了,那么应该消费该消息的Consumer是否还能再消费该消息呢?同一个Queue中的消息不可能被同一个Group中的不同Consumer同时消费。所以,消费同一个Queue的不同选择KEY的消息的Consumer一定属于不同的Group。而不同Group中的Consumer间的消费是相互隔离的,互不影响的。

示例代码

import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

/**
 * 顺序消息
 */
public class OrderedProducer {

    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置发送超时时限为20s,默认3s
        producer.setSendMsgTimeout(20000);
        // 开启Producer
        producer.start();
        for (int i = 0; i < 100; i++) {
            Integer orderId = i;
            byte[] body = ("Hi, Ordered Msg "+i).getBytes();
            Message msg = new Message("testTopic","TagA",body);
            SendResult sendResult = producer.send(msg, (list, message, o) -> {
                Integer id = (Integer) o;
                int index = id % list.size();
                return list.get(index);
            },orderId);
            System.out.println("发送结果:"+sendResult);
        }
        producer.shutdown();
    }
}

延时消息

什么是延时消息

​ 当消息写入到Broker后,在指定的时长后才可被消费处理的消息,称为延时消息。

​ 采用RocketMQ的延时消息可以实现定时任务的功能,而无需使用定时器。典型的应用场景是,电商交易中超时未支付关闭订单的场景,12306平台订票超时未支付取消订票的场景。

在电商平台中,订单创建时会发送一条延时消息。这条消息将会在30分钟后投递给后台业务系统(Consumer),后台业务系统收到该消息后会判断对应的订单是否已经完成支付。如果未完成,则取消订单,将商品再次放回到库存;如果完成支付,则忽略。

在12306平台中,车票预定成功后就会发送一条延迟消息。这条消息将会在45分钟后投递给后台业务系统(Consumer),后台业务系统收到该消息后会判断对应的订单是否已经完成支付。如果未完成,则取消预定,将车票再次返回到票池;如果完成支付,则忽略。

延时消息等级

​ 延迟消息的延迟时长不支持随意时长的延迟,是通过特定的延迟等级来指定的。延迟等级定义在RocketMQ服务端的MessageStoreConfig类中的如下变量中:

image-20230515163112856

​ 即,若指定的延迟等级为3,则表示延迟时长为10s,即延迟等级是1开始的。

​ 当然,如果需要自定义的延迟等级,可以通过在Broker加载的配置中新增如下配置(例如下面增加了1天这个等级 1d)。配置文件在RocketMQ安装目录下的conf目录中。

messageDelayLevel = 1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h 1d

延时消息实现原理

image-20230515163342104

​ 具体实现方案是:

修改消息

image-20230515163410340

​ Producer将消息发送到Broker后,Broker会首先将消息写入到commitlog文件,然后需要将其分发到相应的consumequeue。不过,在分发之前,系统会先判断消息中是否带有延迟等级。若没有,则直接正常分发;若有,则需要经历一个复杂的过程:

  • 修改消息的Topic为SCHEDULE_TOPIC_XXXXX

  • 根据延迟等级,在consumequeue目录中的SCHEDULE_TOPIC_XXXXXX主题下创建出相应的queueId目录与consumequeue文件(如果没有这些目录和文件的话)。

    延迟等级 delayLevel与queueId的对应关系为 queueId = delayLevel - 1

    需要注意,在创建queueId目录时,并不是一次性的将所有延迟等级对应的目录全部创建完毕,而是用到哪个延迟等级创建哪个目录。

image-20230515163932891

  • 修改消息索引单元内容。索引单元的Message Tag HashCode部分原本存放的是消息的Tag的Hash值,现修改为消息的投递时间。投递时间是指该消息被重新修改为原Topic后再次被写入到commitlog中的时间。投递时间 = 消息存储时间 + 延迟等级时间。消息存储时间指的是消息被发送到Broker时的时间戳。

  • 将消息索引写入到SCHEDULE_TOPIC_XXXXX主题下相应的consumequeue中。

SCHEDULE_TOPIC_XXXXX目录中各个延迟等级Queue中的消息是如何排序的?

是按照消息投递时间排序的。一个Broker中同一等级的所有延迟消息会被写入到consumequeue目录中SCHEDULE_TOPIC_XXXXX目录下相同Queue中。即一个Queue中消息投递时间的延迟等级相同的。那么投递时间就取决于消息存储时间了。即按照消息被发送到Broker的时间进行排序的。

投递延时消息

​ Broker内部有一个延迟消息服务类ScheuleMessageService,其会消费SCHEDULE_TOPIC_XXXXX中的消息,即按照每条消息的投递时间,将延迟消息投递到目标Topic中。不过,在投递之前会从commitlog中将原来写入的消息再次读出,并将其原来的延迟等级设置为0,即原消息变为了一条不延迟的普通消息。然后再次将消息投递到目标Topic中。

ScheuleMessageService在Broker启动时,会创建并启动一个定时器Timer,用于执行相应的定时任务。系统会根据延迟等级的个数,定义相应数量的TimerTask,每个TimerTask负责一个延迟等级消息的消费与投递。每个TimerTask都会检测相应Queue队列的第一条消息是否到期。若第一条消息未到期,则后面的所有消息更不会到期(消息是按照投递时间排序的);若第一条消息到期了,则将该消息投递到目标Topic,即消费该消息。

将消息重新写入commitlog

​ 延迟消息服务类ScheuleMessageService将延迟消息再次发送给了commitlog,并再次形成新的消息索引条目,分发到相应的Queue。

这其实就是一次普通消息发送。只不过这次的消息Producer是延迟消息服务类ScheuleMessageService。

示例代码

定义DelayProducer类
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * 延时消息生产者
 */
public class DelayProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置发送超时时限为20s,默认3s
        producer.setSendMsgTimeout(20000);
        // 开启Producer
        producer.start();
        for (int i = 0; i < 100; i++) {
            Integer orderId = i;
            byte[] body = ("Hi, Delay Msg "+i).getBytes();
            Message msg = new Message("TopicB","TagB",body);
            // 指定消息延迟时间等级为3级,即延迟10秒
            msg.setDelayTimeLevel(3);
            SendResult sendResult = producer.send(msg);
            System.out.println("发送时间:"+new SimpleDateFormat("mm:ss").format(new Date()));
            System.out.println("发送结果:"+sendResult);
        }
        producer.shutdown();
    }
}
定义DelayConsumer类
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * 延时消息消费者
 */
public class DelayConsumer {
    public static void main(String[] args) throws MQClientException {
        // 定义一个push消费者
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
        // 指定NameServer
        consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定从第一条消息开始消费
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        // 指定消费Topic与Tag
        consumer.subscribe("TopicB", "*");
        // 注册消息监听器
        // 一旦Broker中有其订阅的消息就会触发该方法的执行,
        // 其返回值为当前consumer消费的状态
        consumer.registerMessageListener((MessageListenerConcurrently) (list, consumeConcurrentlyContext) -> {
            // 逐条消费消息
            for (MessageExt msg : list) {
                // 输出消息被消费的时间
                System.out.println("消费时间:"+new SimpleDateFormat("mm:ss").format(new Date()));
                System.out.println("消息信息:"+msg);
            }
            // 返回消费状态:消费成功
            return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
        });
        // 开启消费者消费
        consumer.start();
        System.out.println("consumer started");
    }
}

事务消息

问题引入

​ 这里有一个需求场景是:工行用户A向建行用户B转账1万元。

​ 我们可以使用同步消息来处理该需求场景:

image-20230515164954560

  • 1、工行系统发送一个给B增宽一万元的同步消息M给Broker
  • 2、消息被Broker成功接收后,向工行系统发送成功ACK
  • 3、工行系统收到成功ACK后从用户A中扣款1万元
  • 4、建行系统从Broker中获取到消息M
  • 5、建行系统消费消息M,即向用户B中增加1万元

这其中是有问题的:若第3步中的扣款操作失败,单消息已经成功发送到了Broker。对于MQ来说,只要消息写入成功,那么这个消息就可以被消费。此时建行系统中用户B增加了1万元。出现了数据不一致的问题。

解决思路

​ 解决思路是,然第1,2,3步具有原子性,要不全部成功,要么全部失败。即消息发送成功后,必须保证扣款成功。如果扣款失败,则回滚发送成功的消息。而该思路即使用事务消息。这里要使用分布式事务解决方案。

image-20230515165448253

​ 使用事务消息来处理该需求场景:

  • 1、事务管理器TM向事务协调器TC发起指令,开启全局事务
  • 2、工行系统发一个给B增款1万元的事务消息M给TC
  • 3、TC想Broker发送半事务消息prepareHalf,将消息M预提交到Broker。此时建行系统是看不到Broker中的消息M的。
  • 4、Broker会将预提交执行结果Report给TC
  • 5、如果预提交失败,则TC会向TM上报预提交失败的响应,全局事务结束;如果预提交成功,TC会调用工行系统的回调操作,去完成工行用户A的预扣款1万元的操作
  • 6、工行系统会向TC发送预扣款执行结果,即本地事务的执行状态
  • 7、TC收到预扣款执行结果后,会将结果上报给TM

预扣款执行结果存在三种可能性:

// 描述本地事务执行状态
public enum LocalTransactionState{
 COMMIT_MESSAGE,// 本地事务提交成功
 ROLLBACK_MESSAGE,// 本地事务执行失败
 UNKNOW,// 不确定,表示需要进行回查以确定本地事务的执行结果
}
  • 8、TM会根据上报结果向TC发出同步的确认执行

    • 若预扣款成功(本地事务状态为COMMIT_MESSAGE),则TM向TC发送Global Commit指令
    • 若预扣款失败(本地事务状态为ROLLBACK_MESSAGE),则TM向TC发送Global Rollback指令
    • 若出现未知状态(本地事务状态为UNKNOW),则会触发工行系统的本地事务状态回查操作。回查操作会将回查结果,即COMMIT_MESSAGE或ROLLBACK_MESSAGE上报给TC。TC将结果上报给TM,TM会再向TC发送最终确认指令Global Commit或Global Rollback
  • 9、TC在接收到指令后会向Broker与工行系统发出确认指令

    • TC接收的若是Global Commit指令,则向Bbroker与工行系统发送Branch Commit指令。此时Broker中的消息M才可被建行系统看到;此时工行用户A的扣款操作才真正被确认
    • TC接收到的若是Global Rollback指令,则向Broker与工行系统发送Bbranch Rollback指令。此时Broker中的消息M将被撤销;工行用户A中的扣款操作将被回滚。

    以上方案就是为了确保消息投递扣款操作能够在一个事务中,要成功都成功,有一个失败,则全部回滚。

    以上方案并不是一个典型的XA模式。因为XA模式中的分支事务是异步的,而事务消息方案中的消息预提交与预扣款操作间是同步的。

基础

分布式事务

​ 对于分布式事务,通俗的说就是,一次操作由若干分支操作组成,这些分支操作分属不同应用,分布在不同服务器上。分布式事务需要保证这些分支操作要么全部成功,要么全部失败。分布式事务与普通事务一样,就是为了保证操作结果的一致性。

事务消息

​ RocketMQ提供了类似X/Open XA的分布式事务功能,通过事务消息能达到分布式事务的最终一致。XA是一种分布式事务解决方案,一种分布式事务处理模式。

半事务消息

​ 暂不能投递的消息,发送方已经成功的将消息发送到了Broker,但是Broker未收到最终确认指令,此时该消息被标记成“暂不能投递”状态,即不能被消费者看到。处于该种状态下的消息即半事务消息。

本地事务状态

​ Producer回调操作执行的结果为本地事务状态,其会发送给TC,而TC会再发送给TM。TM会根据TC发送来的本地事务状态来决定全局事务确认指令。

// 描述本地事务执行状态
public enum LocalTransactionState{
    COMMIT_MESSAGE,// 本地事务提交成功
    ROLLBACK_MESSAGE,// 本地事务执行失败
    UNKNOW,// 不确定,表示需要进行回查以确定本地事务的执行结果
}
消息回查

image-20230515172017814

​ 消息回查,即重新查询本地事务的执行状态。本例中就是重新到DB中查看预扣款操作是否执行成功。

注意:消息回查不是重新执行回调操作。回调操作是进行预扣款操作,而消息回查则是查看预扣款操作执行的结果。

引发消息回查的消息有两个:

  • 1):回调操作返回UNKNOWN
  • 2):TC没有接收到TM的最终全局事务指令
RocketMQ中的消息回查设置

​ 关于消息回查,有三个常见的属性设置。它们都在Broker加载的配置文件中设置,例如:

  • transactionTimeout=20,指定TM在20秒内应将最终确认状态发送给TC,否则引发消息回查。默认为60秒
  • transactionCheckMax=5,指定最多回查5次,超过后将丢弃消息并记录错误日志。默认15次。
  • transactionCheckInterval=10,指定设置的多次消息回查的时间间隔为10秒,默认为60秒

XA模式三剑客

XA协议

​ XA(Unix Transaction)是一种分布式事务解决方案,一种分布式事务处理模式,是基于XA协议的。XA协议由Tuxedo(Transaction for Unix has been Extended for Distributed Operation,分布式操作扩展之后的Unix事务系统)首先提出的,并交给X/Open组织,作为资源管理器与事务管理器的接口标准。

​ XA模式中有三个重要组件:TC、TM、RM。

TC

​ Transaction Coordinator,事务协调者。维护全局和分支事务的状态,驱动全局事务提交或回滚。

RocketMQ中Broker充当着TC。

TM

​ Transaction Manager,事务管理器。定义全局事务的范围:开始全局事务、提交或回滚全局事务。它实际是全局事务的发起者。

RocketMQ中事务消息的Producer充当着TM。

RM

​ Resource Manager,资源管理器。管理分支事务处理的资源,与TC交谈以注册分支事务和报告分支事务的状态,并驱动分支事务提交或回滚。

RocketMQ中事务消息的Producer及Broker均是RM。

XA模式架构

image-20230516164049395

​ XA模式是一个典型的2PC,其执行原理如下:

  • 1、TM向TC发起指令,开启一个全局事务
  • 2、根据业务要求,各个RM会逐个向TC注册分支事务,然后TC会逐个向RM发出预知行指令。
  • 3、各个RM在接收到指令后会进行本地事务预执行。
  • 4、RM将预执行结果Report给TC。当然,这个结果可能是成功,也可能是失败
  • 5、TC在接收到各个RM的Report后会将汇总结果上报给TM,根据汇总结果TM会向TC发出确认指令。
    • 若所有结果都是成功响应,则向TC发送Global Commit指令
    • 只要有结果是失败响应,则向TC发送Global Rollback指令
  • 6、TC在接收到指令后再次向RM发送确认指令。

事务消息方案并不是一个典型的XA模式。因为XA模式中的分支事务是异步的,而事务消息方案中的消息预提交和预扣款操作间是同步的。

注意

  • 事务消息不支持延时消息
  • 对于事务消息要做好幂等性检查,因为事务消息可能不止一次被消费(因为存在回滚后在提交的情况)

示例代码

定义事务监听器
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.client.producer.LocalTransactionState;
import org.apache.rocketmq.client.producer.TransactionListener;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageExt;

/**
 * 工行事务监听器
 */
public class ICBCTransactionListener implements TransactionListener {
    // 回调操作方法
    // 消息预提交成功就会出发该方法的执行,用于完成本地事务
    @Override
    public LocalTransactionState executeLocalTransaction(Message message, Object o) {
        System.out.println("预提交消息成功:" + message);
        // 假设接收到 TAGA的消息就表示扣款成功,TAGB的消息标识扣款失败
        // TAGC的消息表示扣款结果不清楚,需要执行消息回查
        if (StringUtils.equals("TAGA", message.getTags())) {
            return LocalTransactionState.COMMIT_MESSAGE;
        } else if (StringUtils.equals("TAGB", message.getTags())) {
            return LocalTransactionState.ROLLBACK_MESSAGE;
        } else if (StringUtils.equals("TAGC", message.getTags())) {
            return LocalTransactionState.UNKNOW;
        }
        return LocalTransactionState.UNKNOW;
    }

    // 消息回查操作
    // 引发消息回查的原因最常见的有两个:
    // 1) 回调操作返回UNKNOW
    // 2) TC没有接收到TM的最终全局事务确认指令
    @Override
    public LocalTransactionState checkLocalTransaction(MessageExt messageExt) {
        System.out.println("执行消息回查" + messageExt.getTags());
        return LocalTransactionState.COMMIT_MESSAGE;
    }
}

定义事务生产者
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.client.producer.TransactionMQProducer;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;
import tech.msop.test.rocketmq.transaction.listener.ICBCTransactionListener;

import java.util.concurrent.*;

public class TransactionProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        TransactionMQProducer producer = new TransactionMQProducer("tpg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置发送超时时限为20s,默认3s
        producer.setSendMsgTimeout(20000);
        /**
         * 定义一个线程池
         *
         * @param corePoolSize 线程池中核心线程数量
         * @param maximumPoolSize 线程池中最多线程数
         * @param keepAliveTime 这是一个时间。当线程池中线程数量大于核心线程数量时,多余空闲线程的存货时长
         * @param unit 时间单位
         * @param workQueue 临时存放任务的队列,其参数就是队列的长度
         * @param threadFactory 线程工厂
         */
        ExecutorService executorService = new ThreadPoolExecutor(2, 5, 100, TimeUnit.SECONDS, new ArrayBlockingQueue<>(200), 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.setTransactionListener(new ICBCTransactionListener());
        // 开启Producer
        producer.start();
        String[] tags = {"TAGA","TAGB","TAGC"};
        for (int i = 0; i < 3; i++) {
            byte[] body = ("Hi Transaction Msg "+i).getBytes();
            Message msg = new Message("TTopic",tags[i],body);
            // 发送事务消息
            // 第二个参数用于指定在执行本地事务时要使用的业务参数
            SendResult sendResult = producer.sendMessageInTransaction(msg,null);
            System.out.println("发送结果:"+sendResult);
        }
        producer.shutdown();
    }
}

定义事务消费者

同普通消息的消费者

import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyContext;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.util.List;

/**
 * 事务消息消费者
 */
public class TransactionConsumer {
    public static void main(String[] args) throws MQClientException {
        // 定义一个push消费者
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
        // 指定NameServer
        consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定从第一条消息开始消费
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        // 指定消费Topic与Tag
        consumer.subscribe("TTopic","*");
        // 注册消息监听器
        consumer.registerMessageListener(new MessageListenerConcurrently() {
            // 一旦Broker中有其订阅的消息就会触发该方法的执行,
            // 其返回值为当前consumer消费的状态
            @Override
            public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list, ConsumeConcurrentlyContext consumeConcurrentlyContext) {
                // 逐条消费消息
                for (MessageExt msg:list){
                    System.out.println(msg);
                }
                // 返回消费状态:消费成功
                return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
            }
        });
        // 开启消费者消费
        consumer.start();
        System.out.println("consumer started");
    }
}

批量消息

批量发送消息

发送限制

​ 生产者进行消息发送时可以一次发送多条消息,这可以大大提交Producer的发送效率。不过需要注意以下几点:

  • 批量发送的消息必须具有相同的Topic
  • 批量发送的消息必须具有相同的刷盘策略
  • 批量发送的消息不能是延时消息与事务消息
批量发送大小

​ 默认情况下,一批发送的消息总大小不能超过4MB字节。如果想超出该值,有两种解决方案:

  • 方案一:将批量消息进行拆分,拆分为若干不大于4M的消息集合分多次批量发送
  • 方案二:在Producer端与Broker端修改属性
    • Producer端需要在发送之前设置Producer的maxMessageSize属性
    • Broker端需要修改其加载的配置文件中的maxMessageSize属性
生产者发送的消息大小

image-20230516165022073

​ 生产者通过send()方法发送的Message,并不是直接讲Message序列化后发送到网络上的,而是通过这个Message生成了一个字符串发送出去的。这个字符串由四部分构成:Topic、消息Body、消息日志(占20字节),以及用于描述消息的一堆属性key-value。这些属性中包含例如生产者地址、生产时间、要发送的QueueId等。最终写入到Broker中消息单元中的数据都是来自于这些属性。

批量消费消息

修改批量属性

image-20230516165441809

​ Consumer的MessageListenerCurrently监听接口的consumerMessage()方法的第一个参数为消息列表,但默认情况下每次只能消费一条消息。若要使其一次可以消费多条消息,则可以通过修改Consumer的consumeMessageBatchMaxSize属性来指定。不过,该值不能超过32.因为默认情况下消费者每次可以拉取的消息最多32条。若要修改一次拉取的最大值,则可通过修改Consumer的pullBatchSize属性来指定。

存在的问题

​ Consumer的pullBatchSize属性与consumerMessageBatchMaxSize属性是否设置的越大越好?当然不是。

  • pullBatchSize值设置的越大,Consumer每拉取一次需要的时间就越长,且在网络上传输出现问题的可能性就越高。若在拉取过程中出现了问题,那么本批次所有消息都需要全部重新拉取。
  • consumerMessageBatchMaxSize值设置的越大,Consumer的消息并发消费能力越低,且这批被消费的消息具有相同的消费结果。因为consumerMessageBatchMaxSize指定的一批消息只会使用一个线程处理,且在处理过程中只要有一个消息处理异常,则这批消息需要全部重新再次消费处理。

示例代码

定义消息列表分割器
import org.apache.rocketmq.common.message.Message;

import java.util.Iterator;
import java.util.List;
import java.util.Map;

/**
 * 消息列表分割器:其只会处理每条消息大小不超过4M的情况。
 * 若存在某条消息,其本身大小大于4M,这个分割器无法处理,
 * 其直接将这条消息构成一个子列表返回。并没有进行分割
 */
public class MessageListSplitter implements Iterator<List<Message>> {
    // 指定极限值为4M
    private final int SIZE_LIMIT = 4 * 1024 * 1024;
    // 存放所有要发送的消息
    private final List<Message> messages;
    // 要进行批量发送消息的小集合起始索引
    private int currIndex;

    public MessageListSplitter(List<Message> messages) {
        this.messages = messages;
    }

    @Override
    public boolean hasNext() {
        // 判断当前开始遍历的消息索引要小于消息总数
        return currIndex < messages.size();
    }

    @Override
    public List<Message> next() {
        int nextIndex = currIndex;
        // 记录当前要发送的这一小批次消息列表的大小
        int totalSize = 0;
        for (; nextIndex < messages.size(); nextIndex++) {
            // 获取当前遍历的消息
            Message message = messages.get(nextIndex);
            // 统计当前遍历的message的大小
            int tmpSize = message.getTopic().length() + message.getBody().length;
            Map<String,String> properties = message.getProperties();
            for (Map.Entry<String,String> entry:properties.entrySet()){
                tmpSize += entry.getKey().length() + entry.getValue().length();
            }
            tmpSize = tmpSize + 20;

            // 判断当前消息本身是否大于4M
            if (tmpSize > SIZE_LIMIT){
                if (nextIndex - currIndex == 0){
                    nextIndex ++;
                }
                break;
            }
            if (tmpSize + totalSize > SIZE_LIMIT){
                break;
            }else {
                totalSize += tmpSize;
            }
        }
        // 获取当前messages列表的子集合[currentIndex,nextIndex)
        List<Message> subList = messages.subList(currIndex,nextIndex);
        // 下次遍历的开始索引
        currIndex = nextIndex;
        return subList;
    }
}

定义批量消息生产者
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.batch.spliter.MessageListSplitter;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.util.ArrayList;
import java.util.List;

/**
 * 批量消息生产者
 */
public class BatchProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置发送超时时限为20s,默认3s
        producer.setSendMsgTimeout(20000);
        // 指定要发送的消息的最大大小,默认是4M
        // 不过,仅修改该属性是不行的,还需要同时修改broker加载的配置文件中的maxMessageSize属性
//        producer.setMaxMessageSize(8 * 1024 * 1024);
        // 开启Producer
        producer.start();
        // 定义要发送的消息集合
        List<Message> messages = new ArrayList<>();
        for (int i = 0; i < 100; i++) {
            byte[] body = ("Hi, Ordered Msg " + i).getBytes();
            Message msg = new Message("TopicA", "TagA", body);
            messages.add(msg);
        }
        // 定义消息列表分割器,将消息列表分割为多个不超出4M大小的小列表
        MessageListSplitter splitter = new MessageListSplitter(messages);
        while (splitter.hasNext()) {
            try {
                List<Message> listItem = splitter.next();
                producer.send(listItem);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        producer.shutdown();
    }
}

定义批量消息消费者
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * 批量消息消费者
 */
public class BatchConsumer {
    public static void main(String[] args) throws MQClientException {
        // 定义一个push消费者
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
        // 指定NameServer
        consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定从第一条消息开始消费
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        // 指定消费Topic与Tag
        consumer.subscribe("TopicA", "*");
        // 指定每次可以消费10条消息,默认为1
        consumer.setConsumeMessageBatchMaxSize(10);
        // 指定每次可以从Broker拉取40条消息,默认为32
        consumer.setPullBatchSize(40);
        // 注册消息监听器
        // 一旦Broker中有其订阅的消息就会触发该方法的执行,
        // 其返回值为当前consumer消费的状态
        consumer.registerMessageListener((MessageListenerConcurrently) (list, consumeConcurrentlyContext) -> {
            // 逐条消费消息
            for (MessageExt msg : list) {
                // 输出消息被消费的时间
                System.out.println("消费时间:"+new SimpleDateFormat("mm:ss").format(new Date()));
                System.out.println("消息信息:"+msg);
            }
            // 返回消费状态:消费成功
            return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
            // 消费异常的返回结果
//            return ConsumeConcurrentlyStatus.RECONSUME_LATER;
        });
        // 开启消费者消费
        consumer.start();
        System.out.println("consumer started");
    }
}

消息过滤

Tag过滤

​ 通过consumer的subscribe()方法指定要订阅消息的Tag。如果订阅多个Tag的消息,Tag间使用或运算符(双竖线||)连接。

DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("CID_EXAMPLE");
consumer.subscribt("TOPIC","TAGA || TAGB || TAGC");

SQL过滤

​ SQL过滤是一种通过特定表达式对事先买入消息中的用户属性进行筛选过滤的方式。通过SQL过滤,可以实现对消息的复杂过滤。不过,只有使用PUSH模式的消费者才能使用SQL过滤。

​ SQL过滤表达式中支持多种常量类型与运算符。

​ 支持的常量类型:

  • 数值:比如:123,1.32
  • 字符:必须用单引号包裹起来,比如:‘abc’
  • 布尔:TRUE或FALSE
  • NULL:特殊的常量,表示空

​ 支持的运算符有:

  • 数值比较:>、>=、<、<=、BETWEEEN、=
  • 字符比较:=、<>、IN
  • 逻辑运算:AND、OR、NOT
  • NULL判断:IS NULL 或者 IS NOT NULL

​ 默认情况下Broker没有开启消息的SQL过滤功能,需要在Broker加载的配置文件中添加如下属性,以开启该功能:

enablePropertyFilter=true

​ 在启动Broker时需要指定这个修改过的配置文件。例如对于单机Broker的启动,其修改的配置文件是conf/broker.conf,启动时使用如下命令:

$ sh bin/mqbroker -n localhost:9876 -c conf/broker.conf &

示例代码

定义Tag过滤Producer
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

/**
 * Tag过滤Producer
 */
public class FilterByTagProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置发送超时时限为20s,默认3s
        producer.setSendMsgTimeout(20000);
        // 开启Producer
        producer.start();
        String[] tags = {"myTagA", "myTagB", "myTagC"};
        for (int i = 0; i < 10; i++) {
            byte[] body = ("Hi Filter Tag Msg " + i).getBytes();
            String tag = tags[i % tags.length];
            Message msg = new Message("TopicTest", tag, body);
            SendResult sendResult = producer.send(msg);
            System.out.println("发送结果:" + sendResult);
        }
        producer.shutdown();
    }
}
定义Tag过滤Consumer
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * Tag过滤消费者
 */
public class FilterByTagConsumer {
    public static void main(String[] args) throws MQClientException {
        // 定义一个push消费者
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
        // 指定NameServer
        consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定从第一条消息开始消费
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        // 指定消费Topic与Tag
        consumer.subscribe("TopicTest", "myTagA || myTagB");
        // 注册消息监听器
        // 一旦Broker中有其订阅的消息就会触发该方法的执行,
        // 其返回值为当前consumer消费的状态
        consumer.registerMessageListener((MessageListenerConcurrently) (list, consumeConcurrentlyContext) -> {
            // 逐条消费消息
            for (MessageExt msg : list) {
                // 输出消息被消费的时间
                System.out.println("消费时间:"+new SimpleDateFormat("mm:ss").format(new Date()));
                System.out.println("消息信息:"+msg);
            }
            // 返回消费状态:消费成功
            return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
        });
        // 开启消费者消费
        consumer.start();
        System.out.println("consumer started");
    }
}

定义SQL过滤Producer
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

/**
 * SQL过滤Producer
 */
public class FilterBySQLProducer {
    public static void main(String[] args) throws Exception {
        // 创建一个Producer,参数为Producer Group名称
        DefaultMQProducer producer = new DefaultMQProducer("pg");
        // 指定NameServer地址
        producer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 设置发送超时时限为20s,默认3s
        producer.setSendMsgTimeout(20000);
        // 开启Producer
        producer.start();
        for (int i = 0; i < 10; i++) {
            byte[] body = ("Hi Filter SQL Msg " + i).getBytes();
            Message msg = new Message("TopicTest", "myTag", body);
            msg.putUserProperty("age", String.valueOf(i));
            SendResult sendResult = producer.send(msg);
            System.out.println("发送结果:" + sendResult);
        }
        producer.shutdown();
    }
}
定义SQL过滤Consumer
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.MessageSelector;
import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.common.message.MessageExt;
import tech.msop.test.rocketmq.constants.RocketMQConstants;

import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * SQL过滤消费者
 */
public class FilterBySQLConsumer {
    public static void main(String[] args) throws MQClientException {
        // 定义一个push消费者
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
        // 指定NameServer
        consumer.setNamesrvAddr(RocketMQConstants.NAMESRV_ADDR);
        // 指定从第一条消息开始消费
        consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_FIRST_OFFSET);
        // 指定消费Topic与Tag
        consumer.subscribe("TopicTest", MessageSelector.bySql("age between 0 and 6"));
        // 注册消息监听器
        // 一旦Broker中有其订阅的消息就会触发该方法的执行,
        // 其返回值为当前consumer消费的状态
        consumer.registerMessageListener((MessageListenerConcurrently) (list, consumeConcurrentlyContext) -> {
            // 逐条消费消息
            for (MessageExt msg : list) {
                // 输出消息被消费的时间
                System.out.println("消费时间:"+new SimpleDateFormat("mm:ss").format(new Date()));
                System.out.println("消息信息:"+msg);
            }
            // 返回消费状态:消费成功
            return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;
        });
        // 开启消费者消费
        consumer.start();
        System.out.println("consumer started");
    }
}

消息发送重试机制

说明

​ Producer对发送失败的消息进行重新发送的机制,称为消息发送重试机制,也称为消息重投机制。

​ 对于消息重投,需要注意以下几点:

  • 生产者在发送消息时,若采用同步或异步发送方式,发送失败会重试,但oneway消息发送方式发送失败是没有重试机制的
  • 只会有普通消息具有发送重试机制的,顺序消息是没有的
  • 消息重投机制可以保证消息尽可能成功、不丢失,但可能会造成消息重复。消息重复在RocketMQ中是无法避免的问题
  • 消息重复在一般情况下不会发生,当出现消息量大、网络抖动,消息重复就会成为大概率事件
  • producer主动重发、consumer负载变化(发生Rebalance,不会导致消息重复,但可能出现重复消费)也会导致重复消息
  • 避免消息重复消费的解决方案是,为消息添加唯一标识(例如消息key),使消费者对消息进行消费判断来避免重复消费
  • 消息发送重试有三种策略可以选择:同步发送失败策略、异步发送失败策略、消息刷盘失败策略

同步发送失败策略

​ 对于普通消息,消息发送默认采用round-robin策略来选择所发送到的队列。如果发送失败,默认重试2次。但在重试时是不会选择上次发送失败的Broker,而是选择其他Broker。当然,若只有一个Broker其也只能发送到该Broker,但其会尽量发送到该Broker的其他Queue。

// 创建一个producer,参数为Producer Group名称
DefaultMQProducer producer = new DefaultMQProducer("pg");
// 指定NameServer地址
producer.setNamesrvAddr("rocketmqos:9876");
// 设置同步发送失败时重试发送的次数,默认为2次
producer.setRetryTimesWhenSendFailed(3);
// 设置发送超时时限为5秒,默认3秒
producer.setSendMsgTimeout(5000);

​ 同时,Broker还具有失败隔离功能,使Producer尽量选择未发生过发送失败的Broker作为目标Broker。其可以保证其他消息尽量不发送到问题Broker,为了提升消息发送效率,降低消息发送耗时。

思考:让我们自己实现失败隔离功能,应该怎么做?

  • 方案一:Producer中维护着某JUC的Map集合,其key是发生失败的时间戳,value为Broker示例。Producer中还维护着一个Set集合,其中存放着所有未发生发送异常的Broker实例。选择目标Broker是从该Set集合中选择的。再定义一个定时任务,定期从Map集合中将长期未发生发送异常的Broker清理出去,并添加到Set集合。
  • 方案二:为Producer中的Broker实例添加一个标识,例如是一个AtomicBoolean属性。只要该Broker上发生过发送异常,就将其置为ture。选择目标Broker就是选择该属性值为false的Broker。再定义一个定时任务,定期将Broker的该属性设置为false。
  • 方案三:为Producer中的Broker实例添加一个标识,例如是一个AtomicLong属性。只要该Broker上发生过发送异常,就将该值增一。选择目标Broker就是选择该属性值最小的Broker。若该值相同,采用轮询方式选择。

​ 如果超过重试次数,则抛出异常,由Producer去保证消息不丢。当然生产者出现RemotingException、MQClientException和MQBrokerException时,Producer会自动重投消息。

异步发送失败策略

​ 异步发送失败重试时,异步重试不会选择其他Broker,仅在同一个Broker上做重试,所以该策略无法保证消息不丢失。

// 创建一个producer,参数为Producer Group名称
DefaultMQProducer producer = new DefaultMQProducer("pg");
// 指定NameServer地址
producer.setNamesrvAddr("rocketmqos:9876");
// 指定异步发送失败后不进行重试发送
producer.setRetryTimesWhenSendAsyncFailed(0);

消息刷盘失败策略

​ 消息刷盘超时(Master或Slave)或Slave不可用(Slave在做数据同步时向Master返回状态不是SEND_OK)时,默认是不会将消息尝试发送到其他Broker的。不过,对于重要消息可以通过在Broker的配置文件设置retryAnotherBrokerWhenNotStoreOK属性为true来开启。

消息消费重试机制

顺序消息的消费重试

​ 对于顺序消息,当Consumer消费消息失败后,为了保证消息的顺序性,其会自动不断的进行消费重试,直到消费成功。消费重试默认间隔时间为1000毫秒。重试期间应用会出现消息消费被堵塞的情况。

DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
// 顺序消息消费失败的消费重试时间间隔,单位毫秒,默认为1000,其取值范围为[10,3000]
consumer.setSuspendCurrentQueueTimeMillis(100);

由于对顺序消息的重试是无休止的,不简单的,直至消费成功,所以,对于顺序消息的消费,务必要保证应用能够及时监控并处理消费失败的情况,避免消费被永久阻塞。

注意:顺序消息没有发送失败重试机制,但具有消费失败重试机制。

无序消息的消费重试

​ 对于无序消息(普通消息、延时消息、事务消息),当Consumer消费消息失败时,可以通过设置返回状态达到消息重试的效果。不过需要注意,无序消息的重试只对集群消费方式生效,广播消费方式不提供失败重试特性。即对于广播消费,消费失败后,失败消息不再重试,继续消费后续消息。

消费重试次数与间隔

​ 对于无序消息集群消费下的重试消费,每条消息默认最多重试16次,但每次重试的间隔时间是不同的,会主键变长。每次重试的间隔时间如下:

重试次数与上次重试的间隔时间重试次数与上次重试的间隔时间
110秒97分钟
230秒108分钟
31分钟119分钟
42分钟1210分钟
53分钟1320分钟
64分钟1430分钟
75分钟151小时
86分钟162小时

若一条消息在一直消费失败的前提下,将会在正常消费后第2小时46分进行第16次重试。若仍然失败,则将消息投递到死信队列

修改消费重试次数

DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("cg");
// 修改消费重试次数
consumer.setMaxReconsumeTimes(10);

对于修改过的重试次数,将按照以下策略执行:

  • 若修改值小于16,则按照指定间隔进行重试
  • 若修改值大于16,则超过16次的重试间隔均为2小时

对于Consumer Group,若仅修改了一个Consumer的消费重试次数,则会应用到该Group中所有其他Consumer示例。若出现多个Consumer均做了修改的情况,则采用覆盖方式生效。即最后被修改的值会覆盖前面设置的值。

重试队列

​ 对于需要重试消费的消息,并不是Consumer在等待了指定时长后再次去拉取原来的消息进行消费,而是将这些需要重试消费的消息放入到了一个特殊的Topic队列中,而后进行再次消费的。这个特殊的队列就是重试队列。

​ 当出现需要进行重试消费的消息时,Broker会为每个消费者组都设置一个Topic名称为**%RETRY%consumerGroup@consumerGroup**的重试队列。

1)这个重试队列是针对消息才组的,而不是针对每个Topic设置的(一个Topic的消息可以让多个消费者组进行消息,所以会为这些消费者组个创建一个重试队列)。

2)只有当出现需要进行重试消费的消息时,才会为该消费者组创建重试队列。

image-20230517165838350

注意:消息重试的时间间隔与延时消息延时等级十分相似,除了没有延时等级的前两个时间外,其他的时间都是等同的。

​ Broker对于重试消息的处理是通过延时消息实现的。先将消息保存到SCHEDULE_TOPIC_XXXXX延迟队列中,延迟时间到后,会将消息投递到%RETRY%consumerGroup@consumerGroup重试队列中。

消费重试配置方式

image-20230517170052693

​ 集群消费方式下,消息消费失败后若希望消费重试,则需要在消息监听器接口的实现中明确进行如下三种方式之一的配置:

  • 方法1:返回ConsumeConcurrentlyStatus.RECONSUME_LATER(推荐)
  • 方法2:返回NULL
  • 方法3:抛出异常

消费不重试配置方式

image-20230517170617380

​ 集群消费方式下,消息消费失败后若不希望消费重试,则在捕获到异常后同样也返回与消费成功后的相同结果,即ConsumeConcurrentlyStatus.CONSUME_SUCCESS,则不进行消费重试。

死信队列

什么是死信队列

​ 当一条消息初次消费失败,消息队列会自动进行消费重试;达到最大重试次数后,若消费依然失败,则表明消费者在正常情况下无法正确地消费该消息,此时,消息队列不会立刻将消息丢弃,而是将其发送到该消费者对应的特殊队列中。这个队列就是死信队列(Dead-Letter Queue,DLQ),而其中的消息则称为死信消息(Dead-Letter Message,DLM)。

死信队列是用来处理无法被正常消费的消息的。

死信队列的特征

​ 死信队列具有如下特征:

  • 死信队列中的消息不会再被消费者正常消息,即DLQ对于消费者是不可见的。
  • 死信存储有效期与正常消息相同,均为3天(commitlog文件的过期时间),3天后会被自动删除
  • 死信队列就是一个特殊的Topic,名称为**%DLQ%consumerGroup@consumerGroup**,即每个消费者组都有一个死信队列
  • 如果一个消费者组未产生死信消息,则不会为其创建相应的死信队列。

死信队列的处理

​ 实际上,当一条消息进入死信队列,就意味着系统中某些地方出现了问题,从而导致消费者无法正常消费该消息,比如代码中原本就存在的bug。因此,对于死信消息,通常需要开发人员进行特殊处理。最关键的步骤是要排查可以因素,解决代码中可能存在的Bug,然后再将原来的死信消息再次进行投递消费。

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。