Java延时队列的使用与实现

WRAOP • 2025-02-05 13:05 • 编程

一、延时队列介绍

延时队列是一种基于优先级队列PriorityQueue和Delay接口实现的队列。该队列中的元素必须实现Delay接口，表明当前元素需要在指定时间之后才能被处理。队列按照Delay时间进行排序，越早需要处理的元素越排在前面。

二、Delay接口与元素实现

Delay接口中定义了getDelay(TimeUnit unit)方法，该方法返回当前元素距离激活时间还有多少时间。元素实现Delay接口时需要实现该方法。

public interface Delayed extends Comparable<Delayed> {
    long getDelay(TimeUnit unit);
}

元素需要实现该接口，并实现getDelay(TimeUnit unit)方法。一般来说，需要获取当前任务的激活时间与现在时间的差值（单位可以是毫秒、秒等），然后调用convert()方法转换为指定的TimeUnit时间单位。

public class Message implements Delayed {
    private int id;
    private String content;
    private long activeTime;

    public Message(int id, String content, long delayTime) {
        this.id = id;
        this.content = content;
        this.activeTime = System.currentTimeMillis() + delayTime;
    }

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

    @Override
    public int compareTo(Delayed o) {
        Message other = (Message) o;
        return Long.compare(activeTime, other.activeTime);
    }

    // getter and setter methods
}

三、延时队列的实现与使用

Java中提供了DelayQueue类实现延时队列。DelayQueue是线程安全的队列，它实现了BlockingQueue接口，提供了以下方法：

add(E e)：将元素添加到队列中，并处于激活状态。
offer(E e, long timeout, TimeUnit unit)：添加一个元素，并阻塞指定的时间等待队列空间。
take()：移除并返回队列头部的元素，若队列为空则阻塞等待。
poll(long timeout, TimeUnit unit)：移除并返回队列头部的元素，若队列为空则阻塞等待指定时间后返回null。
isEmpty()：判断队列是否为空。
size()：返回队列中的元素个数。

public class DelayQueueDemo {
    public static void main(String[] args) throws InterruptedException {
        DelayQueue<Message> queue = new DelayQueue<>();

        // add messages to the queue
        queue.add(new Message(1, "Hello World!", 5_000)); // delay 5s
        queue.add(new Message(2, "Hi World!", 3_000)); // delay 3s
        queue.add(new Message(3, "Goodbye World!", 7_000)); // delay 7s

        // take and print messages from the queue
        while(!queue.isEmpty()) {
            Message message = queue.take();
            System.out.println(message.getContent());
        }
    }
}

四、延时队列的应用场景

延时队列常用于定时任务、定时器等需要在指定时间后进行操作的场景。例如：

短信/邮件的发送定时处理
缓存失效的自动清除
高并发下的请求限流与熔断
定时扫描并处理过期的数据

五、总结

Java延时队列是一种基于优先级队列PriorityQueue和Delay接口实现的队列。其常用于定时任务、定时器等需要在指定时间后进行操作的场景。可以通过实现Delay接口和Comparator接口来自定义元素，放入DelayQueue中实现自己的业务逻辑。

原创文章，作者：WRAOP，如若转载，请注明出处：https://www.506064.com/n/334823.html