并发容器 ArrayBlockingQueue

 

 

目录

 

BlockingQueue

ArrayBlockingQueue

---

BlockingQueue

在所有的并发容器中，BlockingQueue是最常见的一种。BlockingQueue是一个带阻塞功能的队列，当入队列时，若队列已满，则阻塞调用者；当出队列时，若队列为空，则阻塞调用者。
在Concurrent包中，BlockingQueue是一个接口，有许多个不同的实现类，如图所示。

该接口的定义如下：

public interface BlockingQueue<E> extends Queue<E> {
//...
boolean add(E e);
boolean offer(E e);
void put(E e) throws InterruptedException;

boolean remove(Object o);
E take() throws InterruptedException;
E poll(long timeout, TimeUnit unit) throws InterruptedException;
//...
}

该接口和JDK集合包中的Queue接口是兼容的，同时在其基础上增加了阻塞功能。在这里，入队提供了add(...)、offer(..)、put(...)3个方法，有什么区别呢？从上面的定义可以看到，add(...)和offer(..)的返回值是布尔类型，而put无返回值，还会抛出中断异常，所以add(...)和offer(..)是无阻塞的，也是Queue本身定义的接口，而put(..)是阻塞的。add(...)和offer(..)的区别不大，当队列为满的时候，前者会抛出异常，后者则直接返回false。出队列与之类似，提供了remove()、poll()、take()等方法，remove()是非阻塞式的，take()和poll()是阻塞式的。

 

ArrayBlockingQueue

ArrayBlockingQueue是一个用数组实现的环形队列，在构造方法中，会要求传入数组的容量。

public ArrayBlockingQueue(int capacity) {
this(capacity, false);
}
public ArrayBlockingQueue(int capacity, boolean fair) {
// ...
}
public ArrayBlockingQueue(int capacity, boolean fair, Collection<? extends
E> c) {
this(capacity, fair);
// ...
}

其核心数据结构如下：

public class ArrayBlockingQueue<E> extends AbstractQueue<E> implements
BlockingQueue<E>, java.io.Serializable {
//...
final Object[] items;
// 队头指针 ,拿索引。items index for next take, poll, peek or remove 
int takeIndex;
// 队尾指针，放索引，items index for next put, offer, or add
int putIndex;
// 队列中的元素数量
int count;
// 核心为1个锁外加两个条件
final ReentrantLock lock;
// 不为空的条件
private final Condition notEmpty;
// 没有满的条件
private final Condition notFull;
//...
}

其put/take方法也很简单，如下所示。

put方法：

public void put(E e) throws InterruptedException {
        Objects.requireNonNull(e);
        final ReentrantLock lock = this.lock;
        lock.lockInterruptibly();// 可中断的Lock // 获取锁
        try {
            while (count == items.length)// 元素个数等于数组长度
                notFull.await();  // 若队列满，阻塞，直到队列非满时，被唤醒
            enqueue(e);// 否则，加入队列
        } finally {
            lock.unlock();
        }
    }

private void enqueue(E e) {
        // assert lock.isHeldByCurrentThread();
        // assert lock.getHoldCount() == 1;
        // assert items[putIndex] == null;
        final Object[] items = this.items;
        items[putIndex] = e;
        if (++putIndex == items.length) putIndex = 0;
        count++;
        notEmpty.signal();// 当元素put到队列后，通知非空条件，唤起被阻塞的出队线程
    }

 offer方法：

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

        Objects.requireNonNull(e);
        long nanos = unit.toNanos(timeout);
        final ReentrantLock lock = this.lock;
        lock.lockInterruptibly();
        try {
            while (count == items.length) {
                // 阻塞，直到队列不满
                // 或者超时时间已过，返回false
                if (nanos <= 0L)
                    return false;
                nanos = notFull.awaitNanos(nanos);
            }
            enqueue(e);
            return true;
        } finally {
            lock.unlock();
        }
    }

通过源码分析，我们可以发现下面的规律：

• 阻塞调用方式 put(e)或 offer(e, timeout, unit)
• 阻塞调用时，唤醒条件为超时或者队列非满（因此，要求在出队时，要发起一个唤醒操作）
• 进队成功之后，执行notEmpty.signal()唤起被阻塞的出队线程

PS：在进行某项业务存储操作时，建议采用offer进行添加，可及时获取boolean进行判断，如用put要考虑阻塞情况（队列的出队操作慢于进队操作），资源占用。

 

take方法：

public E take() throws InterruptedException {
        final ReentrantLock lock = this.lock;
        lock.lockInterruptibly();// 获取锁
        try {
            while (count == 0)
                notEmpty.await();// take的时候，如果队列为空，则阻塞
            return dequeue();
        } finally {
            lock.unlock();
        }
    }

 

 private E dequeue() {
        // assert lock.isHeldByCurrentThread();
        // assert lock.getHoldCount() == 1;
        // assert items[takeIndex] != null;
        final Object[] items = this.items;
        @SuppressWarnings("unchecked")
        E e = (E) items[takeIndex];
        items[takeIndex] = null;
        if (++takeIndex == items.length) // 如果拿的索引+1等于数组长度,说明拿到队尾了，
          takeIndex = 0;// 从头在开始拿
        count--;// 元素个数减一
        if (itrs != null)
            itrs.elementDequeued();
        notFull.signal();// take结束，通知非满条件，也可以唤醒被阻塞的入队线程
        return e;
    }

 

ArrayBlockingQueue 队列我们可以在创建线程池时进行使用。

new ThreadPoolExecutor(1, 1, 0L, TimeUnit.MILLISECONDS, new ArrayBlockingQueue<Runnable>(2));