【Java 集合】LinkedBlockingQueue

2023-12-20 16:28

本文主要是介绍【Java 集合】LinkedBlockingQueue,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

LinkedBlockingQueue, 顾名思义: 基于链表的阻塞队列, 位于 JUC (java.util.concurrent) 下, 是一个线程安全的集合, 其本身具备了

  1. 不支持 null 元素: 存入 null 元素会抛出异常
  2. 固定不限容量: 在不手动设置容量时, 最大可以支持 Integer.MAX_VALUE 个元素, 也就是理论上的无限个数
  3. 有序性: 内部采用链表作为底层数据结构,保持了元素的有序性。这意味着当你向队列中添加元素时,它们将按照添加的顺序排列,而消费者线程将按照相同的顺序取出这些元素
  4. LinkedBlockingQueue 会在队列满时, 阻塞添加数据的线程直至队列非满状态, 同样, 在队列空时, 阻塞获取数据的线程直至队列重新非空
  5. 支持锁公平性配置: 在初始化时可以指定是否使用公平锁, 默认为非公平锁。公平锁通常会降低吞吐量, 但是减少了可变性和避免了线程饥饿问题

1 实现的数据结构

内部的实现结构就是一个链表, 而且是所有链表中最简单的单向链表, 所以就不展开了。

2 源码分析

2.1 LinkedBlockingQueue 链表节点的定义

我们知道 LinkedBlockingQueue 的底层实现结构就是一个链表, 而链表绕不开的一个概念就是节点, 所以我们先来看一下 LinkedBlockingQueue 的节点定义。

public class LinkedBlockingQueue<E> {// 链表节点类, 数据的主要存储地方static class Node<E> {// 节点的数据E item;// 下一个节点Node<E> next;Node(E x) { item = x; }}
}

通过节点 Node 的定义中就一个数据域和一个指向下一个节点的指针, 明确 LinkedBlockingQueue 就是一个单向链表。

2.2 LinkedBlockingQueue 持有的属性

public class LinkedBlockingQueue<E> {/** 队列的大小,默认为 Integer.MAX_VALUE */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();
}

和 ArrayBlockingQueue 类似, 都是通过可重入锁控制并发, 通过 Condition 来实现线程的挂起和唤醒。
和 ArrayBlockingQueue 不同的时, 添加和删除数据, ArrayBlockingQueue 使用的是同一把锁, 而 LinkedBlockingQueue 将添加数据和删除数据分开, 使用了两把锁, 这样可以提高并发度。

2.3 LinkedBlockingQueue 构造函数

public class LinkedBlockingQueue<E> {// 无参构造函数public LinkedBlockingQueue() {// 默认大小为 Integer.MAX_VALUEthis(Integer.MAX_VALUE);}// 指定容量的构造函数public LinkedBlockingQueue(int capacity) {if (capacity <= 0)throw new IllegalArgumentException();this.capacity = capacity;// 初始化头尾节点last = head = new Node<E>(null);}// 指定集合的构造函数public LinkedBlockingQueue(Collection<? extends E> c) {// 同样默认为 Integer.MAX_VALUE 的容量this(Integer.MAX_VALUE);final ReentrantLock putLock = this.putLock;// 获取添加数据锁putLock.lock();try {int n = 0;for (E e : c) {// 非空校验if (e == null)throw new NullPointerException();// 达到容量了if (n == capacity)throw new IllegalStateException("Queue full");// 将当前的数据封装为 Node 节点, 添加到队列中// 先知道这个方法的作用, 后面新增数据的时候有分析    enqueue(new Node<E>(e));// 添加个数 + 1++n;}// 当前链表的个数设置为添加的个数count.set(n);} finally {putLock.unlock();}}
}

从 LinkedBlockingQueue 的 3 个构造函数中可以看出, 都需要通过指定容量构造函数先构建出一个只有 1 个节点, 同时节点数据为空的链表, 如下:

Alt 'LinkedBlockingQueue 初始状态'

在构建为这个空链表后, 实际我们的 LinkedBlockingQueue 算是初始化完成了。
而指定集合的构造函数, 会在初始化完成, 加上写锁后, 将入参的集合元素添加到当前的队列中。

同 ArrayBlockingQueue 一样, 在构造函数中使用了 ReentrantLock 来加锁, 通过这个锁达到 Happens-Before 的监视器原则, 达到后面可见性。

2.4 LinkedBlockingQueue 支持的方法

2.4.1 数据入队方法

LinkedBlockingQueue 提供了多种入队操作的实现来满足不同情况下的需求,入队操作有如下几种:

  1. boolean add(E e)
  2. boolean offer(E e)
  3. boolean offer(E e, long timeout, TimeUnit unit)
  4. void put(E e)

add(E e)

public class LinkedBlockingQueue<E> {public boolean add(E e) {// 直接调用自身的 offer 方法, 添加成功就返回 true// offer 方法后面分析if (offer(e))return true;elsethrow new IllegalStateException("Queue full");}
}

offer(E e)

顺着 add 方法, 看一下 offer 方法:

public class LinkedBlockingQueue<E> {public boolean offer(E e) {// 新增的数据为空, 直接抛异常if (e == null)throw new NullPointerException();final AtomicInteger count = this.count;// 达到了容量上限, 直接返回if (count.get() == capacity)return false;int c = -1;// 将当前的数据封装为 Node 节点Node<E> node = new Node<E>(e);final ReentrantLock putLock = this.putLock;// 获取添加锁putLock.lock();try {// 当前队列的容量还没达到上线if (count.get() < capacity) {// 将当前这个节点添加到队列的尾部enqueue(node);// 获取当前的数组容量c = count.getAndIncrement();// 当前的数组容量 + 1 后还是小于容量上限if (c + 1 < capacity)// 唤醒阻塞在非满条件上的线程notFull.signal();}} finally {putLock.unlock();}// 上面 c = count.getAndIncrement(), 会先将 count 当前的值赋给 c, 然后在 count 再 + 1// 所以这里 c == 0, 表示当前链表新增了一个节点前,没有数据, 尝试唤醒阻塞在非空条件上的线程if (c == 0)signalNotEmpty();return c >= 0;}private void enqueue(Node<E> node) {// 简单的链表新增节点操作// 把当前链表的尾结点的下一个节点设置为新增的节点// 把当前的尾节点更新为新增的节点last = last.next = node;}private void signalNotEmpty() {final ReentrantLock takeLock = this.takeLock;// 获取读锁takeLock.lock();try {// 唤醒在非空条件上阻塞的线程 (唤醒前需要获取对应的锁, 所以上面先获取了读锁)notEmpty.signal();} finally {// 释放读锁takeLock.unlock();}}
}

offer 方法概括一下就是:

  1. 非空和队列容量判断
  2. 获取写锁成功时, 将当前的元素封装为节点放到链表的尾部
  3. 添加成功后, 队列的容量没有达到上限, 唤醒在非满条件上阻塞的线程, 然后释放锁
  4. 如果当前添加的元素是当前链表的第一个元素, 唤醒在非空条件上阻塞的线程

offer(E e, long timeout, TimeUnit unit)

offer(E e, long timeout, TimeUnit unit) 方法只是在 offer(E e) 的基础上增加了超时时间的概念。在队列上阻塞了多少时间后, 队列还是满的, 就返回。

public class LinkedBlockingQueue<E> {public boolean offer(E e, long timeout, TimeUnit unit) throws InterruptedException {if (e == null)throw new NullPointerException();// 时间转为纳秒long nanos = unit.toNanos(timeout);int c = -1;final ReentrantLock putLock = this.putLock;final AtomicInteger count = this.count;// 获取可中断的写锁putLock.lockInterruptibly();try {// while 循环的目的是防止在中断后没有到达传入的 timeout 时间, 可以继续重试// 等待超时时间达到了 nanos,就直接返回 falsewhile (count.get() == capacity) {if (nanos <= 0)return false;// 将当前线程阻塞在非空条件上     nanos = notFull.awaitNanos(nanos);}// 把当前的节点放到队列的尾部enqueue(new Node<E>(e));// 获取当前的数组容量c = count.getAndIncrement();// 当前的数组容量 + 1 后还是小于容量上限if (c + 1 < capacity)// 唤醒阻塞在非空条件上的线程notFull.signal();} finally {putLock.unlock();}// 上面 c = count.getAndIncrement(), 会先将 count 当前的值赋给 c, 然后在 count 再 + 1// 所以这里 c == 0, 表示当前链表新增了一个节点前,没有数据, 尝试唤醒阻塞在非空条件上的线程if (c == 0)signalNotEmpty();return true;}
}

可以看到主要的逻辑和 offer(E e) 方法一样, 只是在队列满时, 使用的是带超时时间的阻塞方法。
该方法会在等待时间到达后, 会自动唤醒, 同时线程时间到达后被唤醒, 队列还是满的, 就返回 false。

put(E e)

public class LinkedBlockingQueue<E> {public void put(E e) throws InterruptedException {// 非空校验if (e == null)throw new NullPointerException();int c = -1;Node<E> node = new Node<E>(e);// 获取添加需要的 putLock, 也就是写锁final ReentrantLock putLock = this.putLock;final AtomicInteger count = this.count;// 获取锁中断putLock.lockInterruptibly();try {// 判断队列是否已满,如果已满阻塞等待// 如果每次唤醒都是满的话,就会一直阻塞while (count.get() == capacity) {notFull.await();}// 把node放入队列中enqueue(node);// 链表中的元素个数 + 1c = count.getAndIncrement();// 再次判断队列是否有可用空间,如果有唤醒下一个线程进行添加操作if (c + 1 < capacity)notFull.signal();}  finally {putLock.unlock();}// 上面 c = count.getAndIncrement(), 会先将 count 当前的值赋给 c, 然后在 count 再 + 1// 所以这里 c == 0, 表示当前链表新增了一个节点前,没有数据, 尝试唤醒阻塞在非空条件上的线程if (c == 0)signalNotEmpty();}}

put 方法和 offer 方法类似, 前者在当队列已满时, 进入阻塞, 后者在队列已满时, 则是返回。

2.3.2 数据出队方法

同入队的方法一样, 出队也有多种实现, LinkedBlockingQueue 提供了好几种出队的方法, 大体如下:

  1. E poll();
  2. E poll(long timeout, TimeUnit unit);
  3. E take()

poll()

public class LinkedBlockingQueue<E> {public E poll() {final AtomicInteger count = this.count;// 当前的数组为节点个数为空, 直接返回 nullif (count.get() == 0)return null;E x = null;int c = -1;final ReentrantLock takeLock = this.takeLock;// 获取读锁takeLock.lock();try {// 当前的数组容量大于 0, 有数据可以获取if (count.get() > 0) {// 获取头节点的数据x = dequeue();// 当前数组的节点个数 - 1c = count.getAndDecrement();// 当前数组的节点个数还是大于 1if (c > 1)// 唤醒阻塞在非空条件上的线程notEmpty.signal();}} finally {// 释放读锁takeLock.unlock();}// 当前的容量等于上限了if (c == capacity)// 唤醒阻塞在非满条件上阻塞的线程signalNotFull();return x;}private E dequeue() {// 简单的链表删除操作Node<E> h = head;// 获取当前头节点的下一个节点Node<E> first = h.next;// 将头节点下一个节点设置为自身, 这里的操作后面分析h.next = h;// 将当前的头节点设置为头节点的下一个节点head = first;// 获取旧的头结点的数据E x = first.item;// 设置旧的节点的数据为 nullfirst.item = null;// 返回旧的节点的数据return x;}private void signalNotFull() {final ReentrantLock putLock = this.putLock;// 获取写锁putLock.lock();try {// 唤醒阻塞在非满条件上的线程notFull.signal();} finally {putLock.unlock();}}
}

poll() 逻辑整理如下

  1. 队列为空, 直接返回 null
  2. 获取读锁, 队列不为空, 获取头节点的数据
  3. 获取后队列还是有数据, 唤醒阻塞在非空条件上的线程, 释放锁
  4. 如果当前的容量等于上限了, 再次唤醒阻塞在非满条件上阻塞的线程
public class LinkedBlockingQueue<E> {public E poll(long timeout, TimeUnit unit) throws InterruptedException {E x = null;int c = -1;long nanos = unit.toNanos(timeout);final AtomicInteger count = this.count;final ReentrantLock takeLock = this.takeLock;takeLock.lockInterruptibly();try {// 当前的数组容量为 0, 队列为空, 将当前线程阻塞在非空条件上while (count.get() == 0) {if (nanos <= 0)return null;// 带超时时间的等待nanos = notEmpty.awaitNanos(nanos);}// 队列不为空, 获取头节点的数据x = dequeue();// 当前数组的节点个数 - 1c = count.getAndDecrement();// 当前数组的节点个数还是大于 1if (c > 1)// 唤醒阻塞在非空条件上的线程notEmpty.signal();} finally {takeLock.unlock();}// 当前的容量等于上限了if (c == capacity)// 唤醒阻塞在非满条件上阻塞的线程signalNotFull();return x;}
}

整个 poll(long timeout, TimeUnit unit) 方法和 poll() 方法类似, 只是在队列为空时, 使用的是带超时时间的阻塞方法。

take()

public class LinkedBlockingQueue<E> {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() 方法和 poll() 方法类似, 当队列为空时, 进入阻塞, 一直等待, 直到被唤醒, 唤醒后调用 dequeue() 方法获取队列中的元素, 而 poll() 方法则是在队列为空时, 直接返回 null。

2.3.3 获取元素方法

获取数据的就一个方法, 同时只支持获取头节点的数据。

public class LinkedBlockingQueue<E> {public E peek() {// 当前数组上的节点为 0, 直接返回if (count.get() == 0)return null;final ReentrantLock takeLock = this.takeLock;// 获取读锁takeLock.lock();try {// 获取头节点的下一个节点 (有数据的头节点)Node<E> first = head.next;// 头节点的下一个节点为 nullif (first == null)// 返回 nullreturn null;else// 返回头节点的数据return first.item;} finally {takeLock.unlock();}}
}

很简单, 就是获取头节点的下一个节点的数据, 有数据返回数据, 没有数据, 就返回 null。

2.3.4 删除元素方法

public class LinkedBlockingQueue<E> {public boolean remove(Object o) {if (o == null) return false;// 两个 lock 全部上锁fullyLock();try {// 从 head 开始遍历元素,直到最后一个元素for (Node<E> trail = head, p = trail.next; p != null; trail = p, p = p.next) {// 如果找到相等的元素,调用 unlink 方法删除元素if (o.equals(p.item)) {unlink(p, trail);return true;}}return false;} finally {// 两个lock全部解锁fullyUnlock();}}void fullyLock() {putLock.lock();takeLock.lock();}void unlink(Node<E> p, Node<E> pred) {// 简单的链表节点删除操作// 将这个节点的数据项设置为 null// 同样是删除元素, 什么这里这里删除的元素 p 不需要想 dequeue 方法一样, 将他的 next 设置为 自身?p.item = null;// 将这个节点的上一个节点的 next 设置为删除节点 p 的下一个节点pred.next = p.next;// 删除的节点为尾结点if (last == p)// 设置尾节点为删除节点的上一个节点last = pred;// 当前数组的容量减 1 后还是等于数组上限  if (count.getAndDecrement() == capacity)// 唤醒 notFull 上等待队列的线程notFull.signal();}void fullyUnlock() {takeLock.unlock();putLock.unlock();}
}

3 问题

在上面的源码的分析中, 留了 2 个比较特殊的地方

  1. dequeue 方法中里的移除的元素 h, 为什么需要将 h.next 指向自身, 而不是 null
  2. 同样删除元素的 unlink 方法却不需要将 p.next = null 或者 p.next = p 这样的操作

这里都可以在 LinkedBlockingQueue 内部的迭代器中找到答案


private class Itr implements Iterator<E> {// 遍历的链表头节点private Node<E> current;// 遍历的链表头节点的数据private E currentElement;// 构造函数Itr() {// 加双锁fullyLock();try {current = head.next;if (current != null)currentElement = current.item;} finally {fullyUnlock();}}// 获取入参节点的下一个节点private Node<E> nextNode(Node<E> p) {for (;;) {// 解决了问题 1Node<E> s = p.next;if (s == p)return head.next;if (s == null || s.item != null)return s;p = s;}}
}

问题 1 在 nextNode 方法中也能找到

  1. 为了正确遍历,nextNode 中通过 s == p 进行判断,当下一个元素是自己本身时,返回 head 的下一个节点。

而第二个问题主要在迭代器的遍历中, 迭代器的遍历分为两步

  1. 加双锁, 然后链表的头节点放到一个临时变量中
  2. 遍历临时变量的元素。在遍历的过程是无锁的, 也就是其他线程可以执行 remove 方法, 如果其他线程通过 unlink 修改了 p 的 next,可能在导致迭代异常

4 总结

它是 BlockingQueue 接口的一种实现,通过链表的形式存储元素,在不明确指定容量时, 可以存储 Integer.MAX_VALUE 个元素, 也就是理论上的无上限, 但是建议在使用中还是指定容量, 避免 OOM 等问题。
同时借助 2 个 ReentrantLock 达到读写互不影响, 提高并发性能。同时 ReentrantLock 的 Condition 提供了阻塞操作,使得在队列已满或为空时,线程能够安全地等待。

5 参考

【细谈Java并发】谈谈LinkedBlockingQueue

这篇关于【Java 集合】LinkedBlockingQueue的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/516670

相关文章

Java实现字节字符转bcd编码

《Java实现字节字符转bcd编码》BCD是一种将十进制数字编码为二进制的表示方式,常用于数字显示和存储,本文将介绍如何在Java中实现字节字符转BCD码的过程,需要的小伙伴可以了解下... 目录前言BCD码是什么Java实现字节转bcd编码方法补充总结前言BCD码(Binary-Coded Decima

SpringBoot全局域名替换的实现

《SpringBoot全局域名替换的实现》本文主要介绍了SpringBoot全局域名替换的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一... 目录 项目结构⚙️ 配置文件application.yml️ 配置类AppProperties.Ja

Java使用Javassist动态生成HelloWorld类

《Java使用Javassist动态生成HelloWorld类》Javassist是一个非常强大的字节码操作和定义库,它允许开发者在运行时创建新的类或者修改现有的类,本文将简单介绍如何使用Javass... 目录1. Javassist简介2. 环境准备3. 动态生成HelloWorld类3.1 创建CtC

JavaScript中的高级调试方法全攻略指南

《JavaScript中的高级调试方法全攻略指南》什么是高级JavaScript调试技巧,它比console.log有何优势,如何使用断点调试定位问题,通过本文,我们将深入解答这些问题,带您从理论到实... 目录观点与案例结合观点1观点2观点3观点4观点5高级调试技巧详解实战案例断点调试:定位变量错误性能分

Java实现将HTML文件与字符串转换为图片

《Java实现将HTML文件与字符串转换为图片》在Java开发中,我们经常会遇到将HTML内容转换为图片的需求,本文小编就来和大家详细讲讲如何使用FreeSpire.DocforJava库来实现这一功... 目录前言核心实现:html 转图片完整代码场景 1:转换本地 HTML 文件为图片场景 2:转换 H

Java使用jar命令配置服务器端口的完整指南

《Java使用jar命令配置服务器端口的完整指南》本文将详细介绍如何使用java-jar命令启动应用,并重点讲解如何配置服务器端口,同时提供一个实用的Web工具来简化这一过程,希望对大家有所帮助... 目录1. Java Jar文件简介1.1 什么是Jar文件1.2 创建可执行Jar文件2. 使用java

SpringBoot实现不同接口指定上传文件大小的具体步骤

《SpringBoot实现不同接口指定上传文件大小的具体步骤》:本文主要介绍在SpringBoot中通过自定义注解、AOP拦截和配置文件实现不同接口上传文件大小限制的方法,强调需设置全局阈值远大于... 目录一  springboot实现不同接口指定文件大小1.1 思路说明1.2 工程启动说明二 具体实施2

Java实现在Word文档中添加文本水印和图片水印的操作指南

《Java实现在Word文档中添加文本水印和图片水印的操作指南》在当今数字时代,文档的自动化处理与安全防护变得尤为重要,无论是为了保护版权、推广品牌,还是为了在文档中加入特定的标识,为Word文档添加... 目录引言Spire.Doc for Java:高效Word文档处理的利器代码实战:使用Java为Wo

SpringBoot日志级别与日志分组详解

《SpringBoot日志级别与日志分组详解》文章介绍了日志级别(ALL至OFF)及其作用,说明SpringBoot默认日志级别为INFO,可通过application.properties调整全局或... 目录日志级别1、级别内容2、调整日志级别调整默认日志级别调整指定类的日志级别项目开发过程中,利用日志

Java中的抽象类与abstract 关键字使用详解

《Java中的抽象类与abstract关键字使用详解》:本文主要介绍Java中的抽象类与abstract关键字使用详解,本文通过实例代码给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧... 目录一、抽象类的概念二、使用 abstract2.1 修饰类 => 抽象类2.2 修饰方法 => 抽象方法,没有