在SpringBoot中使用redis中的zset实现延迟任务

为什么使用zset实现延迟任务

ZSET(有序集合)适合实现延迟任务的原因主要有以下几点:

  1. 排序特性ZSET根据分数(score)自动排序,这使得我们可以将任务的执行时间作为分数,从而能够轻松地获取到即将执行的任务。
  2. 范围查询ZSET支持范围查询,我们可以查询分数在某个区间内的所有元素,这对于获取所有已到期的任务非常方便。
  3. 高效性ZSET的操作通常都是高效的,比如添加、删除和范围查询等操作的时间复杂度通常为O(log(N)),这使得它适合用于需要频繁操作的任务队列。
  4. 原子操作ZSET提供了原子操作,比如添加元素时如果元素已存在则会更新其分数,这有助于避免在并发环境中出现数据一致性问题。
  5. 灵活性ZSET允许同一个集合中存在多个相同分数的元素,这使得我们可以存储多个同一时间到期的任务。

实现步骤

1、添加redis依赖

xml 复制代码
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

2、配置redis

yaml 复制代码
spring.redis.host=localhost
spring.redis.port=6379

3、创建ZSET操作服务

创建一个服务类来封装对ZSET的操作:

java 复制代码
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Service;
import java.util.Set;
@Service
public class DelayTaskService {
    // 定义一个常量作为延迟任务队列的键
    private static final String DELAY_TASK_KEY = "delayTaskQueue";
    // 注入RedisTemplate用于操作Redis
    private final RedisTemplate<String, String> redisTemplate;
    // 通过构造函数注入RedisTemplate
    @Autowired
    public DelayTaskService(RedisTemplate<String, String> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }
    /**
     * 添加一个延迟任务
     * @param taskId 任务ID
     * @param delayMillis 延迟执行的毫秒数
     */
    public void addTask(String taskId, long delayMillis) {
        // 计算任务的执行时间(当前时间 + 延迟时间)
        long executeTime = System.currentTimeMillis() + delayMillis;
        // 将任务添加到ZSET中,分数为执行时间
        redisTemplate.opsForZSet().add(DELAY_TASK_KEY, taskId, executeTime);
    }
    /**
     * 处理到期的任务
     */
    public void processTasks() {
        // 获取当前时间
        long now = System.currentTimeMillis();
        // 查询ZSET中小于或等于当前时间的所有任务
        Set<String> tasks = redisTemplate.opsForZSet().rangeByScore(DELAY_TASK_KEY, 0, now);
        // 如果有任务,则进行处理
        if (tasks != null && !tasks.isEmpty()) {
            for (String taskId : tasks) {
                // 从ZSET中移除任务,如果移除成功(任务存在),则处理任务
                if (redisTemplate.opsForZSet().remove(DELAY_TASK_KEY, taskId) > 0) {
                    handleTask(taskId);
                }
            }
        }
    }
    /**
     * 处理具体的任务
     * @param taskId 任务ID
     */
    private void handleTask(String taskId) {
        // 这里实现任务处理的逻辑
        System.out.println("处理任务:" + taskId);
    }
}

4、定时轮询ZSET

创建一个定时任务来定期检查ZSET中是否有到期的任务:

java 复制代码
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
@Component
@EnableScheduling
public class TaskScheduler {
    // 注入DelayTaskService用于处理任务
    private final DelayTaskService delayTaskService;
    // 通过构造函数注入DelayTaskService
    @Autowired
    public TaskScheduler(DelayTaskService delayTaskService) {
        this.delayTaskService = delayTaskService;
    }
    /**
     * 定时检查并处理任务
     */
    @Scheduled(fixedRate = 1000) // 每1000毫秒执行一次
    public void checkAndProcessTasks() {
        // 调用DelayTaskService的processTasks方法
        delayTaskService.processTasks();
    }
}

DelayTaskService负责添加和处理延迟任务,而TaskScheduler负责定期调用processTasks方法来检查和处理到期的任务。

相关推荐
重庆小透明6 分钟前
力扣刷题记录【1】146.LRU缓存
java·后端·学习·算法·leetcode·缓存
lang2015092811 分钟前
Reactor操作符的共享与复用
java
TTc_22 分钟前
@Transactional事务注解的批量回滚机制
java·事务
博观而约取31 分钟前
Django 数据迁移全解析:makemigrations & migrate 常见错误与解决方案
后端·python·django
码不停蹄的玄黓1 小时前
MySQL分布式ID冲突详解:场景、原因与解决方案
数据库·分布式·mysql·id冲突
wei_shuo1 小时前
飞算 JavaAI 开发助手:深度学习驱动下的 Java 全链路智能开发新范式
java·开发语言·飞算javaai
寻月隐君1 小时前
Rust 异步编程实践:从 Tokio 基础到阻塞任务处理模式
后端·rust·github
GO兔1 小时前
开篇:GORM入门——Go语言的ORM王者
开发语言·后端·golang·go
Sincerelyplz1 小时前
【Temproal】快速了解Temproal的核心概念以及使用
笔记·后端·开源
爱上语文1 小时前
Redis基础(6):SpringDataRedis
数据库·redis·后端