JUC源码解析-阻塞队列-LinkedBlockingQueue与ArrayBlockingQueue

什么是阻塞队列？

阻塞队列（BlockingQueue）是一个支持两个附加操作的队列。这两个附加的操作是：

• 在队列为空时，获取元素的线程会阻塞等待，直到队列变为非空或超时。
• 当队列满时，存储元素的线程会等待队列可用。

阻塞队列常用于生产者和消费者的场景。

阻塞队列提供了四种处理方法:

• 抛出异常：是指当阻塞队列满时候，再往队列里插入元素，会抛出IllegalStateException(“Queue full”)异常。当队列为空时，从队列里获取元素时会抛出NoSuchElementException异常 。
• 返回特殊值：插入方法会返回是否成功，成功则返回true。移除方法，则是从队列里拿出一个元素，如果没有则返回null
• 一直阻塞：当阻塞队列满时，如果生产者线程往队列里put元素，队列会一直阻塞生产者线程，直到拿到数据，或者响应中断退出。当队列空时，消费者线程试图从队列里take元素，队列也会阻塞消费者线程，直到队列可用。
• 超时退出：当阻塞队列满时，队列会阻塞生产者线程一段时间，如果超过一定的时间，生产者线程就会退出。

ArrayBlockingQueue

ArrayBlockingQueue的内部是通过一个可重入锁ReentrantLock和两个Condition条件对象来实现阻塞。

关于 ReentrantLock
关于 Condition

public class ArrayBlockingQueue<E> extends AbstractQueue<E>
        implements BlockingQueue<E>, java.io.Serializable {
   

	// 存储数据的数组
    final Object[] items;

	// 获取数据的索引，用于take，poll，remove等方法
    int takeIndex;

	// 添加数据的索引，用于 put, offer, add等方法
    int putIndex;

	// 元素个数
    int count;

	// 锁
    final ReentrantLock lock;

	// Condition 内有个队连，有别于AQS的同步队列，我们称它为等待队列
	// await 会先构建当前线程的节点放入等待队列，之后阻塞线程
	// signal 一般不会唤醒线程，而是将节点放回AQS的同步队列，等待被唤醒
	
	// notEmpty队列里放的是执行获取操作的线程，它们之前由于数组为空无法执行获取操作而阻塞
	// 执行取操作的线程由于数组为空而被放入notEmpty的等待队列中等待
	// 当执行添加操作的线程执行成功后调用notEmpty.signal将notEmpty队列中的
	// 头节点放回AQS的同步队列，其代表的线程在队列中等待被唤醒
    private final Condition notEmpty;

	// notFull 的队列中放的是执行插入操作的线程，它们之前由于数组满无法执行添加操作而阻塞
    private final Condition notFull;
    
    // 迭代器
    transient Itrs itrs = null;

在上面提到了AQS的同步队列与Condition的等待队列，它们在我的关于 ReentrantLock 和 Condition两篇源码分析文章中有详细介绍。

这里贴张图比那与理解：

第一条队列是AQS的同步队列，下面是Condition的等待队列。

关于 notEmpty 与 notFull 不太好记忆，可以将 notEmpty 与 get 联系在一起；notFull 与 put 联系在一起。

add，offer添加操作

    public boolean add(E e) {
   
        return super.add(e);
    }
    
// 跳到AbstractQueue：
    public boolean add(E e) {
   
        if (offer(e))
            return true;
        else
            throw new IllegalStateException("Queue full");
    }
    
// offer方法在ArrayBlockingQueue的实现
    public boolean offer(E e) {
   
        checkNotNull(e);
        final ReentrantLock lock = this.lock;
        lock.lock();
        try {
   
            if (count == items.length)
                return false;
            else {
   
                enqueue(e);
                return true;
            }
        } finally {
   
            lock.unlock();
        }
    }

    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();
    }

对重点进行下总结：

• 添加操作先获取锁；
• add 方法在数组满时抛异常，offer 则返回 false
• 唤醒一个notEmpty等待队列中的线程