基于Redis的消息持久化定时设置（redis 消息设置时间）

基于Redis的消息持久化定时设置

Redis是一款内存型的键值存储数据库，支持数据持久化，常被用于缓存和消息队列等场景。在应用中，我们常常需要对消息进行定时发送，这时就需要实现一个基于Redis的消息持久化定时设置功能。

实现思路

我们通过Redis的ZSET数据结构来存储消息，其中score表示消息发送的时间戳，value表示消息的内容。通过zadd命令添加消息到ZSET中，然后使用一个定时器定时扫描ZSET，将该发送的消息发送出去。

1. 添加消息

使用zadd命令将消息添加到ZSET中，消息内容使用JSON字符串进行序列化，添加消息代码如下：

import redis
import time
import json
client = redis.Redis(host='localhost', port=6379, db=0)

def add_job(job, delay):
    """添加任务"""
    ts = time.time() + delay
    client.zadd('jobs', {json.dumps(job): ts})
job = {'type': 'eml', 'title': 'Hello', 'content': 'World'}
delay = 10 # 延时10秒发送
add_job(job, delay)

以上代码将一个发送邮件的任务添加到ZSET中，延时10秒发送。

2. 扫描消息

使用一个定时器每隔1秒钟扫描一次ZSET，将需要发送的消息发送出去，代码如下：

import redis
import time
import json
client = redis.Redis(host='localhost', port=6379, db=0)

def scan_jobs():
    """扫描任务"""
    while True:
        ts = time.time()
        msgs = client.zrangebyscore('jobs', 0, ts) # 获取需要发送的消息
        if not msgs:
            time.sleep(1)
            continue

        for msg in msgs:
            client.zrem('jobs', msg) # 从ZSET中删除该消息
            job = json.loads(msg)
            # 发送消息...
            print(job)
scan_jobs()

以上代码通过zrangebyscore命令获取需要发送的消息，然后依次发送。发送完毕后，将该消息从ZSET中删除。

完整代码如下：

import redis
import time
import json
client = redis.Redis(host='localhost', port=6379, db=0)

def add_job(job, delay):
    """添加任务"""
    ts = time.time() + delay
    client.zadd('jobs', {json.dumps(job): ts})
def scan_jobs():
    """扫描任务"""
    while True:
        ts = time.time()
        msgs = client.zrangebyscore('jobs', 0, ts)
        if not msgs:
            time.sleep(1)
            continue

        for msg in msgs:
            client.zrem('jobs', msg)
            job = json.loads(msg)
            # 发送消息...
            print(job)
job = {'type': 'eml', 'title': 'Hello', 'content': 'World'}
delay = 10
add_job(job, delay)

scan_jobs()