一、概述
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) {