生产者与消费者模式,数组阻塞队列(ArrayBlockingQueue)
生产者消费者模式最核心的部分是生产者与消费者之间的特殊容器,而阻塞队列是特殊容器最常见的实现。JDK 中定义了阻塞队列接口 BlockingQueue,JDK 通过该接口为我们提供了很多种阻塞队列的实现,其中包括本节的主角 ArrayBlockingQueue,该类位于 java.util.concurrent.ArrayBlockingQueue.java。该类需要实现的核心方法如下,下面我们详细分析 ArrayBlockingQueue 的实现原理。
从名字可以看出它的存储结构就是一个数组,即基于数组实现了一个 FIFO 的阻塞队列。新元素都插入到队列尾部,于是最先进入的元素在队列头而最后进入的元素在队列尾部。该数组是有界的,所以构造时需要制定数组的大小。此外,该阻塞队列还提供公平和非公平两种模式。
使用例子
在分析 ArrayBlockingQueue 的实现原理之前我们先通过几个例子来了解它的使用。例子一中我们创建了一个 ArrayBlockingQueue 对象,然后通过五个线程去生产数据并 put 到阻塞队列,二主线程则充当消费端,不断将阻塞队列中的数据取出消费。程序输出如下:
主要展示了阻塞队列空间满了产生阻塞,我们创建一个最大长度为 4 的阻塞队列,然后通过五个线程分别产生一个数据 put 到阻塞队列中,但由于阻塞队列最大长度为 4,所以在 put 四个数据后会产生阻塞。
主要展示了阻塞队列为空时产生阻塞,我们创建一个最大长度为 10 的阻塞队列,然后没有生产者产生数据,却又一个消费者调用 take 消费数据。由于阻塞队列为空,所以产生了阻塞。
实现原理
我们先看 ArrayBlockingQueue 类的属性和构造函数。其中 count 变量表示队列的长度大小,takeIndex 和 putIndex 分别表示队列入队和出队的索引,items 是一个 Object 数组用于保存队列的元素,lock 是并发控制锁,notEmpty 和 notFull 分别表示队列非空时和非满时的条件。提供两个构造函数,我们在创建时指定队列的大小,另外一个是公平模式参数,默认是非公平的。锁对象使用的是 ReentrantLock 对象,公平机制也使用的是它的,它的公平机制我们在前面的章节中已经深入分析过了。notEmpty 和 notFull 条件对象通过锁对象的 newCondition 创建。
put 和 take 方法是阻塞队列的入队和出队方法,它们会间接调用 enqueue 和 dequeue 方法。其中 put 方法会检查入队的元素不能为 null,需要先获取锁后才执行 enqueue 方法维护数组,如果数组长度已经达到最大长度则调用 notFull 条件的 await 方法等待。take 方法会先获取锁后才执行 dequeue 方法维护数组,如果数组长度为 0 则调用 notEmpty 条件的 await 方法等待。
offer 和 poll 方法则是支持超时的阻塞队列的入队和出队方法,可以看到主要的区别就在于它们分别调用了 notFull 和 notEmpty 条件的 awaitNanos 方法进行等待。入队时如果超过指定的超时时间则会返回 false,表示入队超时导致失败。而出队时如果超过指定的超时时间则会返回 null,表示出队超时导致失败。
总结
围绕着 JDK 中 BlockingQueue 的其中一种实现,即 ArrayBlockingQueue。它的基本结构是使用数组来保存阻塞队列的元素,数组的长度有限制,并且提供了公平和非公平模式。对它的实现原理进行分析,为了方便我们仅保留了 ArrayBlockingQueue 最核心的源码,只有我们能理解透彻这些内容就已经能掌握它的实现原理了。推荐阅读
版权声明: 本文为 InfoQ 作者【码农架构】的原创文章。
原文链接:【http://xie.infoq.cn/article/08a9380b219dd1ac8188d419e】。文章转载请联系作者。
评论 (1 条评论)