package com.brycen.part3.threadpool.collections;
import java.util.Iterator;
import java.util.concurrent.*;
public class DelayQueueExample {
    public static void main(String[] args) throws InterruptedException {
        DelayQueue<DelayElement> delayQueue = new DelayQueue<>();
        delayQueue.add(new DelayElement<String>("test1",1000));
        delayQueue.add(new DelayElement<String>("test2",800));
        delayQueue.add(new DelayElement<String>("test3",2000));
        delayQueue.add(new DelayElement<String>("test4",3000));
        System.out.println("隊(duì)列大?。?+delayQueue.size());
        //立即返回，同時(shí)刪除元素，如果沒(méi)有過(guò)期元素則返回null
        System.out.println("poll方法獲?。?+delayQueue.poll());
        //立即返回，不會(huì)刪除元素，如果沒(méi)有過(guò)期元素也會(huì)返回該元素，只有當(dāng)隊(duì)列為null時(shí)才會(huì)返回null
        System.out.println("peek方法獲取："+delayQueue.peek());
        //進(jìn)行阻塞
        System.out.println("take方法獲?。?+delayQueue.take().getData());
        //立即返回，但不能保證順序
        Iterator<DelayElement> iterator = delayQueue.iterator();
        while (iterator.hasNext()){
            DelayElement element = iterator.next();
            System.out.println("iterator獲?。?+element.getData());
        }
    }
    static class DelayElement<E> implements Delayed{
        private final long expireTime;
        private final E e;
        public DelayElement(E e,long delay) {
            this.expireTime = System.currentTimeMillis()+delay;
            this.e = e;
        }
        @Override
        public long getDelay(TimeUnit unit) {
            //判斷是否過(guò)期
            return unit.convert(expireTime-System.currentTimeMillis(),TimeUnit.MILLISECONDS);
        }
        @Override
        public int compareTo(Delayed delayed) {
            //排序規(guī)則：按照過(guò)期時(shí)間排序
            DelayElement that = (DelayElement)delayed;
            if (this.expireTime<that.expireTime){
                return -1;
            }else if (this.expireTime>that.expireTime){
                return 1;
            }else {
                return 0;
            }
        }
        public E getData() {
            return e;
        }
    }
}