Java8 - ArrayBlockingQueue源码

一、概述

ArrayBlockingQueue 是一个基于循环数组的有界阻塞队列，一旦创建了队列，那么其容量就不能再改变了，它不允许 null 值的插入。它是通过使用 ReentrantLock 来保证线程安全。

ArrayBlockingQueue 支持对生产者线程和消费者线程进行公平的调度，默认情况下是不保证公平性的。公平性通常会降低吞吐量，但是可以减少了可变性和避免了线程饥饿。

二、循环数组

PS：接触过循环数组这种数据结构的，可以直接跳过这一小节。

通常，队列的实现方式有数组和链表两种方式。对于数组这种实现方式来说，我们可以通过维护一个队尾指针，使得在入队的时候可以在O(1)的时间内完成；但是对于出队操作，在删除队头元素之后，必须将数组中的所有元素都往前移动一个位置，这个操作的复杂度达到了O(n)，效果并不是很好。如下图所示：

为了解决这个问题，我们可以使用另外一种逻辑结构来处理数组中各个位置之间的关系。假设现在我们有一个数组A[1…n]，我们可以把它想象成一个环型结构，即A[n]之后是A[1]，如下图所示：那么我们便可以使用两个指针，分别维护队头和队尾两个位置，使入队和出队操作都可以在O(1)的时间内完成，如下图所示。当然，这个环形结构只是逻辑上的结构，实际的物理结构还是一个普通的数据。

到此，我们对 ArrayBlockingQueue 使用的数据结构有了一定的了解，那么接下来我们就来看看它是怎么实现的。

三、属性

//队列的底层结构
final Object[] items;

//队头指针
int takeIndex;

//队尾指针
int putIndex;

//队列中的元素个数
int count;

final ReentrantLock lock;

//并发时的两种状态
private final Condition notEmpty;
private final Condition notFull;

四、方法

1、基本方法

下面的代码是入队和出队操作的实现，我们可以看到，putIndex 指向的是下一个插入的位置，而 takeIndex 指向的是队头元素的位置，如下图，通过这两个指针，我们就可以在 O(1) 的时间内完成入队和出队操作。

//入队
private void enqueue(E x) {
    // assert lock.getHoldCount() == 1;
    // assert items[putIndex] == null;
    final Object[] items = this.items;
    items[putIndex] = x;
    if (++putIndex == items.length)
        putIndex = 0;
    count++;
    //唤醒
    notEmpty.signal();
}

//出队
private E dequeue() {
    // assert lock.getHoldCount() == 1;
    // assert items[takeIndex] != null;
    final Object[] items = this.items;
    @SuppressWarnings("unchecked")
    E x = (E) items[takeIndex];
    items[takeIndex] = null;
    if (++takeIndex == items.length)
        takeIndex = 0;
    count--;
    if (itrs != null)
        itrs.elementDequeued();
    //唤醒
    notFull.signal();
    return x;
}

除了入队和出队操作之外，基本操作中还包含删除指定下标元素的操作，从代码中我们可以看到，删除指定下标的元素之后，必须把后面的元素都往前移动，所以这个操作的代价是非常高的。

//删除指定下标的元素
void removeAt(final int removeIndex) {