Redis模拟延时队列 实现日程提醒

使用Redis模拟延时队列

实际上通过MQ实现延时队列更加方便，只是在实际业务中种种原因导致最终选择使用redis作为该业务实现的中间件，顺便记录一下。

该业务是用于日程短信提醒，用户添加日程后，就会被放入redis队列中等待被执行发送短信提醒业务。

本文介绍如何使用Redis来实现一个简单的延时任务队列，通过这个示例，可以帮助你理解如何利用Redis的有序集合特性来管理和执行延时任务。

设计思路

Redis有序集合（Sorted Set）可以很好地用来实现延时队列的功能。通过将任务的执行时间作为分数（score）来存入有序集合中，并定期检查集合中小于等于当前时间的任务来触发执行。

代码实现

JedisCluster连接初始化

首先，我们需要初始化JedisCluster连接来与Redis集群进行交互。

private static final String ZSET_KEY = "sms_delayed_tasks";
private JedisCluster jedisCluster;

public void RedisClusterScheduler() {
    Set<HostAndPort> nodes = new HashSet<>();
    //从配置文件中读取redis集群配置
    for (String node : AcpCore.getProp("spring.redis.cluster.nodes").split(",")) {
        String[] hostPort = node.split(":");
        nodes.add(new HostAndPort(hostPort[0], Integer.parseInt(hostPort[1])));
    }

    GenericObjectPoolConfig<Jedis> poolConfig = new GenericObjectPoolConfig<>();
    poolConfig.setMaxTotal(128);
    poolConfig.setMaxIdle(128);
    poolConfig.setMinIdle(16);

    jedisCluster = new JedisCluster(nodes, 2000, 2000, 5, AcpCore.getProp("spring.redis.password"), poolConfig);

    if (!isCalled) {
        isCalled = true;
        startTaskChecker();
    }
}

添加延时任务

我们可以通过指定任务和执行时间来添加延时任务。该方法将执行时间转换为时间戳，并将任务存储在Redis有序集合中。

public void addDelayedTask(String task, String time) {
    long executeTime = convertToTimestamp(time);
    if (executeTime > System.currentTimeMillis() / 1000) {
        jedisCluster.zadd(ZSET_KEY, executeTime, task);
        log.info("添加任务: " + task + " 执行时间: " + executeTime);
    } else {
        log.info("任务时间必须在当前时间之后: " + task);
    }
}

private long convertToTimestamp(String time) {
    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
    try {
        return sdf.parse(time).getTime() / 1000;
    } catch (ParseException e) {
        e.printStackTrace();
        return System.currentTimeMillis() / 1000;
    }
}

检查和执行任务

通过一个定时任务不断检查当前时间之前的任务并执行。

private void startTaskChecker() {
    executorService.submit(() -> {
        while (!Thread.currentThread().isInterrupted()) {
            try {
                checkAndExecuteTasks();
                Thread.sleep(1000);
            } catch (Exception e) {
                log.info(new Date() + "发生异常但不中断，异常是：" + e);
            }
        }
    });
}

private void checkAndExecuteTasks() {
    long currentTime = System.currentTimeMillis() / 1000;
    Set<String> tasks = jedisCluster.zrangeByScore(ZSET_KEY, 0, currentTime);
    
    for (String task : tasks) {
        jedisCluster.zrem(ZSET_KEY, task);
        executeTask(task);
    }
}

执行任务的逻辑

假设任务内容是一个JSON对象，执行逻辑在这里可以是任何操作，比如调用外部服务、发送消息等。

private void executeTask(String taskJson) {
    JSONObject task = JSONObject.parseObject(taskJson);
    // 在此处添加具体的业务逻辑
    log.info("执行任务: " + task);
}

总结

通过Redis的有序集合和简单的定时器，能够实现一个简洁有效的延时任务队列。

当然，这个示例是一个简化的模型，在生产环境中，你需要考虑任务的幂等性、系统崩溃后的恢复策略、任务的优先级等问题。希望本文能为你提供实现延时队列的思路和参考。