Redis 如何实现延时队列？

延时队列的实现步骤

• 1.使用 Sorted Set 存储消息：将消息存储在 Sorted Set 中，使用消息的预期处理时间（时间戳）作为排序分数。
• 2.定期检查和处理消息：消费者定期检查 Sorted Set，处理那些已经到达处理时间的消息。

生产者代码
生产者将消息推入延时队列。

import time
import redis

def add_message_to_delay_queue(r, queue_name, message, delay):
    score = time.time() + delay
    r.zadd(queue_name, {message: score})
    //使用 Redis 的 ZADD 命令将消息添加到 queue_name 对应的 Sorted Set 中。message 作为键，score 作为消息的处理时间（排序分数）。

r = redis.StrictRedis(host='localhost', port=6379, db=0)

# 添加消息到延时队列，延时 5 秒
add_message_to_delay_queue(r, 'delay_queue', 'message1', 5)
//将消息 message1 添加到 delay_queue 队列中，延迟时间为 5 秒。计算出的 score 是当前时间加上 5 秒，表示消息将在 5 秒后处理。
add_message_to_delay_queue(r, 'delay_queue', 'message2', 10)

消费者代码
消费者从延时队列中获取已经到期的消息并处理它们。

import time
import redis

def get_messages_from_delay_queue(r, queue_name):
    now = time.time()
    messages = r.zrangebyscore(queue_name, 0, now)
    //使用 Redis 的 ZRANGEBYSCORE 命令，从 queue_name 对应的 Sorted Set 中获取分数在 0 到 now 之间的所有消息。这些消息的分数（时间戳）表示它们已经到期，可以处理。
    
    if messages:
        r.zremrangebyscore(queue_name, 0, now)
        //如果有到期的消息，使用 ZREMRANGEBYSCORE 命令将这些消息从 Sorted Set 中删除。
    return messages

r = redis.StrictRedis(host='localhost', port=6379, db=0)

while True:
    messages = get_messages_from_delay_queue(r, 'delay_queue')
    for message in messages:
        print(f'Consumed: {message.decode("utf-8")}')
    
    # 等待一段时间再检查，避免频繁轮询造成资源浪费
    time.sleep(1)