RabbitMQ——实现延迟队列

165 阅读4分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

定时任务各种各样,常见的定时任务例如日志备份,我们可能在每天凌晨 3 点去备份,这种固定时间的定时任务我们一般采用 cron 表达式就能轻松的实现,还有一些比较特殊的定时任务,向大家看电影中的定时炸弹,3分钟后爆炸,这种定时任务就不太好用 cron 去描述,因为开始时间不确定,在实际开发中有的时候也会遇到类似的需求,例如:

  • 在电商项目中,当我们下单之后,一般需要 20 分钟之内或者 30 分钟之内付款,否则订单就会进入异常处理逻辑中,被取消,那么进入到异常处理逻辑中,就可以当成是一个延迟队列
  • 我买了一个智能砂锅,可以用来煮粥,上班前把素材都放到锅里,然后设置几点几分开始煮粥,这样下班后就可以喝到香喷喷的粥了,那么这个煮粥的指令也可以看成是一个延迟任务,放到一个延迟队列中,时间到了再执行
  • 公司的会议预定系统,在会议预定成功后,会在会议开始前半小时通知所有预定该会议的用户
  • 安全工单超过 24 小时未处理,则自动拉企业微信群提醒相关责任人
  • 用户下单外卖以后,距离超时时间还有 10 分钟时提醒外卖小哥即将超时
  • ...等等很多场景下都需要延迟队列

整体上来说,在 RabbitMQ 上实现定时任务有两种方式:

  • 利用 RabbitMQ 自带的消息过期和死信队列机制,实现定时任务
  • 使用 RabbitMQ 的 rabbitmq_delayed_message_exchange 插件来实现定时任务,这种方案较简单

两种用法分别来看

1. 用插件

1.1 安装插件

rabbitmq_delayed_message_exchange 插件地址:github.com/rabbitmq/ra…

下载完成后在命令行执行如下命令将下载文件拷贝到 Docker 容器中去

docker cp rabbitmq_delayed_message_exchange-3.9.0.ez some-rabbit:/plugins

这里第一个参数是宿主机上的文件地址,第二个参数是拷贝到容器的位置

进入到 RabbitMQ 容器中

docker exec -it some-rabbit /bin/bash

进入到容器之后,执行如下命令启用插件:

rabbitmq-plugins enable rabbitmq_delayed_message_exchange

1.2 消息收发

创建一个 Spring Boot 项目,引入 Web 和 RabbitMQ 依赖,如下:

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-amqp</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.springframework.amqp</groupId>
        <artifactId>spring-rabbit-test</artifactId>
        <scope>test</scope>
    </dependency>
</dependencies>

application.properties 中配置一下 RabbitMQ 的连接信息,如下:

server.port=8081
# 配置RabbitMQ
spring.rabbitmq.host=ip
# RabbitMQ通信端口
spring.rabbitmq.port=5672
spring.rabbitmq.username=guest
spring.rabbitmq.password=guest
# 虚拟主机 默认为/
spring.rabbitmq.virtual-host=/

提供一个 RabbitMQ 的配置类:

@Configuration
public class RabbitConfig {
    public static final String DELAY_QUEUE_NAME = "delay_queue";
    public static final String EXCHANGE_NAME = "delay_exchange";
    public static final String EXCHANGE_TYPE = "x-delayed-message";

    @Bean
    Queue queue() {
        return new Queue(DELAY_QUEUE_NAME, true, false, false);
    }

    @Bean
    CustomExchange customExchange() {
        Map<String, Object> args = new HashMap<>();
        args.put("x-delayed-type", "direct");
        /*
        五个参数
            交换机名称
            交换机类型,这个地方是固定的
            交换机是否持久化
            如果没有队列绑定到交换机,交换机是否删除
            其他参数
         */
        return new CustomExchange(EXCHANGE_NAME, EXCHANGE_TYPE, true, false,args);
    }

    @Bean
    Binding binding() {
        return BindingBuilder.bind(queue())
                .to(customExchange()).with(DELAY_QUEUE_NAME).noargs();
    }
}

这里使用的交换机是 CustomExchange,这是一个 Spring 中提供的交换机,创建 CustomExchange 时有五个参数,含义分别如下:

  • 交换机名称。
  • 交换机类型,这个地方是固定的。
  • 交换机是否持久化。
  • 如果没有队列绑定到交换机,交换机是否删除。
  • 其他参数。

最后一个 args 参数中,指定了交换机消息分发的类型,这个类型就是大家熟知的 direct、fanout、topic 以及 header 几种,用了哪种类型,将来交换机分发消息就按哪种方式来。

创建一个消息消费者:

@Component
@Slf4j
public class MsgReceiver {
    @RabbitListener(queues = RabbitConfig.DELAY_QUEUE_NAME)
    public void handleMsg(String msg) {
        log.info("handleMsg,{}",msg);
    }
}

测试发送消息的接口如下,在消息头中设置消息的延迟时间:

@RestController
public class HelloController {
    @Autowired
    RabbitTemplate rabbitTemplate;

    @GetMapping("/hello")
    public void hello() throws UnsupportedEncodingException {
        Message msg = MessageBuilder.withBody(("hello rabbitmq_delayed_message_exchange,"+new Date()).getBytes("UTF-8")).setHeader("x-delay", 3000).build();
        rabbitTemplate.convertAndSend(RabbitConfig.EXCHANGE_NAME, RabbitConfig.DELAY_QUEUE_NAME, msg);
    }
}

启动项目,请求hello接口,查看日志,消息发送后3秒后打印

从日志中可以看到消息延迟已经实现了