使用 DelayQueue做延时消息处理

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