最近比较闲,看某项目源码时看到有用DelayQueue类来做延时的消息推送。
DelayQueue是Delayed元素的一个无界阻塞队列,只有在延迟期满时才能从中提取元素。该队列的头部 是延迟期满后保存时间最长的Delayed元素。如果延迟都还没有期满,则队列没有头部,并且poll将返回null。当一个元素的 getDelay( TimeUnit.NANOSECONDS) 方法返回一个小于等于 0 的值时,将发生到期。即使无法使用 take 或 poll 移除未到期的元素,也不会将这些元素作为正常元素对待。例如,size 方法同时返回到期和未到期元素的计数。此队列不允许使用 null 元素。
消息类,继承Delayed:
package com.study.queue;
import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;
public class Message implements Delayed {
private int id;
private String body;
private long excuteTime;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getBody() {
return body;
}
public void setBody(String body) {
this.body = body;
}
public long getExcuteTime() {
return excuteTime;
}
public void setExcuteTime(long excuteTime) {
this.excuteTime = excuteTime;
}
public Message(int id, String body, long delayTime) {
this.id = id;
this.body = body;
this.excuteTime = TimeUnit.NANOSECONDS.convert(delayTime, TimeUnit.MILLISECONDS) + System.nanoTime();
}
@Override
public long getDelay(TimeUnit unit) {
return unit.convert(this.excuteTime - System.nanoTime(), TimeUnit.NANOSECONDS);
}
@Override
public int compareTo(Delayed o) {
Message msg = (Message) o;
return Long.valueOf(this.excuteTime) > Long.valueOf(msg.excuteTime) ? 1 : (Long.valueOf(this.excuteTime) < Long.valueOf(msg.id) ? -1 : 0);
}
}
消费类:检查消费情况
package com.study.queue;
import java.util.concurrent.DelayQueue;
public class Consumer implements Runnable {
private DelayQueue<Message> queue;
public Consumer(DelayQueue<Message> queue) {
this.queue = queue;
}
@Override
public void run() {
while (true) {
try {
Message take = queue.take();
System.out.println("消费信息:" + take.getExcuteTime() + ":" + take.getBody());
} catch (Exception e) {
e.printStackTrace();
}
}
}
}
主程序:把Message放入延时队列里,打印出过时的Message
package com.study.queue;
import java.util.concurrent.DelayQueue;
public class DelayQueueTest {
public static void main(String[] args) {
// 创建延时队列
DelayQueue<Message> queue = new DelayQueue<Message>();
// 添加延时消息,m1 延时5s
Message m1 = new Message(1, "world", 3000);
// 添加延时消息,m2 延时3s
Message m2 = new Message(2, "hello", 10000);
queue.offer(m2);
queue.offer(m1);
// 启动消费线程
new Thread(new Consumer(queue)).start();
}
}
这里有个需要注意的bug,这个重载方法里需要返回的是纳秒,如传入的是毫秒,则回造成循环过快,造成cpu资源浪费。