Java并发包源码学习：阻塞队列实现之LBQ源码解析(1)

// 传入指定的容量

public LinkedBlockingQueue(int capacity) {

if (capacity <= 0) throw new IllegalArgumentException();

this.capacity = capacity;

// 初始化last 和 head指针

last = head = new Node(null);

}

// 传入指定集合对象，容量视为Integer.MAX_VALUE，直接构造queue

public LinkedBlockingQueue(Collection<? extends E> c) {

this(Integer.MAX_VALUE);

// 写线程获取putLock

final ReentrantLock putLock = this.putLock;

putLock.lock(); // Never contended, but necessary for visibility

try {

int n = 0;

for (E e : c) {

if (e == null)

throw new NullPointerException();

if (n == capacity)

throw new IllegalStateException(“Queue full”);

enqueue(new Node(e));

++n;

}

count.set(n);

} finally {

putLock.unlock();

}

}

出队和入队操作

=======

队列的操作最核心的部分莫过于入队和出队了，后面分析的方法基本上都基于这两个工具方法。

LinkedBlockingQueue的出队和入队相对ArrayBlockingQueue来说就简单很多啦：

入队enqueue

=========

private void enqueue(Node node) {

// assert putLock.isHeldByCurrentThread();

// assert last.next == null;

last = last.next = node;

}

1. 将node连接到last的后面。

2. 更新last指针的位置，指向node。

出队dequeue

=========

private E dequeue() {

// assert takeLock.isHeldByCurrentThread();

// assert head.item == null;

Node h = head;

Node first = h.next;

h.next = h; // help GC

head = first; // head向后移一位

E x = first.item;

first.item = null;

return x;

}

队列中的元素实际上是从head.first开始的，那么移除队头，其实就是将head指向head.next即可。

阻塞式操作

=====

E take() 阻塞式获取

==============

take操作将会获取当前队列头部元素并移除，如果队列为空则阻塞当前线程直到队列不为空，退出阻塞时返回获取的元素。

如果线程在阻塞时被其他线程设置了中断标志，则抛出InterruptedException异常并返回。

public E take() throws InterruptedException {

E x;

int c = -1;

final AtomicInteger count = this.count;

// 首先要获取takeLock

final ReentrantLock takeLock = this.takeLock;

takeLock.lockInterruptibly();

try {

// 如果队列为空， notEmpty不满足，就等着

while (count.get() == 0) {

notEmpty.await();

}

// 出队

x = dequeue();

// c先赋值为count的值， count 减 1

c = count.getAndDecrement();

// 这次出队后至少还有一个元素，唤醒notEmpty中的读线程

if (c > 1)

notEmpty.signal();

} finally {

takeLock.unlock();

}

// c == capacity 表示在该元素出队之前，队列是满的

if (c == capacity)

// 因为在这之前队列是满的，可能会有写线程在等着，这里做个唤醒

signalNotFull();

return x;

}

// 用于唤醒写线程

private void signalNotFull() {

final ReentrantLock putLock = this.putLock;

// 获取putLock

putLock.lock();

try {

notFull.signal();

} finally {

putLock.unlock();

}

}

void put(E e) 阻塞式插入

===================

put操作将向队尾插入元素，如果队列未满则插入，如果队列已满，则阻塞当前线程直到队列不满。

如果线程在阻塞时被其他线程设置了中断标志，则抛出InterruptedException异常并返回。

public void put(E e) throws InterruptedException {

if (e == null) throw new NullPointerException();

// 所有的插入操作 都约定 本地变量c 作为是否失败的标识

int c = -1;

Node node = new Node(e);

final ReentrantLock putLock = this.putLock;

final AtomicInteger count = this.count;

// 插入操作获取 putLock

putLock.lockInterruptibly();

try {

// 队列满，这时notFull条件不满足，await

while (count.get() == capacity) {

notFull.await();

}

enqueue(node);

// c先返回count的值 ， 原子变量 + 1 ，

c = count.getAndIncrement();

// 至少还有一个空位可以插入，notFull条件是满足的，唤醒它

if (c + 1 < capacity)

notFull.signal();

} finally {

putLock.unlock();

}

// c == 0 表示在该元素入队之前，队列是空的

if (c == 0)

// 因为在这之前队列是空的，可能会有读线程在等着，这里做个唤醒

signalNotEmpty();

}

// 用于唤醒读线程

private void signalNotEmpty() {

final ReentrantLock takeLock = this.takeLock;

// 获取takeLock

takeLock.lock();

try {

// 唤醒

notEmpty.signal();

} finally {

takeLock.unlock();

}

}

E poll(timeout, unit) 阻塞式超时获取

=============================

在take阻塞式获取方法的基础上额外增加超时功能，传入一个timeout，获取不到而阻塞的时候，如果时间到了，即使还获取不到，也只能立即返回null。

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 {

// 这里就是超时机制的逻辑所在

while (count.get() == 0) {

if (nanos <= 0)

return null;

nanos = notEmpty.awaitNanos(nanos);

}

x = dequeue();

c = count.getAndDecrement();

if (c > 1)

notEmpty.signal();

} finally {

takeLock.unlock();

}

if (c == capacity)

signalNotFull();

return x;

}

boolean offer(e, timeout, unit) 阻塞式超时插入

=======================================

在put阻塞式插入方法的基础上额外增加超时功能，传入一个timeout，获取不到而阻塞的时候，如果时间到了，即使还获取不到，也只能立即返回null。

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 (count.get() == capacity) {

if (nanos <= 0)

return false;

nanos = notFull.awaitNanos(nanos);

}

enqueue(new Node(e));

c = count.getAndIncrement();

if (c + 1 < capacity)

notFull.signal();

} finally {

putLock.unlock();

}

if (c == 0)

signalNotEmpty();

return true;

}

其他常规操作

======

boolean offer(E e)

==================

offer(E e)是非阻塞的方法，向队尾插入一个元素，如果队列未满，则插入成功并返回true；如果队列已满则丢弃当前元素，并返回false。

public boolean offer(E e) {

if (e == null) throw new NullPointerException();

final AtomicInteger count = this.count;

// 此时队列已满，直接返回false

if (count.get() == capacity)

return false;

int c = -1;

Node node = new Node(e);

// 插入操作 获取putLock

final ReentrantLock putLock = this.putLock;

putLock.lock();

try {

// 加锁后再校验一次

if (count.get() < capacity) {

enqueue(node);

c = count.getAndIncrement();

if (c + 1 < capacity)

notFull.signal();

}

} finally {

putLock.unlock();

}

if (c == 0)

signalNotEmpty();

return c >= 0; // 只要不是-1，就代表成功~

}

E poll()

========

从队列头部获取并移除第一个元素，如果队列为空则返回null。

public E poll() {

final AtomicInteger count = this.count;

if (count.get() == 0)

return null;

E x = null;

int c = -1;

final ReentrantLock takeLock = this.takeLock;

takeLock.lock();

try {

// 如果队列不为空，则出队， 并递减计数

if (count.get() > 0) {

x = dequeue();

c = count.getAndDecrement();

if (c > 1)

notEmpty.signal();

}

} finally {

takeLock.unlock();

}

if (c == capacity)

signalNotFull();

return x;

}

E peek()

========

瞅一瞅队头的元素是啥，如果队列为空，则返回null。

public E peek() {

if (count.get() == 0)

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长，自己不成体系的自学效果低效漫长且无助。

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，不论你是刚入门Java开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

public E peek() {

if (count.get() == 0)

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长，自己不成体系的自学效果低效漫长且无助。

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

[外链图片转存中…(img-fuDCWt1J-1715736788554)]

[外链图片转存中…(img-htkorIi0-1715736788555)]

[外链图片转存中…(img-tAkgm335-1715736788555)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，不论你是刚入门Java开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！