要实现任务延时可以使用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()
方法是一个阻塞方法,如果队列为空,则会阻塞等待直到有任务可取。
运行上述代码,可以看到每个任务都会根据其延时时间依次被消费。