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返回值判断任务是否需要被立刻执行,如果需要的话,就返回任务,如果不需要就会等待这个任务到延迟时间的剩余时间,当时间到了就会将任务返回。