【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

相关文章

Spring boot整合dubbo+zookeeper的详细过程

《Springboot整合dubbo+zookeeper的详细过程》本文讲解SpringBoot整合Dubbo与Zookeeper实现API、Provider、Consumer模式,包含依赖配置、... 目录Spring boot整合dubbo+zookeeper1.创建父工程2.父工程引入依赖3.创建ap

SpringBoot结合Docker进行容器化处理指南

《SpringBoot结合Docker进行容器化处理指南》在当今快速发展的软件工程领域,SpringBoot和Docker已经成为现代Java开发者的必备工具,本文将深入讲解如何将一个SpringBo... 目录前言一、为什么选择 Spring Bootjavascript + docker1. 快速部署与

Spring Boot spring-boot-maven-plugin 参数配置详解(最新推荐)

《SpringBootspring-boot-maven-plugin参数配置详解(最新推荐)》文章介绍了SpringBootMaven插件的5个核心目标(repackage、run、start... 目录一 spring-boot-maven-plugin 插件的5个Goals二 应用场景1 重新打包应用

SpringBoot+EasyExcel实现自定义复杂样式导入导出

《SpringBoot+EasyExcel实现自定义复杂样式导入导出》这篇文章主要为大家详细介绍了SpringBoot如何结果EasyExcel实现自定义复杂样式导入导出功能,文中的示例代码讲解详细,... 目录安装处理自定义导出复杂场景1、列不固定,动态列2、动态下拉3、自定义锁定行/列,添加密码4、合并

Spring Boot集成Druid实现数据源管理与监控的详细步骤

《SpringBoot集成Druid实现数据源管理与监控的详细步骤》本文介绍如何在SpringBoot项目中集成Druid数据库连接池,包括环境搭建、Maven依赖配置、SpringBoot配置文件... 目录1. 引言1.1 环境准备1.2 Druid介绍2. 配置Druid连接池3. 查看Druid监控

Java中读取YAML文件配置信息常见问题及解决方法

《Java中读取YAML文件配置信息常见问题及解决方法》:本文主要介绍Java中读取YAML文件配置信息常见问题及解决方法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要... 目录1 使用Spring Boot的@ConfigurationProperties2. 使用@Valu

创建Java keystore文件的完整指南及详细步骤

《创建Javakeystore文件的完整指南及详细步骤》本文详解Java中keystore的创建与配置,涵盖私钥管理、自签名与CA证书生成、SSL/TLS应用,强调安全存储及验证机制,确保通信加密和... 目录1. 秘密键(私钥)的理解与管理私钥的定义与重要性私钥的管理策略私钥的生成与存储2. 证书的创建与

浅析Spring如何控制Bean的加载顺序

《浅析Spring如何控制Bean的加载顺序》在大多数情况下,我们不需要手动控制Bean的加载顺序,因为Spring的IoC容器足够智能,但在某些特殊场景下,这种隐式的依赖关系可能不存在,下面我们就来... 目录核心原则:依赖驱动加载手动控制 Bean 加载顺序的方法方法 1:使用@DependsOn(最直

SpringBoot中如何使用Assert进行断言校验

《SpringBoot中如何使用Assert进行断言校验》Java提供了内置的assert机制,而Spring框架也提供了更强大的Assert工具类来帮助开发者进行参数校验和状态检查,下... 目录前言一、Java 原生assert简介1.1 使用方式1.2 示例代码1.3 优缺点分析二、Spring Fr

java使用protobuf-maven-plugin的插件编译proto文件详解

《java使用protobuf-maven-plugin的插件编译proto文件详解》:本文主要介绍java使用protobuf-maven-plugin的插件编译proto文件,具有很好的参考价... 目录protobuf文件作为数据传输和存储的协议主要介绍在Java使用maven编译proto文件的插件