LinkedBlockingQueue#put操作

JDK提供了7大阻塞队列,常用于实现生产者和消费者,LinkedBlockingQueue是最常用之一.

public void put(E e) throws InterruptedException {
    if (e == null) throw new NullPointerException();
    int c = -1;
    Node<E> node = new Node<E>(e);
    final ReentrantLock putLock = this.putLock;
    final AtomicInteger count = this.count;
    // 可中断的获取锁操作,即线程由于未获取到锁而处于Blocked状态时,线程可以被中断而不再继续等待.
    putLock.lockInterruptibly();
    try {
        // 如果队列中元素数量和队列容量相等,即此时队列已经满了,则当前线程需要等待.
        // 之所以使用while循环,是当线程被唤醒之后,依然需要判断队列是否满.
        // 即只有队列不满的情况下,当前线程才可以添加元素        
        while (count.get() == capacity) {
            notFull.await();
        }
        // 入队,即添加元素
        enqueue(node);
        c = count.getAndIncrement();
        if (c + 1 < capacity)
            // 通知其他生产者
            notFull.signal();
    } finally {
        putLock.unlock();
    }
    if (c == 0)
        // 通知消费者
        signalNotEmpty();
}

总结一下添加操作流程

1.获取putLock锁
2.如果队列已满, 则等待(notFull.await())
3.元素入队
4.当前生产者添加元素之后如果队列还没有满, 则通知其他生产者添加元素(notFull.signal())
5.释放putLock锁
6.如果队列中已经有元素,则通知消费者

首先是获取锁,拿到锁之后,如果队列满了则等待,如果队列不满则添加元素,

添加元素之后,如果队列还没有满,则通知其他生产者

添加元素之后,如果队列已经有元素了,则通知消费者

---

个人站点
语雀

公众号