美文网首页JUC并发包
JUC下的阻塞队列-DelayQueue

JUC下的阻塞队列-DelayQueue

作者: 于情于你 | 来源:发表于2021-04-21 16:57 被阅读0次

        DelayQueue是一个无界阻塞队列,可以用做延时处理,延时处理就是可以为队列中元素设定一个过期时间,队列中的元素按照自定义的delay时间进行排序,其中的元素只有在时间到期时才能从被提取。
        队列中的头节点是最接近过期的元素。
        该队列中的元素必须实现Delayed接口,该接口强制实现下列两个方法。

    compareTo(Delayed o) :Delayed接口继承了Comparable接口,因此有了这个方法
    getDelay(TimeUnit unit):这个方法返回到激活日期的剩余时间,时间单位由单位参数指定。Delayed接口的方法
    

        该队列实现了Leader-Follower模式,减少不必要的线程等待.关于Leader-Follower可以去看这篇文章:https://www.cnblogs.com/yeyang/p/12580600.html

    1.当存在多个take线程时,同时只生效一个(队列中有多个已经过期的元素除外,因为take源码中,如果元素过期了,是直接删除并返回元素的),即leader线程
    2.当leader存在时,其它的take线程均为follower,其等待是通过condition实现的
    3.当leader不存在时,当前线程即成为leader,在delay之后,将leader角色释放还原
    4.最后如果队列还有内容,且leader空缺,则调用一次condition的signal,唤醒挂起的take线程,其中之一将成为新的leader

    DelayQueue主体结构:
    public class DelayQueue<E extends Delayed> extends AbstractQueue<E>
       implements BlockingQueue<E> {
    
       private final transient ReentrantLock lock = new ReentrantLock();
       // 主体队列,用作元素的delay时间排序
       private final PriorityQueue<E> q = new PriorityQueue<E>();
    
       // leader线程
       private Thread leader = null;
    
        // 条件等待队列
       private final Condition available = lock.newCondition();
    }
    
    
    DelayQueue构造函数
    public DelayQueue() {}
    
     public DelayQueue(Collection<? extends E> c) {
            this.addAll(c);
    }
    
    入队API:
    public boolean add(E e) {
          return offer(e);
      }
    
    public boolean offer(E e) {
          final ReentrantLock lock = this.lock;
          lock.lock();
          try {
              // 新元素入队
              q.offer(e);
              // 如果入队之后,队头元素就是刚入队的元素。
              // 说明入队之前队列是空的,现在有元素了,唤醒take线程
              if (q.peek() == e) {
                  leader = null;
                  available.signal();
              }
              return true;
          } finally {
              lock.unlock();
          }
      }
    
    public void put(E e) {
          offer(e);
      }
    
    
    出队API:
    poll 删除并且返回队列的头部
      public E poll() {
            final ReentrantLock lock = this.lock;
            lock.lock();
            try {
                // 获取队头元素
                E first = q.peek();
                // 队头是空的,或者队头的过期时间没有到,则返回null。否则出队并删除
                if (first == null || first.getDelay(NANOSECONDS) > 0)
                    return null;
                else
                    return q.poll();
            } finally {
                lock.unlock();
            }
        }
    
    take 返回并删除此队列的头,如有必要,等待直到延迟过期的元素在此队列上可用。
     public E take() throws InterruptedException {
            // 同时只允许一个线程访问
            final ReentrantLock lock = this.lock;
            lock.lockInterruptibly();
            try {
                for (;;) {
                    E first = q.peek();
                    // 队头是空的,take线程条件等待
                    if (first == null)
                        available.await();
                    else {
                         // 若头节点的过期时间到了,直接删除并返回头节点
                        long delay = first.getDelay(NANOSECONDS);
                        if (delay <= 0)
                            return q.poll();
                        first = null;  // don't retain ref while waiting 疑问点
                        // 头节点过期时间没到,且leader线程不是空的,当前take线程等待,让leader执行
                        if (leader != null)
                            available.await();
                        else {
                       // 头节点过期时间没到,且leader线程是空的,让当前线程成为leader线程
                        // 并且等待delay时间,delay到了,元素就可以被取了
                            Thread thisThread = Thread.currentThread();
                            leader = thisThread;
                            try {
                                available.awaitNanos(delay);
                            } finally {
                                // delay过去了,如果leader还是当前线程,那么把leader置空,留给下一任leader
                                if (leader == thisThread)
                                    leader = null;
                            }
                        }
                    }
                }
            } finally {
                // 最后如果没有leader了,并且队列还不是空的,那么唤醒take线程
                if (leader == null && q.peek() != null)
                    available.signal();
                lock.unlock();
            }
        }
    
    peek,返回队列的头部,并且不删除元素
    public E peek() {
            final ReentrantLock lock = this.lock;
            lock.lock();
            try {
                return q.peek();
            } finally {
                lock.unlock();
            }
        }
    
    疑问点解答

    Q:为什么在take方法中,如果队列中有元素,但是过期时间还没到的时候,要写first = null;这一句?
    A:如果不设置first = null,将会引起内存泄露。
        线程A到达,队首元素没有到期,设置leader = 线程A,并且执行available.awaitNanos(delay);等待元素过期。
        这时线程B来了,因为leader != null,则会available.await();阻塞,线程C、D、E同理。
    线程A阻塞完毕了,再次循环,获取列首元素成功,出列。
        这个时候列首元素应该会被回收掉,但是问题是它还被线程B、线程C持有着,所以不会回收,如果线程增多,且队首元素无限期的不能回收,就会造成内存泄漏。

    相关文章

      网友评论

        本文标题:JUC下的阻塞队列-DelayQueue

        本文链接:https://www.haomeiwen.com/subject/fflzlltx.html