java DelayQueue 实现延时队列

原创

mob64ca12df9869 2024-03-03 03:59:27 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12df9869的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java DelayQueue 实现延时队列

延时队列（DelayQueue）是 Java 中的一个特殊队列，用于存储实现了 Delayed 接口的元素。这些元素在被放入队列时都会被设置一个延时时间，只有当延时时间到达后才能从队列中取出。DelayQueue 内部使用优先队列（PriorityQueue）来实现，其中元素根据延时时间进行排序。

Delayed 接口

Delayed 接口是一个泛型接口，它包含两个方法：

long getDelay(TimeUnit unit)：返回对象的剩余延时时间。
int compareTo(Delayed o)：比较两个对象的延时时间。

DelayQueue 示例

下面是一个使用 DelayQueue 实现延时队列的示例代码：

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

class DelayedElement implements Delayed {
    private String data;
    private long delayTime;

    public DelayedElement(String data, long delayTime) {
        this.data = data;
        this.delayTime = System.currentTimeMillis() + delayTime;
    }

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

    @Override
    public int compareTo(Delayed o) {
        return Long.compare(this.delayTime, ((DelayedElement) o).delayTime);
    }

    @Override
    public String toString() {
        return "DelayedElement{data='" + data + '\'' + '}';
    }
}

public class DelayQueueExample {
    public static void main(String[] args) throws InterruptedException {
        DelayQueue<DelayedElement> delayQueue = new DelayQueue<>();
        delayQueue.put(new DelayedElement("Element 1", 2000));
        delayQueue.put(new DelayedElement("Element 2", 4000));

        System.out.println("Elements in queue:");
        delayQueue.forEach(System.out::println);

        System.out.println("Waiting for elements to be ready...");
        while (!delayQueue.isEmpty()) {
            DelayedElement element = delayQueue.take();
            System.out.println("Taken element: " + element);
        }
    }
}

DelayQueue 类图

下面是 DelayQueue 的类图，描述了 DelayQueue 的继承关系：

erDiagram
    Delayed ||--|> DelayedElement : 实现
    Object <|-- DelayedElement : 继承
    Delayed ||--|> DelayQueue : 使用
    Object <|-- DelayQueue : 继承