基于消息队列（RabbitMQ）实现延迟任务
原创基于消息队列（RabbitMQ）实现延迟任务
原创
一、序言
延迟任务应用广泛，延迟任务典型应用场景有订单超时自动取消；支付回调重试。其中订单超时取消具有幂等性属性，无需考虑重复消费问题；支付回调重试需要考虑重复消费问题。
延迟任务具有如下特点：在未来的某个时间点执行；一般仅执行一次。
1、实现原理
生产者将带有延迟信息的消息发送到RabbitMQ交换机中，等待延迟时间结束方将消息转发到绑定的队列中，消费者通过监听队列消费消息。延迟任务的关键在消息在交换机中停留。
显而易见，基于RabbitMQ实现延迟任务对服务器的可靠性要求极高，交换机内部消息无持久化机制，比如单机模式服务重启，未开始的延迟任务均丢失。
2、组件选型
二、方案设计
（一）服务器
RabbitMQ服务需要安装x-delayed-message插件以处理延迟消息。
（二）生产者
延迟任务的实现对生产者的要求是将消息可靠的投递到交换机，因此使用confirm确认机制即可。
订单生成之后，先入库，然后以订单ID为key将订单详情存入Redis中（持久化），向RabbitMQ发送异步confirm确定请求。如果收到正常投递返回，则删除Redis中订单ID为key的数据，回收内存，否则以订单ID为key，从Redis中查询出订单数据，重新发送。
（三）消费者
延迟任务的实现对消费者的要求是以信息不丢失的方式消费消息，具体表现在：手动确认消息的消费，防止消息丢失；消费端持续稳定，防止消息堆积；消息消费失败有重试机制。
考虑到订单延迟取消属于幂等性操作，因此无需考虑消息的重复消费问题。
三、SpringBoot实现
实现部分仅贴一部分核心源码，完整项目请访问GitHub。
（一）生产者
考虑到下单是极为重要的操作，因此首先将订单落库、存盘，然后进行后续操作。
for (long i = 1; i <= 10; i++) {
/* 1.模拟生成订单 */
BuOrder order = createOrder(i);
/* 2.订单入库 */
orderService.removeById(order);
orderService.saveOrUpdate(order);
/* 3.将订单存入信息Redis */
RedisUtils.setObject(RabbitTemplateConfig.ORDER_PREFIX + i, order);
/* 4.向RabbitMQ异步投递消息 */
rabbitTemplate.convertAndSend(RabbitmqConfig.DELAY_EXCHANGE_NAME, RabbitmqConfig.DELAY_KEY, order, RabbitUtils.setDelay(30000), RabbitUtils.correlationData(order.getOrderId()));
}生产者可靠投递消息
public void confirm(CorrelationData correlationData, boolean ack, String cause) {
if (correlationData == null) {
return;
}
String key = ORDER_PREFIX + correlationData.getId();
if (ack) {
/* 如果消息投递成功，则删除Redis中订单数据，回收内存 */
RedisUtils.deleteObject(key);
} else {
/* 从Redis中读取订单数据，重新投递 */
BuOrder order = RedisUtils.getObject(key, BuOrder.class);
/* 重新投递消息 */
rabbitTemplate.convertAndSend(RabbitmqConfig.DELAY_EXCHANGE_NAME, RabbitmqConfig.DELAY_KEY, order, RabbitUtils.setDelay(30000), RabbitUtils.correlationData(order.getOrderId()));
}
}（二）消费者
消费者端手动确认，避免消息丢失；失败自动重试。
@RabbitListener(queues = RabbitmqConfig.DELAY_QUEUE_NAME)
public void consumeNode01(Channel channel, Message message, BuOrder order) throws IOException {
if (Objects.equals(0, order.getOrderStatus())) {
/* 修改订单状态，设置为关闭状态 */
orderService.updateById(new BuOrder(order.getOrderId(), -1));
channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
log.info(String.format("消费者节点01消费编号为【%s】的消息", order.getOrderId()));
}
}消费者可靠消费应至少开启两个及以上应用，确保消息队列中不积压消息。
（三）通用工具包
上述代码涉及一个工具类RabbitUtils，存在于如下依赖中，主要封装RabbitMQ极常用的工具方法。
<dependency>
<groupId>xin.altitude.cms</groupId>
<artifactId>ucode-cms-common</artifactId>
<version>1.4.3.1</version>
</dependency>原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。