Java延迟队列 DelayQueue

需求
下单完成,10分钟后,如果没有支付,发送消息,提醒支付。
分析
发消息好解决,这里主要讨论延时的问题,Java里,有个DelayQueue,可以实现延时的功能。

先贴运行结果、为了看到效果,代码里延时的是3s
Sat Feb 16 12:25:09 CST 2019启动,并加入了两条消息
Sat Feb 16 12:25:12 CST 2019 ..... 发送消息id 1
Sat Feb 16 12:25:12 CST 2019 ..... 发送消息id 2

代码:

public class Tt {
    private DelayQueue<OrderInfo> messageQueue = new DelayQueue<OrderInfo>(); //消息队列
    public static void main(String[] args) {
        Tt t = new Tt();
        OrderInfo oi = new OrderInfo("1", new Date().getTime() + 3000); // 加3000,表示延时3秒 
        OrderInfo oi2 = new OrderInfo("2", new Date().getTime() + 3000);
        t.messageQueue.offer(oi); // 当接收到订单后,往队列里放一条消息
        t.messageQueue.offer(oi2);
        System.out.println(new Date().toString() + "启动,并加入了两条消息"); // 模拟接收到了两个订单
        
        // 这里是消费线程,随程序运行启动
        t.handle();
    }
    private void handle() {
        ThreadPoolHolder.getThreadPool().submit(new Runnable() {
            @Override
            public void run() {
                while (true) {
                    try {
                        OrderInfo message = messageQueue.take(); // take方法,有消息(延时时间<0的消息)取消息,没有,阻塞住。
                        // 这里查一遍订单.. 符合需求发消息,不符合不发
                        System.out.println(new Date().toString() + " ..... 发送消息id " + message.getId());
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            }
        });
    }
}

// 消息体,必须实现Delayed接口
public class OrderInfo implements Delayed {

    private String id;

    public OrderInfo(String id, long excuteTime) {
        this.id = id;
        this.excuteTime = excuteTime;
    }

    private long excuteTime;// 延迟时长,这个是必须的属性因为要按照这个判断延时时长。

    public String getId() {
        return id;
    }

    @Override
    public long getDelay(TimeUnit unit) {
        return unit.convert(excuteTime,TimeUnit.NANOSECONDS) - unit.convert(System.currentTimeMillis(),TimeUnit.NANOSECONDS);
    }

    @Override
    public int compareTo(Delayed o) {
        OrderInfo msg = (OrderInfo) o;
        return Integer.valueOf(this.id) > Integer.valueOf(msg.id) ? 1
                : (Integer.valueOf(this.id) < Integer.valueOf(msg.id) ? -1 : 0);
    }
}

你可能感兴趣的:(java)