在阿里面试官面前现场手撕DelayQueue源码！

如果不想在世界上虚度一生，那就要学习一辈子。 ——高尔基

0 前言

延迟元素的无边界阻塞队列，在该队列中，仅当元素的延迟到期时才可以使用它. 队首是该 Delayed 元素，其延迟在过去最远过期. 如果没有延迟已经过期，就没有head, poll将返回null. 当元素的getDelay（TimeUnit.NANOSECONDS）方法返回的值小于或等于零时，就会发生过期. 即使未到期的元素无法使用take或poll删除，它们也被视为普通的元素。 例如，size方法返回过期和未过期元素的计数. 此队列不允许空元素. 该类及其迭代器实现集合和迭代器接口的所有可选方法。方法Iterator()中提供的迭代器不能保证以任何特定的顺序遍历DelayQueue中的元素.

此类是Java Collections Framework的成员.

1 继承体系

• 该队列里的元素必须实现Delayed接口才能入队 混合式的接口，用于标记在给定延迟后应作用的对象。此接口的实现还必须定义一个compareTo方法，该方法提供与其getDelay方法一致的顺序.

2 属性

• 锁

• PriorityQueue队列里的元素会根据某些属性排列先后的顺序，这里正好可以利用Delayed接口里的getDelay的返回值来进行排序，delayQueue其实就是在每次往优先级队列中添加元素,然后以元素的delay/过期值作为排序的因素,以此来达到先过期的元素会拍在队首,每次从队列里取出来都是最先要过期的元素

• 指定用于等待队首元素的线程。 Leader-Follower模式的变体用于最大程度地减少不必要的定时等待.当一个线程成为leader时，它仅等待下一个延迟过去，但是其他线程将无限期地等待.leader线程必须在从take（）或poll（…）返回之前向其他线程发出信号，除非其他线程成为过渡期间的leader。.每当队首被具有更早到期时间的元素替换时，leader字段都会被重置为null来无效，并且会发出一些等待线程（但不一定是当前leader）的信号。 因此，等待线程必须准备好在等待时获得并失去leader能力.

• 当更新的元素在队首变得可用或新的线程可能需要成为 leader 时，会发出条件信号

3 构造方法 3.1 无参

• 创建一个新的 DelayQueue，它初始是空的

3.2 有参

• 创建一个DelayQueue，初始包含Delayed实例的给定集合的元素。

4 新增数据

先看看继承自 BlockingQueue 的方法

put

• 将指定的元素插入此延迟队列。 由于队列无界，因此此方法将永远不会阻塞. 可以看到 put 调用的是 offer

DelayQueue#offer

• 将指定的元素插入此延迟队列

执行流程

1.加锁 2.元素添加到优先级队列中 3.检验元素是否为队首，是则设置 leader 为null, 并唤醒一个消费线程 4.解锁

其内部调用的是 PriorityQueue 的 offer 方法

PriorityQueue#offer

将指定的元素插入此优先级队列.

public boolean offer(E e) { // 若元素为 null,抛NPE if (e == null) throw new NullPointerException(); // 修改计数器加一 modCount++; int i = size; // 如果队列大小 > 容量  if (i >= queue.length) // => 扩容 grow(i + 1); size = i + 1; // 若队列空，则当前元素正好处于队首 if (i == 0) queue[0] = e; else // 若队列非空，根据优先级排序 siftUp(i, e); return true; } 

执行流程

1. 元素判空
2. 队列扩容判断
3. 根据元素的 compareTo 方法进行排序，希望最终排序的结果是从小到大的，因为想让队首的都是过期的数据，需要在 compareTo 方法实现.

5 取数据 take

检索并删除此队列的头，如有必要，请等待直到延迟过期的元素在此队列上可用

public E take() throws InterruptedException { final ReentrantLock lock = this.lock; // 获取可中断锁 lock.lockInterruptibly(); try { for (;;) { // 从优先级队列中获取队首 E first = q.peek(); if (first == null) // 队首为 null,说明无元素，当前线程加入等待队列，并阻塞 available.await(); else { // 获取延迟时间 long delay = first.getDelay(NANOSECONDS); if (delay <= 0) // 已到期，获取并删除头部元素 return q.poll(); first = null; // 在等待时不要保留引用 if (leader != null) available.await(); else { Thread thisThread = Thread.currentThread(); leader = thisThread; try { // 线程节点进入等待队列 available.awaitNanos(delay); } finally { if (leader == thisThread) leader = null; } } } } } finally { // 若leader == null且还存在元素，则唤醒一个消费线程 if (leader == null && q.peek() != null) available.signal(); // 解锁 lock.unlock(); } } 

执行流程

1. 加锁
2. 取出优先级队列的队首
3. 若队列为空,阻塞
4. 若队首非空,获得这个元素的delay时间值，如果first的延迟delay时间值为0的话,说明该元素已经到了可以使用的时间,调用poll方法弹出该元素,跳出方法
5. 若first的延迟delay时间值非0,释放元素first的引用,避免内存泄露
6. 循环以上操作，直至return

take 方法是会无限阻塞，直到队头的过期时间到了才会返回. 如果不想无限阻塞，可以尝试 poll 方法，设置超时时间，在超时时间内，队头元素还没有过期的> 话，就会返回 null.

6 解密 leader 元素

leader 是一个Thread元素,表示当前获取到锁的消费者线程.

• 以take代码段为例

若 leader 非 null,说明已有消费者线程获取锁，直接阻塞当前线程.

若 leader 为 null，把当前线程赋给 leader，并等待剩余的到期时间，最后释放 leader. 这里假设有多个消费者线程执行 take 取数据,若没有leader != null判断，这些线程都会无限循环，直到返回第一个元素，这显然很浪费系统资源. 所以 leader 在这里相当于一个线程标识，避免消费者线程的无脑竞争.

• 注意这里因为first是队首的引用，阻塞时会有很多线程同时持有队首引用，可能导致内存溢出,所以需要手动释放.

7 总结

DelayQueue 使用排序和超时机制即实现了延迟队列.充分利用已有的 PriorityQueue 排序功能，超时阻塞又恰当好处的利用了锁的等待，在已有机制的基础上进行封装.在实际开发中，可以多多实践这一思想，使代码架构具备高复用性.