Java线程池队列ArrayBlockingQueue的详细原理分析-刘宇
CSDN博客地址:https://blog.csdn.net/liuyu973971883
文章来源:转载,原文地址:https://blog.csdn.net/weixin_41622183/article/details/89220043,感谢这位老哥的辛勤付出,写的非常棒,各位看完别忘了给这位老哥点个赞啊。如有侵权,请联系删除。
一、什么是ArrayBlockingQueue?
- ArrayBlockingQueue是一个阻塞的队列,继承了AbstractBlockingQueue,间接的实现了Queue接口和Collection接口。
- 底层以数组的形式保存数据,所以它是基于数组的阻塞队列。
- ArrayBlockingQueue是有边界值的,在创建ArrayBlockingQueue时就要确定好该队列的大小,一旦创建,该队列大小不可更改。
- 内部的全局锁是使用的ReentrantLock。
二、使用场景
- 先进先出(队首是最旧的元素;队尾是新添加的元素)
- 有界队列(即初始化时指定的容量,就是队列最大的容量,不会出现扩容,容量满,则阻塞进队操作;容量空,则阻塞出队操作)
- 队列不支持空元素
三、ArrayBlockingQueue的类图
四、BlockingQueue讲解
我们先来知道 ArrayBlockingQueue 是 BlockingQueue 的实现类,那我们需要先看看 BlockingQueue 提供了哪些方法。
public interface BlockingQueue<E> extends Queue<E> {
//将对象塞入队列,如果塞入成功返回true, 否则返回false。
boolean add(E e);
//将对象塞入到队列中,如果设置成功返回true, 否则返回false
boolean offer(E e);
//将元素塞入到队列中,如果队列中已经满了,
//则该方法会一直阻塞,直到队列中有多余的空间。
void put(E e) throws InterruptedException;
//将对象塞入队列并设置时间
//如果塞入成功返回 true, 否则返回 false.
boolean offer(E e, long timeout, TimeUnit unit)
throws InterruptedException;
//从队列中取对象,如果队列中没有对象,
//线程会一直阻塞,直到队列中有对象,并且该方法取得了该对象。
E take() throws InterruptedException;
//在给定的时间里,从队列中获取对象,
//时间到了直接调用普通的poll方法,为null则直接返回null。
E poll(long timeout, TimeUnit unit)
throws InterruptedException;
//获取队列中剩余长度。
int remainingCapacity();
//从队列中移除指定的值。
boolean remove(Object o);
//判断队列中包含该对象。
public boolean contains(Object o);
//将队列中对象,全部移除,并加到传入集合中。
int drainTo(Collection<? super E> c);
//指定最多数量限制将队列中对,全部移除,并家到传入的集合中。
int drainTo(Collection<? super E> c, int maxElements);
}
五、ArrayBlockingQueue源码解析
1、参数解释
/** 队列中存放的值 */
final Object[] items;
/** 值的索引,这是取出位置的索引*/
int takeIndex;
/** 值的索引,这是插入位置的索引*/
int putIndex;
/** 队列中有多少个元素 */
int count;
2、构造方法
/**
* capacity 表示数组中最大容量,默认使用非公平锁
*/
public ArrayBlockingQueue(int capacity) {
this(capacity, false);
}
/**
* capacity 表示数组中最大容量
* fair 为 false 时使用非公平锁,true 时使用公平锁
*/
public ArrayBlockingQueue(int capacity, boolean fair) {
if (capacity <= 0)
throw new IllegalArgumentException();
this.items = new Object[capacity];
lock = new ReentrantLock(fair);
notEmpty = lock.newCondition();
notFull = lock.newCondition();
}
/**
* capacity 表示数组中最大容量
* fair 为 false 时使用非公平锁,true 时使用公平锁
* c 初始化时,可以加入将我们有的集合加入该队列中
*/
public ArrayBlockingQueue(int capacity, boolean fair,
Collection<? extends E> c) {
this(capacity, fair);
final ReentrantLock lock = this.lock;
lock.lock(); // Lock only for visibility, not mutual exclusion
try {
int i = 0;
try {
for (E e : c) {
checkNotNull(e); //判空
items[i++] = e;
}
} catch (ArrayIndexOutOfBoundsException ex) {
throw new IllegalArgumentException();
}
count = i;
putIndex = (i == capacity) ? 0 : i;
} finally {
lock.unlock();
}
}
3、核心方法
这里仅介绍会阻塞的 take 和 put 方法,其他方法和前面讲的队列类似。
3.1、put方法
入队,该方法为阻塞方法
public void put(E e) throws InterruptedException {
//判空
checkNotNull(e);
//显示锁
final ReentrantLock lock = this.lock;
//可中断锁
lock.lockInterruptibly();
try {
//判断队列元素是否以及满了,满了就阻塞
while (count == items.length)
notFull.await();
//入队方法
enqueue(e);
} finally {
//释放锁
lock.unlock();
}
}
3.2、入队的具体实现
如下代码中,当插入值时发现这个值是队列最后一个位置,会将插入队列的下一个入队位置的索引设置为 0。这样一做,就变成了环形队列的意思了。
/** 变成环形队列 */
if (++putIndex == items.length)
putIndex = 0;
private void enqueue(E x) {
//拿到当前队列数组
final Object[] items = this.items;
//将值放入队列
items[putIndex] = x;
//判断队列是否以及满了
if (++putIndex == items.length)
//满了就将下一个入队索引设置为 0
putIndex = 0;
//队列中数量 +1
count++;
//唤醒其他阻塞的出队操作
notEmpty.signal();
}
3.3、入队图解
下面我们来构造一个长度为 6 的环形数组,如下图是初始化情况:
刚刚开始构造完毕时,插入索引和取出索引都在同一个位置,数组内值为空。
接着,我们将元素 A 进行入队操作。元素 A 入队后,取出值的索引不会变化,但是入队索引会偏移到下一个位置,如下图所示:
我们继续进行入队操作,直到 putIndex 索引指向最后一个位置,如下图所示:
这时,我们将元素 F 进行入队,我们会看到,putIndex 索引和 takeIndex 索引,又到回了初始时的位置,这就是上面说的环形队列,此时如果再有线程进行入队操作时,线程便会堵塞,直到元素被消费。
3.4、take方法
出队方法,该方法为阻塞方法。
public E take() throws InterruptedException {
final ReentrantLock lock = this.lock;
//可中断锁
lock.lockInterruptibly();
try {
//判断队列是否为空
while (count == 0)
//线程阻塞
notEmpty.await();
//返回出队的值
return dequeue();
} finally {
//释放锁
lock.unlock();
}
}
3.5、出队的具体实现
private E dequeue() {
//取得当前对象
final Object[] items = this.items;
//拿出队列中的值
E x = (E) items[takeIndex];
//将该出队索引指向位置的值置空
items[takeIndex] = null;
//如果出的是最后一个元素
if (++takeIndex == items.length)
//又回到原来的点
takeIndex = 0;
//出队值索引指向下一个位置
count--;
if (itrs != null)
itrs.elementDequeued();
notFull.signal();
return x;
}
3.6、出队图解
初始化情况是队列满的情况。
这时来了一个线程,将队列中第一个元素取走。此时,队列中 putIndex 索引没有变化,而 takeIndex 索引指向了下一个位置。
假设又来一条线程执行取出值操作,takeIdex 索引将会指向下一个位置。
我们继续进行出队操作,直到 takeIndex 索引指向最后一个位置,如下图所示:
再次进行取出时,我们可以看到似乎有一次回到了原点。这就环形队列队列。