百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT技术 > 正文

如何在Spring Boot通过Redis模拟延时队列实现日程提醒功能?

wptr33 2024-12-27 17:15 18 浏览

在开发过程中日程提醒功能作为一个比较常见的功能出现,例如在一些场景中,用户设置了一个日程提醒之后,需要在指定的时间将提醒发送给用户提醒对应的操作,为了实现这种操作功能,我们可以通过Redis来模拟一个延时队列来实现日程消息提醒功能。

下面我们就来介绍一下如何在Spring Boot中如何结合Redis模拟延时队列实现日程提醒功能。

方案介绍

在Redis中有一个有序集合Sorted Set,我们可以通过这个有序集合来实现延时消息队列,通过Redis的ZADD和ZRANGEBYSCORE命令,将任务按照时间顺序进行存储,然后通过任务检索操作将到期任务梳理出来进行执行。具体操作如下所示。

实现步骤

首先第一步需要添加Redis的POM依赖,如下所示。

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-redis</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter</artifactId>
    </dependency>
</dependencies>

接下来就是利用有序集合实现延时队列操作。我们知道在Redis有序集合中,包含了如下的几个内容,用来表示创建的每个任务操作。

  • Key:集合的名称,比如 schedule:tasks。
  • Score:任务的触发时间戳(Unix 时间)。
  • Value:任务的唯一标识或任务详情。

操作步骤如下所示。

  • 第一步、使用 ZADD 添加任务到队列,Score 为触发时间。
  • 第二步、使用 ZRANGEBYSCORE 获取当前时间前的任务。
  • 第三步、触发任务后,使用 ZREM 删除任务。

完成上面的设计之后,接下来就是需要在Spring Boot的配置文件中添加Redis的连接配置,如下所示。

spring:
  redis:
    host: localhost
    port: 6379
    database: 0

根据上面的设计方案,我们需要创建一个Redis任务对象,如下所示,包含了任务的唯一标识以及内容情况,还有需要执行任务的时间。

@Data
@AllArgsConstructor
@NoArgsConstructor
public class ScheduleTask {
    private String taskId;   // 任务唯一标识
    private String content;  // 提醒内容
    private Long executeTime; // 任务执行时间(Unix 时间戳)
}

实现任务管理逻辑

接下来就是创建Redis延时队列用来实现延迟操作。

@Service
public class TaskService {

    private final String TASK_KEY = "schedule:tasks";

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    // 添加任务
    public void addTask(ScheduleTask task) {
        redisTemplate.opsForZSet().add(TASK_KEY, 
                                       JSON.toJSONString(task), 
                                       task.getExecuteTime());
    }

    // 获取到期任务
    public List<ScheduleTask> fetchDueTasks(long currentTime) {
        Set<String> tasks = redisTemplate.opsForZSet()
                                         .rangeByScore(TASK_KEY, 0, currentTime);
        List<ScheduleTask> result = tasks.stream()
                                         .map(task -> JSON.parseObject(task, ScheduleTask.class))
                                         .collect(Collectors.toList());
        redisTemplate.opsForZSet().removeRangeByScore(TASK_KEY, 0, currentTime);
        return result;
    }
}

定义好队列实现之后,接下,我们就可以创建一个定时任务然后定时检查是否有到期的任务需要执行,如下所示,每秒执行一次,检查队列中是否存在到期的消息需要执行。

@Component
public class TaskScheduler {

    @Autowired
    private TaskService taskService;

    @Scheduled(fixedRate = 1000) // 每秒轮询一次
    public void scheduleTasks() {
        long currentTime = System.currentTimeMillis();
        List<ScheduleTask> dueTasks = taskService.fetchDueTasks(currentTime);

        for (ScheduleTask task : dueTasks) {
            // 模拟发送提醒
            System.out.println("提醒用户:" + task.getContent());
        }
    }
}

测试操作

完成上面的定时任务构建之后,接下来我们可以创建一个RESTFul接口用来创建延时任务并且调用TaskService服务将任务添加到延时队列中。如下所示。

@RestController
@RequestMapping("/tasks")
public class TaskController {

    @Autowired
    private TaskService taskService;

    @PostMapping("/add")
    public ResponseEntity<String> addTask(@RequestBody ScheduleTask task) {
        taskService.addTask(task);
        return ResponseEntity.ok("任务添加成功");
    }
}

总结

到这里,Spring Boot通过Redis实现延时队列操作,通过Redis的Sorted Set具有高性能的排序和检索能力,适合实现延时任务队列。但是于此同时需要通过定时任务轮询检查到期时间,所以也是一种性能的消耗。但是在一些简单场景中,通过这种方式来实现日程提醒、订单超时处理等延时触发场景不失为一种合适的选择。

相关推荐

Python自动化脚本应用与示例(python办公自动化脚本)

Python是编写自动化脚本的绝佳选择,因其语法简洁、库丰富且跨平台兼容性强。以下是Python自动化脚本的常见应用场景及示例,帮助你快速上手:一、常见自动化场景文件与目录操作...

Python文件操作常用库高级应用教程

本文是在前面《Python文件操作常用库使用教程》的基础上,进一步学习Python文件操作库的高级应用。一、高级文件系统监控1.1watchdog库-实时文件系统监控安装与基本使用:...

Python办公自动化系列篇之六:文件系统与操作系统任务

作为高效办公自动化领域的主流编程语言,Python凭借其优雅的语法结构、完善的技术生态及成熟的第三方工具库集合,已成为企业数字化转型过程中提升运营效率的理想选择。该语言在结构化数据处理、自动化文档生成...

14《Python 办公自动化教程》os 模块操作文件与文件夹

在日常工作中,我们经常会和文件、文件夹打交道,比如将服务器上指定目录下文件进行归档,或将爬虫爬取的数据根据时间创建对应的文件夹/文件,如果这些还依靠手动来进行操作,无疑是费时费力的,这时候Pyt...

python中os模块详解(python os.path模块)

os模块是Python标准库中的一个模块,它提供了与操作系统交互的方法。使用os模块可以方便地执行许多常见的系统任务,如文件和目录操作、进程管理、环境变量管理等。下面是os模块中一些常用的函数和方法:...

21-Python-文件操作(python文件的操作步骤)

在Python中,文件操作是非常重要的一部分,它允许我们读取、写入和修改文件。下面将详细讲解Python文件操作的各个方面,并给出相应的示例。1-打开文件...

轻松玩转Python文件操作:移动、删除

哈喽,大家好,我是木头左!Python文件操作基础在处理计算机文件时,经常需要执行如移动和删除等基本操作。Python提供了一些内置的库来帮助完成这些任务,其中最常用的就是os模块和shutil模块。...

Python 初学者练习:删除文件和文件夹

在本教程中,你将学习如何在Python中删除文件和文件夹。使用os.remove()函数删除文件...

引人遐想,用 Python 获取你想要的“某个人”摄像头照片

仅用来学习,希望给你们有提供到学习上的作用。1.安装库需要安装python3.5以上版本,在官网下载即可。然后安装库opencv-python,安装方式为打开终端输入命令行。...

Python如何使用临时文件和目录(python目录下文件)

在某些项目中,有时候会有大量的临时数据,比如各种日志,这时候我们要做数据分析,并把最后的结果储存起来,这些大量的临时数据如果常驻内存,将消耗大量内存资源,我们可以使用临时文件,存储这些临时数据。使用标...

Linux 下海量文件删除方法效率对比,最慢的竟然是 rm

Linux下海量文件删除方法效率对比,本次参赛选手一共6位,分别是:rm、find、findwithdelete、rsync、Python、Perl.首先建立50万个文件$testfor...

Python 开发工程师必会的 5 个系统命令操作库

当我们需要编写自动化脚本、部署工具、监控程序时,熟练操作系统命令几乎是必备技能。今天就来聊聊我在实际项目中高频使用的5个系统命令操作库,这些可都是能让你效率翻倍的"瑞士军刀"。一...

Python常用文件操作库使用详解(python文件操作选项)

Python生态系统提供了丰富的文件操作库,可以处理各种复杂的文件操作需求。本教程将介绍Python中最常用的文件操作库及其实际应用。一、标准库核心模块1.1os模块-操作系统接口主要功能...

11. 文件与IO操作(文件io和网络io)

本章深入探讨Go语言文件处理与IO操作的核心技术,结合高性能实践与安全规范,提供企业级解决方案。11.1文件读写11.1.1基础操作...

Python os模块的20个应用实例(python中 import os模块用法)

在Python中,...