温馨提示×

java delayqueue怎么实现任务延时

小亿
93
2023-11-09 09:17:07
栏目: 编程语言

要实现任务延时可以使用Java的DelayQueue类。DelayQueue是一个无界阻塞队列,其中的元素只能在其到期时才能从队列中取出。可以使用Delayed接口表示延时任务,该接口包含了一个getDelay(TimeUnit unit)方法,用于获取任务的剩余延时时间。

下面是一个使用DelayQueue实现任务延时的示例代码:

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

// 延时任务类
class DelayedTask implements Delayed {
    private String taskName;
    private long delayTime;

    public DelayedTask(String taskName, long delayTime) {
        this.taskName = taskName;
        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 other) {
        if (this.delayTime < ((DelayedTask) other).delayTime) {
            return -1;
        } else if (this.delayTime > ((DelayedTask) other).delayTime) {
            return 1;
        } else {
            return 0;
        }
    }

    @Override
    public String toString() {
        return taskName;
    }
}

public class DelayQueueExample {
    public static void main(String[] args) {
        DelayQueue<DelayedTask> delayQueue = new DelayQueue<>();

        // 添加延时任务到队列
        delayQueue.offer(new DelayedTask("Task 1", 1000));
        delayQueue.offer(new DelayedTask("Task 2", 2000));
        delayQueue.offer(new DelayedTask("Task 3", 3000));

        // 消费延时任务
        while (!delayQueue.isEmpty()) {
            try {
                DelayedTask task = delayQueue.take();
                System.out.println("Consuming task: " + task);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

在上面的示例代码中,我们先定义了一个DelayedTask类,该类实现了Delayed接口。然后在main()方法中创建一个DelayQueue对象,并添加了三个延时任务到队列中。最后使用take()方法从队列中取出并消费延时任务。注意,take()方法是一个阻塞方法,如果队列为空,则会阻塞等待直到有任务可取。

运行上述代码,可以看到每个任务都会根据其延时时间依次被消费。

0