JUC 源码解读系列--ArrayBlockingQueue 篇

ArrayBlockingQueue 提供的功能与 LinkedBlockingQueue 基本相同，且都是 FIFO 的队列。两者的差别主要在于底层的数据结构不同，本篇的初衷在于熟悉 ArrayBlockingQueue 中的游标的设计。

1. offer()

public boolean offer(E e, long timeout, TimeUnit unit)
    throws InterruptedException {

	// 非空校验
    checkNotNull(e);
    long nanos = unit.toNanos(timeout);
    final ReentrantLock lock = this.lock;
    // 获得锁
    lock.lockInterruptibly();
    try {
    	// 一直阻塞，直至超时或者数组有空间可插入
        while (count == items.length) {
            if (nanos <= 0)
                return false;
            nanos = notFull.awaitNanos(nanos);
        }
        // 元素入队
        enqueue(e);
        return true;
    } finally {
        lock.unlock();
    }
}

enqueue()

private void enqueue(E x) {
    // assert lock.getHoldCount() == 1;
    // assert items[putIndex] == null;
    final Object[] items = this.items;
    items[putIndex] = x;
    // 移动下一次 put 元素的位置，溢出则从头开始
    if (++putIndex == items.length)
        putIndex = 0;
    count++;
    // 唤醒一个阻塞在获取操作的线程
    notEmpty.signal();
}

2. poll()

public E poll(long timeout, TimeUnit unit) throws InterruptedException {
    long nanos = unit.toNanos(timeout);
    final ReentrantLock lock = this.lock;
    // 获得锁
    lock.lockInterruptibly();
    try {
    	// 一直阻塞，直至超时或者数组中有元素可取
        while (count == 0) {
            if (nanos <= 0)
                return null;
            nanos = notEmpty.awaitNanos(nanos);
        }
        // 出队
        return dequeue();
    } finally {
        lock.unlock();
    }
}

dequeue()

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;
    // 移动下一次 take 元素的位置，溢出则从头开始
    if (++takeIndex == items.length)
        takeIndex = 0;
    count--;
    if (itrs != null)
        itrs.elementDequeued();
    notFull.signal();
    return x;
}

3. 总结

ArrayBlockingQueue 中使用两个指针（游标） takeIndex 与 putIndex 来确保 FIFO 的特性。

当获取元素时，如果数组中没有可用元素，会直接返回空（无参的 poll 方法）或者一直阻塞（带超时时间的 poll 方法，take 方法）直至超时或者有元素可用，获取到元素后，将 takeIndex 向后移动，表示下一个获取元素的位置。

当插入元素时，如果数组中没有可用空间，会直接返回失败（add方法，无超时时间的 offer 方法）或者一直阻塞（带超时时间的 offer 方法，put 方法）直至超时或者有空间可插入，元素插入成功后，将 putIndex 向后移动，表示下一个插入元素的位置。