Redis实现优先级队列

Posted 2020-12-30 徐小贱的博客

实现优先级队列

优先级队列是什么？

优先级队列 是不同于先进先出队列的另一种队列。每次从队列中取出的是具有最高优先权的元素。

redis如何实现优先级队列

关键命令：BRPOP（从队列右边弹出）/BLPOP（从队列左边弹出）

BRPOP/BLPOP和RPOP/LPOP命令类似，都是从队列里弹出一个元素，只不过BRPOP在队列没有新元素的时候会阻塞，它接受一个timeout参数，用来表示阻塞的时间，当超过该时间还是未获得新元素则会返回nil，如果为0则表示一直阻塞下去。除此之外它还可以接受多个KEY，如：>BRPOP KEY1,KEY2 TIMEOUT。

• 如果多个KEY只有其中一个获得了新元素，则BLPOP会返回该新元素以及获得该元素的KEY

• 如果多个KEY都获得了新元素，则会返回BLPOP/BRPOP命令后面的KEY从左到右第一个KEY的KEY和VALUE

实际操作

只有一个KEY获得新元素

• 打开两个redis-cli实例A和B
• 在客户端A使用BLPOP获取queue:1,queue:2,queue:3

BLPOP queue:1 queue:2 queue:3 0

• 在客户端B往queue:2里面放一个元素

LPUSH queue:2 task

• 回到客户端A 原本处于阻塞状态，现在输出了刚才添加的元素以及所属的KEY,如下：

1. "queue:2" ----KEY
2. "task" ----最新获取的元素
3. (467.91s) ----阻塞的时长

多个KEY都获得新元素

• 基于之前创建的客户端和队列，在客户端B往queue:2和queue:3中各放入一个新元素

LPUSH queue:2 task1
LPUSH queue:3 task2

• 客户端A中使用BLPOP弹出新获得的元素

BLPOP queue:1 queue:2 queue:3 0

立马输出了刚才添加的元素以及所属的KEY,如下：

1. "queue:2" -----获得新元素的所有KEY中从左到右第一个KEY
2. "task1" -----对应KEY的VALUE

总结

通过BLPOP/BRPOP多个KEY都获得新元素时，只返回获得新元素的所有KEY中从左到右第一个KEY和对应的VALUE的特性，我们可以把想优先执行的任务队列对应的KEY放到最左边就可以实现（粗略）的优先级队列了。之所以说是“粗略”，是因为它的优先级是队列级别，优先级高的队列，里面所有的元素的优先级都比后面的队列元素优先级高；而每个队列内部还是FIFO（先进先出）。