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.如果队列中已经有元素,则通知消费者
首先是获取锁,拿到锁之后,如果队列满了则等待,如果队列不满则添加元素,
添加元素之后,如果队列还没有满,则通知其他生产者
添加元素之后,如果队列已经有元素了,则通知消费者
个人站点: https://www.infuq.com/
公众号: Netty历险记
网友评论