使用 DelayQueue做延时消息处理

DelayQueue是一个基于优先级队列的无界阻塞队列,用于存储延迟执行的任务。任务需要实现Delayed接口来计算延迟时间,并通过compareTo方法进行排序。队列按照延迟时间顺序执行任务,take方法会阻塞直到有任务达到执行时间。示例代码展示了如何创建和使用DelayQueue进行任务调度。
摘要由CSDN通过智能技术生成

DelayQueue是JDK提供的api,是一个延迟队列

DelayQueue泛型参数得实现Delayed接口,Delayed继承了Comparable接口。
在这里插入图片描述

在这里插入图片描述

getDelay方法返回这个任务还剩多久时间可以执行,小于0的时候说明可以这个延迟任务到了执行的时间了。

compareTo这个是对任务排序的,保证最先到延迟时间的任务排到队列的头。

demo

import lombok.Getter;
import lombok.extern.slf4j.Slf4j;

import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;

@Slf4j
@Getter
public class MymisticTask implements Delayed {

    private String task;  //延迟任务的具体的内容

    private Long triggerTime;  //延迟时间,秒为单位

    //将triggerTime转成毫秒
    public MymisticTask(String task, Long triggerTime) {
        this.task = task;
        this.triggerTime = triggerTime * 1000 + System.currentTimeMillis();
    }

    @Override
    public long getDelay(TimeUnit unit) {
        return unit.convert(triggerTime - System.currentTimeMillis() , TimeUnit.MILLISECONDS);
    }

    @Override
    public int compareTo(Delayed o) {
        return this.triggerTime.compareTo(((MymisticTask) o).triggerTime);
    }
}

测试


import lombok.extern.slf4j.Slf4j;
import java.time.LocalTime;
import java.util.concurrent.DelayQueue;

@Slf4j
public class DelayQueueTest {

    public static void main(String[] args) {
        DelayQueue<MymisticTask> delayQueue = new DelayQueue<>();

        new Thread(() -> {
            while (true) {
                try {
                    MymisticTask task = delayQueue.take();
                    log.info("{} 获取到延时任务: {}", LocalTime.now(), task.getTask());
                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
            }

        }).start();

        log.info("{} 提交延时任务", LocalTime.now());


        delayQueue.offer(new MymisticTask("task No:1 => 5L ", 5L));
        delayQueue.offer(new MymisticTask("task No:2 => 8L", 8L));
        delayQueue.offer(new MymisticTask("task No:3 => 3L ", 3L));

    }


}

测试结果

在这里插入图片描述

实现原理

在这里插入图片描述
offer 方法在提交任务的时候,会通过根据compareTo的实现对任务进行排序,将最先需要被执行的任务放到队列头。

take 方法获取任务的时候,会拿到队列头部的元素,也就是队列中最早需要被执行的任务,通过getDelay返回值判断任务是否需要被立刻执行,如果需要的话,就返回任务,如果不需要就会等待这个任务到延迟时间的剩余时间,当时间到了就会将任务返回。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

pywanggui

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值