一、认识阻塞队列
我们学习过数据结构的话,应该知道队列这样一种数据结构,他的特点就是陷阱先出(FIFO),那么什么是阻塞队列呢?
我们可以想象下面两种场景
场景一: 向队列里面添加元素的时候,队列满了
场景二: 从队列获取元素, 但是队列为空
按场景一来说,如果队列满了,我们再往里面添加的话,应该添加不进去,那么改怎么处理呢,是报错,还是等待队列有空闲位置(有数据被取出)的时候加到队列里面?
按场景二来说,如果队列为空的时候,我们取数据,直接返回null,还是等待直到队列里面有数据了,返回第一个数据呢
其实针对上述两种场景,普通的队列有一些默认实现,但是在多线程这一章节,基于队列实现了一种阻塞队列。
那什么是阻塞队列呢?从名字可以看出阻塞队列其实也就是队列的一种特殊情况,只不过是添加了阻塞的效果,它主要由以下两个特点:
1)支持阻塞的插入方法:意思是当队列满时,队列会阻塞插入元素的线程,直到队列不满。
2)支持阻塞的移除方法:意思是在队列为空时,获取元素的线程会等待队列变为非空
阻塞队列的学习有助于我们对一些中间件的理解,在生产者-消费者模式中广泛应用
二、阻塞队列示例
我们用一个非常简单的例子来掩饰阻塞队列的阻塞效果,首先我们创建一个队列,然后直接用take()方法取值
public class Demo {
static BlockingQueue<String> blockingQueue = new ArrayBlockingQueue<>(1);
public static void main(String[] args) throws InterruptedException {
blockingQueue.take();
}
}
运行效果,我们从控制台可以看到,程序一直在运行状态未结束,说明被阻塞了,因为队列为空,所以取值的时候就被阻塞了
然后我们再测试一种添加元素被阻塞情况,我们创建一个容量为1的阻塞队列,通过代码块给他添加一个元素,再通过在main方法里面添加元素,代码如下
public class Demo {
static BlockingQueue<String> blockingQueue = new ArrayBlockingQueue<>(1);
static{
try {
blockingQueue.put("存放元素");
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public static void main(String[] args) throws InterruptedException {
blockingQueue.put("测试添加元素");
}
}
会发现效果同样是被阻塞了
然后我们模拟一个生产者-消费者的模型
生产者代码:
public class Product extends Thread{
private ArrayBlockingQueue<String> blockingQueue;
public Product(ArrayBlockingQueue<String> blockingQueue) {
this.blockingQueue = blockingQueue;
}
@Override
public void run() {
try {
System.out.println("添加元素成功");
blockingQueue.put("添加元素");
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
消费者代码
public class Consumer extends Thread{
private ArrayBlockingQueue<String> blockingQueue;
public Consumer(ArrayBlockingQueue<String> blockingQueue) {
this.blockingQueue = blockingQueue;
}
@Override
public void run() {
try {
System.out.println("取值成功-"+blockingQueue.take());
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
测试代码
public class Test {
public static void main( String[] args ) throws InterruptedException {
ArrayBlockingQueue<String> queue = new ArrayBlockingQueue<>(1);
Product product = new Product(queue);
Consumer consumer = new Consumer(queue);
consumer.start(); // 先从队列取值线程
Thread.sleep(1000);
product.start(); // 然后再往队列添加值线程
}
}
执行多次效果都一样,先打印添加元素,再取出元素
添加元素成功
取值成功-添加元素
Process finished with exit code 0
其实在阻塞队列里面提供了很多阻塞方法,我这里只是以put()和take()方法演示的,以下是总结的阻塞队列方法和区别
三、阻塞队列原理-源码分析
我们分析一下阻塞队列的源码,来看一下阻塞队列是怎么实现的,以ArrayBlockingQueue为例,以下代码中,是我从源码中抽取的核心代码
public class ArrayBlockingQueue<E> extends AbstractQueue<E>
implements BlockingQueue<E>, java.io.Serializable {
// 定义一个数组,存储元素
final Object[] items;
// 定义获取元素的指针
int takeIndex;
// 定义存放元素的指针
int putIndex;
// 定义元素数量
int count;
// 定义一个重入锁
final ReentrantLock lock;
// 定义了一个数组元素不能为空的Condition条件
private final Condition notEmpty;
// 定义了一个数组元素不能满了的Condition条件
private final Condition notFull;
// 定义一个构造函数
public ArrayBlockingQueue(int capacity, boolean fair){
if (capacity <= 0)
throw new IllegalArgumentException();
this.items = new Object[capacity];
// 调用构造函数的时候初始化锁和Condition条件
lock = new ReentrantLock(fair);
notEmpty = lock.newCondition();
notFull = lock.newCondition();
}
// 添加元素的方法
public void put(E e) throws InterruptedException{
final ReentrantLock lock = this.lock;
// 因为方法阻塞的时候可能会被中断,所以不是调用lock.lock()方法,lockInterruptibly()方法在线程被中断时会抛出异常
lock.lockInterruptibly();
try {
// 如果当前队列满了,调用notFull条件的await()方法阻塞
if (count == items.length){
notFull.await();
}else {
// 队列不满,添加元素进队列
enqueue(e);
}
} finally {
lock.unlock();
}
}
// 定义一个获取元素的阻塞方法
public E take() throws InterruptedException {
final ReentrantLock lock = this.lock;
// 因为方法阻塞的时候可能会被中断,所以不是调用lock.lock()方法,lockInterruptibly()方法在线程被中断时会抛出异常
lock.lockInterruptibly();
try {
// 如果当前队列元素为空,调用notEmpty条件await()方法阻塞
while (count == 0) {
notEmpty.await();
}
return dequeue(); // 不为空的话,就取出元素
} finally {
lock.unlock();
}
}
// 添加一个元素进队列
private void enqueue(E x) {
final Object[] items = this.items;
// 把元素放在此位置
items[putIndex] = x;
// 存放元素的指针+1
putIndex+=1;
// 判断队列是否已满,如果满了的话,把指针从头开始
if (putIndex == items.length) {
putIndex = 0;
}
// 元素个数+1
count++;
// 如果在第一个元素添加之前,可能有获取过元素,然后被阻塞了,此处调用notEmpty条件的signal()方法唤醒获取元素的方法
notEmpty.signal();
}
// 取出当前队列中的一个元素
private E dequeue() {
final Object[] items = this.items;
// 取出获取指针指向的元素
E x = (E) items[takeIndex];
// 将当前位置元素置为null
items[takeIndex] = null;
// 获取指针+1
++takeIndex;
// 如果获取指针的位置到了最后一个元素,再下一个元素就是从0开始
if (takeIndex == items.length) {
takeIndex = 0;
}
// 取出元素后,数量-1
count--;
// 调用notEmpty条件的signal()方法唤醒队列满时,仍然添加元素时被阻塞的线程
notFull.signal();
return x;
}
}
以上就是阻塞队列put()/take()方法的核心原理,我们可以看到,阻塞队列是基于lock锁和Condition条件实现的,至于其他方法,如offer方法,核心代码一样,无非是临界点(队列满或者队列为空)的时候处理方式不一样而已
public boolean offer(E e) {
checkNotNull(e);
final ReentrantLock lock = this.lock;
lock.lock();
try {
if (count == items.length)
// 此处是返回false,而不是阻塞
return false;
else {
enqueue(e);
return true;
}
} finally {
lock.unlock();
}
}
四、阻塞队列总结
阻塞队列的实现是基于锁和condition实现的,所以掌握阻塞队列,需要掌握这两个只是点
链接一:ReentrantLock锁
链接二:condition线程通信
本文是综合自己的认识和参考各类资料(书本及网上资料)编写,若有侵权请联系作者,所有内容仅代表个人认知观点,如有错误,欢迎校正; 邮箱:1354518382@qq.com 博客地址:https://blog.csdn.net/qq_35576976/