springcloudalibaba架构(24):RocketMQ延时消息

前言

延时消息。
通常在电商项目中，某些热卖商品如果被下单锁定成功后，为了避免热卖商品被长时间锁定但是不付款，影响商品的售卖，一般都会设置5分钟未付款自动取消订单。而要实现该功能，有两种方案

• 轮询
  后台启动一个线程，定期查看订单是否已支付，到期未支付取消订单，释放库存。
• 延迟队列
  用户下单成功后，秒杀系统发送一个5分钟的延迟消息，当监听到消息队列时，判断是否已支付，如果未支付则取消订单，释放库存。

官网地址:延时消息样例

延迟消息的使用

RocketMQ延时队列的核心思路：所有的延迟消息由Producer发出之后，都会被存放到一个Topic(SCHEDULE_TOPIC_XXXX)下，不同的延迟消息级别会对应不同的队列序号，当到延迟消息时间后，由定时线程读取SCHEDULE_TOPIC_XXXX下下的相应消息，再存到指定的Topic下，此时消息对于Consumer端可见，从而被Consumer消费。

注意: RocketMQ不支持任意时间的延时，只支持一下固定时间的延时等级，延时等级定义在MessageStoreConfig类

// org/apache/rocketmq/store/config/MessageStoreConfig.java
private String messageDelayLevel = "1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h";

官方地址:https://github.com/apache/rocketmq/blob/master/store/src/main/java/org/apache/rocketmq/store/config/MessageStoreConfig.java

延时消息样例

import com.lcz.util.DateUtils;
import lombok.extern.slf4j.Slf4j;
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.producer.DefaultMQProducer;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageExt;
import org.junit.Test;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.List;
import java.util.concurrent.TimeUnit;


@SpringBootTest
@Slf4j
public class RocketMqTest {

    @Test
    public void testProducer() throws Exception {
        // 创建消息生产者，设置生产者组名
        DefaultMQProducer producer = new DefaultMQProducer("myproducergroup");
        // 为生产者设置NameServer
        producer.setNamesrvAddr("127.0.0.1:9876");
        //启动生产者
        producer.start();
        //构建消息
        Message message = new Message("mytopic", "tag222", "hello world".getBytes());
        message.setDelayTimeLevel(3);//等级3标识10秒后才可以消费
        //发送消息
        SendResult result = producer.send(message, 5000);
        System.out.println("时间:" + DateUtils.getNow() + ",发送消息:" + result);
        //关闭生产者
        producer.shutdown();


    }

    @Test
    public void testConsumer() throws Exception {
        DefaultMQPushConsumer consumer = new DefaultMQPushConsumer("myconsumergroup");
        consumer.setNamesrvAddr("127.0.0.1:9876");
        consumer.subscribe("mytopic", "*");
        consumer.registerMessageListener(new MessageListenerConcurrently() {
            @Override
            public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list, ConsumeConcurrentlyContext consumeConcurrentlyContext) {
                System.out.println("时间:" + DateUtils.getNow() + ",收到的消息:" + list);
                return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;//返回状态是消费成功还是消费失败
            }
        });

        consumer.start();
        System.out.println("时间:" + DateUtils.getNow() + ",启动consumer成功");
        TimeUnit.HOURS.sleep(1);
    }
}

运行测试用例，可以看到消息实在10s后被消费。

消息延迟等级

前面的demo直接声明数值设置延时等级，阅读代码时很不直观，将其定义成常量，这样才在使用时，就比较方便直观。

package com.lcz.config;

/**
 * RocketMq消息延迟等级
 */
public class RocketMqMessageDelayLevelConstant {
    // https://github.com/apache/rocketmq/blob/master/store/src/main/java/org/apache/rocketmq/store/config/MessageStoreConfig.java
    //private String messageDelayLevel = "1s 5s 10s 30s 1m 2m 3m 4m 5m 6m 7m 8m 9m 10m 20m 30m 1h 2h";
    public final static int Level_1s=1;
    public final static int Level_5s=2;
    public final static int Level_10s=3;
    public final static int Level_30s=4;
    public final static int Level_1m=5;
    public final static int Level_2m=6;
    public final static int Level_3m=7;
    public final static int Level_4m=8;
    public final static int Level_5m=9;
    public final static int Level_6m=10;
    public final static int Level_7m=11;
    public final static int Level_8m=12;
    public final static int Level_9m=13;
    public final static int Level_10m=14;
    public final static int Level_20m=15;
    public final static int Level_30m=16;
    public final static int Level_1h=17;
    public final static int Level_2h=18;
}