飞道的博客

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

297人阅读  评论(0)

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

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 排序功能,超时阻塞又恰当好处的利用了锁的等待,在已有机制的基础上进行封装.在实际开发中,可以多多实践这一思想,使代码架构具备高复用性.


转载:https://blog.csdn.net/qq_33589510/article/details/105898614
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场