Java并发包源码学习：阻塞队列实现之LBQ源码解析，守护线程面试题

// assert head.item == null;

Node h = head;

Node first = h.next;

【一线大厂Java面试题解析+后端开发学习笔记+最新架构讲解视频+实战项目源码讲义】

浏览器打开：qq.cn.hn/FTf 开源分享

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)

return null;

final ReentrantLock takeLock = this.takeLock;

takeLock.lock();

try {

// 实际上第一个元素是head.next

Node first = head.next;

if (first == null)

return null;

else

return first.item;

} finally {

takeLock.unlock();

}

}

Boolean remove(Object o)

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

移除队列中与元素o相等【指的是equals方法判定相同】的元素，移除成功返回true，如果队列为空或没有匹配元素，则返回false。

public boolean remove(Object o) {

if (o == null) return false;

fullyLock();

try {

// trail 和 p 同时向后遍历， 如果p匹配了，就让trail.next = p.next代表移除p

for (Node 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();

}

}

// trail为p的前驱， 希望移除p节点

void unlink(Node p, Node trail) {

// assert isFullyLocked();

// p.next is not changed, to allow iterators that are

// traversing p to maintain their weak-consistency guarantee.

p.item = null;

trail.next = p.next;// 移除p

// 如果p已经是最后一个节点了，就更新一下last

if (last == p)

last = trail;

// 移除一个节点之后，队列从满到未满， 唤醒notFull

if (count.getAndDecrement() == capacity)

notFull.signal();

}

//----- 多个锁 获取和释放的顺序是 相反的

// 同时上锁

void fullyLock() {

putLock.lock();

takeLock.lock();

}

// 同时解锁

void fullyUnlock() {

takeLock.unlock();

putLock.unlock();

}