Redis实现消息队列与延时消息队列

import redis
import time

# 连接Redis
r = redis.Redis(host='localhost', port=6379, db=0)

# 生产者：发送延时消息
def send_delay_message(queue_name, message, delay_seconds):
    timestamp = int(time.time()) + delay_seconds
    queue_name_key = f'{queue_name}:{timestamp}'
    r.set(queue_name_key, message)
    r.zadd('delayqueue', {queue_name_key: timestamp})

# 消费者：消费延时消息
def consume_delay_message(queue_name):
    while True:
        # 获取当前时间的前一秒，确保可以获取到已到期的消息
        now = int(time.time()) - 1
        # 获取到期的消息
        messages = r.zrangebyscore('delayqueue', 0, now)
        for message in messages:
            # 移除延时队列中已经处理过的消息
            r.zrem('delayqueue', message)
            # 处理消息
            print(f'Consuming message: {r.get(message).decode()}')
            r.delete(message)
        time.sleep(5)  # 每5秒检查一次延时消息

# 使用示例
send_delay_message('myqueue', 'Hello, Redis!', 10)  # 10秒后消费
consume_delay_message('myqueue')  # 消费延时消息