RabbitMQ实现延迟任务详解(Golang版)

5,078 阅读4分钟

介绍

延迟任务在业务中是一个很常见的需求,比如:

  1. 订单下单15分钟之后,用户没有支付,则自动取消订单
  2. 用户做了某些操作,5分钟之后发短信提醒用户

诸如此类的场景比比皆是,一种最常见的实现方式,就是开启一个定时任务,然后一直轮询数据库,这种实现方式在数据量小的时候还好,但是数据量一旦过大,这轮询数据库就会给数据库造成很大的压力,此时全面扫表的实现方式就显得不可靠了。

另外一种实现方式,就是用延迟队列的方式来实现,但是RabbitMQ本身是没有实现延迟队列的,不过可以使用TTL+死信队列的方式来实现延迟队列。

消息的TTL

TTL全称Time To Live,即生存时间。消息的TTL也就是消息的生存时间。在RabbitMQ中设置TTL有两种

  • 第一种是声明队列的时候,在队列的属性中设置TTL,这样该队列中的消息都会有相同的有效期
  • 第二种是发送消息时给消息设置属性,可以为每条消息都设置不同的TTL

如果两者都设置,生存时间取两者最小的那一个。这里我们采用第二种,即为每条消息设置TTL

死信交换机/死信队列

一个消息在满足如下的条件的时候,就会变成“死信”,并且能被投递到死信交换机(Dead-Letter-Exchange),最后进入到死信交换机绑定的队列,也称死信队列(Dead-Letter-Queue)

  • 消息被拒绝而且requeue=false
  • 消息的TTL到了,即消息过期
  • 队列排满了,排在前面的消息会被丢弃或者扔到死信路由上

死信交换机和普通的交换机是没有区别的,只是某一个设置死信交换机的队列中有消息过期了,会自动触发消息的转发,发送到死信交换机中去,再由死信交换机转发到死信队列中。死信队列也是一个普通的队列,并没有什么其它特殊的。

延迟队列的实现

接着来看看TTL+死信交换机是如何实现延迟队列的

上面的流程就是实现延迟队列的思路,比方说15分钟取消订单,那么用户下单之后,消息的TTL设置为15分钟,当消息在Queue1待的时间到了15分钟,那么就会被转发到Dead-Letter-Exchange,从而转发到Dead-Letter-Queue,最后被消费者消费,实现延迟任务。

先在RabbitMQ控制台创建一个名为dlx的交换机,作为死信交换机,并绑定上一个dlxQueue队列,作为Dead-Letter-Queue

// 生产者.go
package main

import (
	"github.com/streadway/amqp"
	"mq/fail"
)

func main() {
	conn, err := amqp.Dial("amqp://123:123@localhost:5672")
	fail.OnError(err)
	defer conn.Close()

	ch, err := conn.Channel()
	fail.OnError(err)
	defer ch.Close()

	args := amqp.Table{"x-dead-letter-exchange": "dlx"} 
	q, err := ch.QueueDeclare("test", true, false, false, false, args) // 声明一个test队列,并设置队列的死信交换机为"dlx"

	body := "hello world1"
	for i := 0; i < 10; i++ {
		err = ch.Publish("", q.Name, false, false, amqp.Publishing{
			Body:       []byte(body),
			Expiration: "5000", // 设置TTL为5秒
		})
		fail.OnError(err)
	}
}

启动生产者,可以看到消息被投递到test队列中

5秒之后,消息被转发到dlxQueue队列中

之后有一个消费者,专门处理这个dlxQueu队列中的消息

// 消费者.go
package main

import (
	"fmt"
	"github.com/streadway/amqp"
	"mq/fail"
)

func main() {
	conn, err := amqp.Dial("amqp://123:123@localhost:5672")
	fail.OnError(err)

	c, err := conn.Channel()
	fail.OnError(err)

	msgs, err := c.Consume("dlxQueue", "", true, false, false, false, nil) //监听dlxQueue队列
	fail.OnError(err)

	for d := range msgs {
		fmt.Printf("收到信息: %s\n", d.Body) // 收到消息,业务处理
	}
}

// 5秒之后,打印
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1
// 收到信息: hello world1

总结

使用TTL+死信交换机实现延迟任务还是非常方便的,除此之外还可以使用相关的插件abbitmq-delayed-message-exchange,来实现延迟队列,也是非常的方便。

Thanks!