本文已参与「新人创作礼」活动,一起开启掘金创作之路。
定时任务各种各样,常见的定时任务例如日志备份,我们可能在每天凌晨 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秒后打印
从日志中可以看到消息延迟已经实现了