Java关于DelayQueue做延时消息推送

最近比较闲，看某项目源码时看到有用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 queue;

    public Consumer(DelayQueue 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 queue = new DelayQueue();
        // 添加延时消息,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资源浪费。