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

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

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

在开发过程中日程提醒功能作为一个比较常见的功能出现,例如在一些场景中,用户设置了一个日程提醒之后,需要在指定的时间将提醒发送给用户提醒对应的操作,为了实现这种操作功能,我们可以通过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具有高性能的排序和检索能力,适合实现延时任务队列。但是于此同时需要通过定时任务轮询检查到期时间,所以也是一种性能的消耗。但是在一些简单场景中,通过这种方式来实现日程提醒、订单超时处理等延时触发场景不失为一种合适的选择。

相关推荐

oracle数据导入导出_oracle数据导入导出工具

关于oracle的数据导入导出,这个功能的使用场景,一般是换服务环境,把原先的oracle数据导入到另外一台oracle数据库,或者导出备份使用。只不过oracle的导入导出命令不好记忆,稍稍有点复杂...

继续学习Python中的while true/break语句

上次讲到if语句的用法,大家在微信公众号问了小编很多问题,那么小编在这几种解决一下,1.else和elif是子模块,不能单独使用2.一个if语句中可以包括很多个elif语句,但结尾只能有一个...

python continue和break的区别_python中break语句和continue语句的区别

python中循环语句经常会使用continue和break,那么这2者的区别是?continue是跳出本次循环,进行下一次循环;break是跳出整个循环;例如:...

简单学Python——关键字6——break和continue

Python退出循环,有break语句和continue语句两种实现方式。break语句和continue语句的区别:break语句作用是终止循环。continue语句作用是跳出本轮循环,继续下一次循...

2-1,0基础学Python之 break退出循环、 continue继续循环 多重循

用for循环或者while循环时,如果要在循环体内直接退出循环,可以使用break语句。比如计算1至100的整数和,我们用while来实现:sum=0x=1whileTrue...

Python 中 break 和 continue 傻傻分不清

大家好啊,我是大田。...

python中的流程控制语句:continue、break 和 return使用方法

Python中,continue、break和return是控制流程的关键语句,用于在循环或函数中提前退出或跳过某些操作。它们的用途和区别如下:1.continue(跳过当前循环的剩余部分,进...

L017:continue和break - 教程文案

continue和break在Python中,continue和break是用于控制循环(如for和while)执行流程的关键字,它们的作用如下:1.continue:跳过当前迭代,...

作为前端开发者,你都经历过怎样的面试?

已经裸辞1个月了,最近开始投简历找工作,遇到各种各样的面试,今天分享一下。其实在职的时候也做过面试官,面试官时,感觉自己问的问题很难区分候选人的能力,最好的办法就是看看候选人的github上的代码仓库...

面试被问 const 是否不可变?这样回答才显功底

作为前端开发者,我在学习ES6特性时,总被const的"善变"搞得一头雾水——为什么用const声明的数组还能push元素?为什么基本类型赋值就会报错?直到翻遍MDN文档、对着内存图反...

2023金九银十必看前端面试题!2w字精品!

导文2023金九银十必看前端面试题!金九银十黄金期来了想要跳槽的小伙伴快来看啊CSS1.请解释CSS的盒模型是什么,并描述其组成部分。...

前端面试总结_前端面试题整理

记得当时大二的时候,看到实验室的学长学姐忙于各种春招,有些收获了大厂offer,有些还在苦苦面试,其实那时候的心里还蛮忐忑的,不知道自己大三的时候会是什么样的一个水平,所以从19年的寒假放完,大二下学...

由浅入深,66条JavaScript面试知识点(七)

作者:JakeZhang转发链接:https://juejin.im/post/5ef8377f6fb9a07e693a6061目录...

2024前端面试真题之—VUE篇_前端面试题vue2020及答案

添加图片注释,不超过140字(可选)...

今年最常见的前端面试题,你会做几道?

在面试或招聘前端开发人员时,期望、现实和需求之间总是存在着巨大差距。面试其实是一个交流想法的地方,挑战人们的思考方式,并客观地分析给定的问题。可以通过面试了解人们如何做出决策,了解一个人对技术和解决问...