java 延迟队列

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

class Task implements Delayed {
    private final String name;
    private final long startTime;

    public Task(String name, long delayInMilliseconds) {
        this.name = name;
        this.startTime = System.currentTimeMillis() + delayInMilliseconds;
    }

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

    @Override
    public int compareTo(Delayed o) {
        if (this.getDelay(TimeUnit.MILLISECONDS) < o.getDelay(TimeUnit.MILLISECONDS)) {
            return -1;
        }
        if (this.getDelay(TimeUnit.MILLISECONDS) > o.getDelay(TimeUnit.MILLISECONDS)) {
            return 1;
        }
        return 0;
    }

    @Override
    public String toString() {
        return "Task{name='" + name + "', startTime=" + startTime + '}';
    }
}

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

        // 添加任务到延迟队列
        queue.put(new Task("Task 1", 2000)); // 延迟2秒
        queue.put(new Task("Task 2", 1000)); // 延迟1秒
        queue.put(new Task("Task 3", 5000)); // 延迟5秒

        // 按照延迟时间顺序取出任务
        while (!queue.isEmpty()) {
            Task task = queue.take(); // 只有当任务的延迟时间到了，才会被取出来
            System.out.println("Executing: " + task);
        }
    }
}

解释说明：

1. Task 类：

   • Task 类实现了 Delayed 接口，表示这是一个可以延迟执行的任务。
   • name 是任务的名称。
   • startTime 是任务开始执行的时间戳，等于当前时间加上延迟时间。
   • getDelay 方法返回任务还需要等待的时间。
   • compareTo 方法用于比较两个任务的延迟时间，确保延迟时间短的任务优先被执行。

2. DelayQueueExample 类：

   • 使用 DelayQueue 来存储和管理延迟任务。
   • put 方法将任务添加到队列中，任务会按照其延迟时间自动排序。
   • take 方法会阻塞直到任务的延迟时间到期，然后返回该任务并执行。

通过这个例子，你可以看到如何使用 DelayQueue 来实现一个简单的延迟任务调度系统。