Flink 核心技术与实战 : www.sanzhishu.top/512.html
概述
BlockingQueue 是 java.util.concurrent 包提供的用于解决并发生产者 - 消费者问题的最有用的类。
BlockingQueue 的特性是在任意时刻只有一个线程可以进行 take 或者 put 操作,并且 BlockingQueue 提供了超时 return null 的机制,在许多生产场景里都可以看到这个工具的身影。
BlockingQueue是一个接口,它的实现类有 ArrayBlockingQueue、DelayQueue、 LinkedBlockingDeque、 LinkedBlockingQueue、PriorityBlockingQueue、 SynchronousQueue等,它们的区别主要体现在存储结构上或对元素 操作上的不同,但是对于 take 与 put 操作的原理,却是类似的。
队列类型
- 无限队列(unbounded queue ) - 几乎可以无限增长
- 有限队列(bounded queue ) - 定义了最大容量
队列数据结构
队列实质就是一种存储数据的结构
- 通常用链表或者数组实现
- 一般而言队列具备先进先出的特性,当然也有双端队列(Deque)优先级队列
- 主要操作:入队(EnQueue)与出队(Dequeue)
; BlockingQueue API
继承关系:BlockingQueue extends Queue extends Collection
BlockingQueue 接口的所有方法可以分为两大类:负责向队列添加元素的方法和检索这些元素的方法。在队列满/空的情况下,来自这两个组的每个方法的行为都不同。
- 添加元素 的常用方法:
void put(E e)
boolean add(E e)
boolean offer(E e)
boolean offer(E e, long timeout, TimeUnit unit)
- 检索元素 的常用方法:
E take()
E poll(long timeout, TimeUnit unit)
int drainTo(Collection<? super E> c)
int drainTo(Collection<? super E> c, int maxElements)
- 还可以使用 Collection(集合)接口中的方法,例如:
void clear()
int size()
int remainingCapacity()
常见的阻塞队列
- ArrayBlockingQueue :由数组支持的有界队列
- LinkedBlockingQueue :由链接节点支持的可选有界队列
- PriorityBlockingQueue :由优先级堆支持的无界优先级队列
- DelayQueue :由优先级堆支持的、基于时间的调度队列
ArrayBlockingQueue
介绍
队列基于数组实现,容量大小在创建 ArrayBlockingQueue 对象时指定。底层数组一旦创建了,容量就不能改变了,因此 ArrayBlockingQueue 是一个容量限制的阻塞队列。因此在队列满的时候执行入队会阻塞,在队列为空时出队也会阻塞。
数据结构如下图:
- 队列创建:
BlockingQueue<string> blockingQueue = new ArrayBlockingQueue<>();</string> - **应用场景:**在线程池中有比较多的应用,生产者消费者场景
- **工作原理:**基于 ReentrantLock 保证线程安全,根据 Condition 实现队列满时的阻塞
- 主要变量:
final Object[] items;
int count;
final ReentrantLock lock;
private final Condition notEmpty;
private final Condition notFull;
方法解析
- put(E e) 方法
public void put(E e) throws InterruptedException {
Objects.requireNonNull(e);
final ReentrantLock lock = this.lock;
lock.lockInterruptibly();
try {
while (count == items.length)
notFull.await();
enqueue(e);
} finally {
lock.unlock();
}
}
put 方法总结:
- ArrayBlockingQueue 不允许添加 null 元素;
- ArrayBlockingQueue 在队列已满的时候,会调用 notFull.await(),释放锁并处于阻塞状态; 一旦 ArrayBlockingQueue 在队列不满的时候,就立即入队。
- E take() 方法
public E take() throws InterruptedException {
final ReentrantLock lock = this.lock;
lock.lockInterruptibly();
try {
while (count == 0)
notEmpty.await();
return dequeue();
} finally {
lock.unlock();
}
}
take 方法总结:
- 取元素时,若队列为空,则调用 notEmpty.await(),进入阻塞状态,直至不为空时,调用 dequeue() 方法出队。
LinkedBlockingQueue
介绍
是一个基于链表的无界队列(理论上有界),向无限队列添加元素的所有操作正常情况下不会阻塞,因此它可以增长到非常大的容量。
- 队列创建:
BlockingQueue<string> blockingQueue = new LinkedBlockingQueue<>();</string>blockingQueue 如果在初始化时没有指定容量,那么容量默认为 Integer.MAX_VALUE - 注意:使用无限 BlockingQueue 设计生产者 - 消费者模型时,消费者需要能够像生产者向队列添加消息一样快地消费消息。否则,内存可能会填满,然后就会得到一个 OutOfMemory 异常。
- 底层数据结构 LinkedBlockingQueue 内部是使用链表实现一个队列的,但是有别于一般的队列,在于该队列至少是有一个节点的,头节点不含有元素。 如果队列为空时,头节点的 next 参数为null,尾节点的 next 参数也为null
- 主要变量
private final int capacity;
private final AtomicInteger count = new AtomicInteger();
transient Node<E> head;
private transient Node<E> last;
private final ReentrantLock takeLock = new ReentrantLock();
private final Condition notEmpty = takeLock.newCondition();
private final ReentrantLock putLock = new ReentrantLock();
private final Condition notFull = putLock.newCondition();
方法解析
- put(E e) 方法
public void put(E e) throws InterruptedException {
if (e == null) throw new NullPointerException();
int c = -1;
Node<E> node = new Node<E>(e);
final ReentrantLock putLock = this.putLock;
final AtomicInteger count = this.count;
putLock.lockInterruptibly();
try {
while (count.get() == capacity) {
notFull.await();
}
enqueue(node);
c = count.getAndIncrement();
if (c + 1 < capacity)
notFull.signal();
} finally {
putLock.unlock();
}
if (c == 0)
signalNotEmpty();
}
put 方法总结:
- LinkedBlockingQueue 不允许插入的元素为 null;
- 同一时刻只有一个线程可以进行入队操作,putLock 在将元素插入队列尾部时加锁了;
- 如果队列满了,则会调用 notFull.await(),将该线程加入到 Condition 等待队列中。await 方法会释放线程占有的锁,这将导致之前由于被阻塞的入队线程将会获取到锁,执行到while循环处,不过可能因为队列仍旧是满的,也被进入到条件队列中;
- 一旦有出队线程取走元素,就会通知到入队等待队列释放线程。那么第一个加入到 Condition 队列中的将会被释放,那么该线程将会重新获得 put 锁,继而执行 enqueue() 方法,将节点插入到队列的尾部;
- 然后得到插入队列前元素的个数,如果插入后队列中还可以继续插入元素,那么就通知 notFull 条件的等待队列中的线程;
- 如果插入队列前个数为 0,那现在插入后,就为 1 了,那就可以通知因为队列为空而导致阻塞的出队线程去取元素了。
- E take() 方法
public E take() throws InterruptedException {
E x;
int c = -1;
final AtomicInteger count = this.count;
final ReentrantLock takeLock = this.takeLock;
takeLock.lockInterruptibly();
try {
while (count.get() == 0) {
notEmpty.await();
}
x = dequeue();
c = count.getAndDecrement();
if (c > 1)
notEmpty.signal();
} finally {
takeLock.unlock();
}
if (c == capacity)
signalNotFull();
return x;
}
take方法总结:
- 同一时刻只有一个线程可以进行出队操作,takeLock 在出队之前加锁了;
- 如果队列中元素为空,那就进入 notEmpty 队列中进行等待。直到队列不为空时,得到队列中的第一个元素。当发现取完发现还有元素可取时,再通知一下 notEmpty 队列中等待的其他线程。最后判断自己取元素前的是不是满的,如果是满的,那自己取完,就不满了,就可以通知在 notFull 队列中等待插入的线程进行 put 了。
- remove() 方法 用于删除队列中一个元素,如果队列中不含有该元素,那么返回 false ;有的话则删除并返回true。 入队和出队都是只获取一个锁,而 remove()方法需要同时获得两把锁。
public boolean remove(Object o) {
if (o == null) return false;
fullyLock();
try {
for (Node<E> trail = head, p = trail.next;
p != null;
trail = p, p = p.next) {
if (o.equals(p.item)) {
unlink(p, trail);
return true;
}
}
return false;
} finally {
fullyUnlock();
}
}
void fullyLock() {
putLock.lock();
takeLock.lock();
}
void fullyUnlock() {
takeLock.unlock();
putLock.unlock();
}
DelayQueue
由优先级堆支持的、基于时间的调度队列,内部基于无界队列 PriorityQueue 实现,而无界队列基于数组的扩容实现。
- 队列创建:
BlockingQueue<string> blockingQueue = new DelayQueue(); </string> - 要求:入队的对象必须要实现 Delayed 接口,而 Delayed 集成自 Comparable 接口
- 应用场景:电影票
- 工作原理:队列内部会根据时间优先级进行排序。延迟类线程池周期执行。
LinkedBlockingQueue 和 ArrayBlockingQueue 的区别
- 底层实现不同 LinkedBlockingQueue 底层实现是链表,ArrayBlockingQueue 底层实现是数组
- 队列容量 LinkedBlockingQueue 默认的队列长度是 Integer.Max,但是可以指定容量。在入队与出队都高并发的情况下,性能比ArrayBlockingQueue 高很多; ArrayBlockingQueue 必须在构造方法中指定队列长度,不可变。在只有入队高并发或出队高并发的情况下,因为操作数组,且不需要扩容,性能很高。
- 锁的数量 LinkedBlockingQueue 有两把锁,可以有两个线程同时进行入队和出队操作,但同时只能有一个线程进行入队或出队操作。 ArrayBlockingQueue 只有一把锁,同时只能有一个线程进行入队和出队操作。
关注夏壹分享发送:资源 获取238本进阶书籍和大厂面试