RocketMQ如何保证消息被有序消费

RocketMQ如何保证消息被有序消费

消费者端如何接收有序消息

队列消费的两种模式

并发消费模式

当同一类消息被送入不同队列,且这些消息在处理上并不需要按时序消费时,可以考虑使用并发消费模式。

并发消费模式生产者会将消息轮询发送到不同的队列当中,这些队列会和消费者实例建立多个连接(线程),将消息并发送入到不同的消费者,因为消费者处理速度有快慢,所以并不能保证物流数据会按1~9的顺序依次消费。

并发消费模式处理效率很高,但无法保证有序性。

RocketMQ如何保证消息被有序消费_第1张图片

有序消费模式

有序消息是指生产者在产生数据的时候,根据Hash规则指定让消息放入哪个队列,在消费者消费时会保证不同消费者针对每一个队列只有唯一的连接(线程)用于消费指定队列。

有序消费模式可以保证消息按队列FIFO顺序依次被消费,但因此失去并发性能,有序消费模式只有在业务要求必须按顺序消费的场景下才允许使用。

RocketMQ如何保证消息被有序消费_第2张图片

RocketMQ如何实现有序消息

要实现RocketMQ有序消息需要两点调整:

  • 生产者端要求按id等唯一标识分配消息队列
  • 消费者端采用专用的监听器保证对队列的单线程应用

下面咱们来看一下代码:

生产者端

SequenceMessageProvider核心代码是在向Broker发送消息时附加MessageQueueSelector对象,在实现select方法时指定存放到哪个队列中。

import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.MessageQueueSelector;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageQueue;

import java.nio.charset.StandardCharsets;
import java.util.List;

@Slf4j
public class SequenceMessageProvider {
    public static void main(String[] args) {
        // 前置准备代码
        DefaultMQProducer producer = new DefaultMQProducer("producer-group");
        producer.setNamesrvAddr("127.0.0.1:9876");
        try {
            producer.start();
            // 模拟10笔订单
            for (Integer orderId = 1; orderId <= 10; orderId++) {
                // 每笔订单要发3条消息:(1)创建订单 (2)订单库存扣减 (3)增加积分
                for (int i = 0; i < 3; i++) {
                    String data = "";
                    switch (i % 3) {
                        case 0:
                            data = orderId + "号创建订单";
                            break;
                        case 1:
                            data = orderId + "号订单减少库存";
                            break;
                        case 2:
                            data = orderId + "号订单增加积分";
                            break;
                    }
                    // 创建消息对象 topic="order",tags="order",key=orderId
                    Message message = new Message("order", "order", orderId.toString(), data.getBytes(StandardCharsets.UTF_8));
                    // 发送消息,实现MessageQueueSelector接口
                    SendResult result = producer.send(message, new MessageQueueSelector() {
                        // select方法决定向broker哪一个队列发送消息
                        @Override
                        public MessageQueue select(List<MessageQueue> list, Message msg, Object arg) {
                            int orderId = Integer.parseInt(msg.getKeys());
                            int index = orderId % list.size();
                            MessageQueue messageQueue = list.get(index);
                            log.info("id:{},data:{},queue:{}", orderId, new String(msg.getBody()), messageQueue);
                            return messageQueue;
                        }
                    }, null);
                }
            }
        } catch (Exception exception) {
            exception.printStackTrace();
        } finally {
            try {
                producer.shutdown();
                log.warn("连接已关闭");
            } catch (Exception exception) {
                exception.printStackTrace();
            }
        }
    }
}

消费者端

消费者端最大的变化是registerMessageListener监听器要实例化MessageListenerOrdery对象,用于为每一个队列分配唯一的连接(线程)进行消费。

每一批消息从Broker投递给消费者都会触发consumeMessage()方法实现对消息的消费。

import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeOrderlyContext;
import org.apache.rocketmq.client.consumer.listener.ConsumeOrderlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerOrderly;
import org.apache.rocketmq.common.message.MessageExt;

import java.util.List;

@Slf4j
public class SequenceMessageConsumer {
    public static void main(String[] args) throws Exception {
        // 声明并初始化一个consumer
        // 需要一个consumer group名字作为构造方法的参数
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("consumer-group");
        // 同样也要设置NamesrvAddr地址,须要与提供者的地址列表保持一致
        consumer.setNamesrvAddr("127.0.0.1:9876");
        // 设置consumer所订阅的Top 和 Tag,*代表全部的Tag
        consumer.subscribe("order", "*");
        // 注册消息监听者,消费者端要增加MessageListenerOrderly监听器,用于实现有序队列
        consumer.registerMessageListener(new MessageListenerOrderly() {
            @Override
            public ConsumeOrderlyStatus consumeMessage(List<MessageExt> list, ConsumeOrderlyContext context) {
                // 遍历输出
                list.forEach(msg -> {
                    log.info("{},{},{}", msg.getKeys(), new String(msg.getBody()), context.getMessageQueue());
                });
                return ConsumeOrderlyStatus.SUCCESS;
            }
        });
        // 启动消费者
        consumer.start();
    }
}

如何实现消息全局顺序消费?

只需要再生产者固定将所有消息发往到0号队列即可保证全局有序,这也意味着全局采用单线程消费,执行效率极差。

    @Override
    public MessageQueue select(List<MessageQueue> list, Message msg, Object arg) {
        MessageQueue messageQueue = list.get(0);
        return messageQueue;
    }

有序消费有什么使用限制吗?

有序消费模式只支持集群模式(CLUSTERING),不支持广播模式(BROADCASTING),采用广播模式会无法接收到数据。

        // 设置为集群模式
        consumer.setMessageModel(MessageModel.CLUSTERING); //支持有序消息,默认模式
        consumer.setMessageModel(MessageModel.BROADCASTING); //不支持有序消息

你可能感兴趣的:(rocketmq)