AbstractQueuedSynchronizer源码解读--续篇之Condition
1. 背景
在之前的AbstractQueuedSynchronizer源码解读中，介绍了AQS的基本概念、互斥锁、共享锁、AQS对同步队列状态流转管理、线程阻塞与唤醒等内容。其中并不涉及Condition相关的内容。本文主要介绍AQS中Condition的实现即ConditionObject类的源码。
Condition在JUC中使用很多，最常见的就是各种BlockingQueue了。
2. Condition是什么
java.util.concurrent.locks.Condition是JUC提供的与Java的Object中wait/notify/notifyAll类似功能的一个接口，通过此接口，线程可以在某个特定的条件下等待/唤醒。
与wait/notify/notifyAll操作需要获得对象监视器类似，一个Condition实例与某个互斥锁绑定，在此Condition实例进行等待/唤醒操作的调用也需要获得互斥锁，线程被唤醒后需要再次获取到锁，否则将继续等待。
而与原生的wait/notify/notifyAll等API不同的地方在于,JUC提供的Condition具有更丰富的功能，例如等待可以响应/不响应中断，可以设定超时时间或是等待到某个具体时间点。
此外一把互斥锁可以绑定多个Condition，这意味着在同一把互斥锁上竞争的线程可以在不同的条件下等待，唤醒时可以根据条件来唤醒线程，这是Object中的wait/notify/notifyAll不具备的机制
3. 代码解读
3.1 套路
JUC中Condition接口的主要实现类是AQS的内部类ConditionObject，它内部维护了一个队列，我们可以称之为条件队列，在某个Condition上等待的线程被signal/signalAll后，ConditionObject会将对应的节点转移到外部类AQS的等待队列中，线程需要获取到AQS等待队列的锁，才可以继续恢复执行后续的用户代码。
这里给出一个流程:
await流程:
1. 创建节点加入到条件队列
2. 释放互斥锁
3. 只要没有转移到同步队列就阻塞(等待其他线程调用signal/signalAll或是被中断)
4. 重新获取互斥锁
signal流程:
1. 将队列中第一个节点转移到同步队列
2. 根据情况决定是否要唤醒对应线程
这里以我之前在[AbstractQueuedSynchronizer源码解读]画的AQS状态流转图来说明下：
如果一个节点通过ConditionObject#await等方法调用初始化后，在被唤醒之后，会将状态切换至0，也即无状态，随后进入AQS的同步队列，此后就与一般的争锁无异了。
3.2 await方法
public final void await() throws InterruptedException {
// 对中断敏感。
if (Thread.interrupted())
throw new InterruptedException();
// 加到条件队列中。
Node node = addConditionWaiter();
// 完全释放互斥锁(无论锁是否可以重入)，如果没有持锁，会抛出异常。
int savedState = fullyRelease(node);
int interruptMode = 0;
/*
* 只要仍未转移到同步队列就阻塞。
* 转移的情况如下:
* 1. 其它线程调用signal将当前线程节点转移到同步队列并唤醒当前线程。
* 2. 其它线程调用signalAll。
* 3. 其它线程中断了当前线程,当前线程会自行尝试进入同步队列。
*/
while (!isOnSyncQueue(node)) {
LockSupport.park(this);
/*
* 获取中断模式。
* 在线程从park中被唤醒的时候，需要判断是否此时被中断，若中断则尝试转移到同步队列。
* 1. 中断且自行进入同步队列,返回THROW_IE(值-1),后续需要抛出InterruptedException。
* 2. 中断且未能自行进入同步队列,则说明有线程调用signal/signalAll唤醒线程并尝试转移到同步队列,
* 返回REINTERRUPT，后续重新中断线程。
* 3. 线程未被中断,返回0，此时需要重试循环判断是否上了同步队列。
*/
if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
break;
}
// 重新获取互斥锁过程中如果中断并且interruptMode不为"抛出异常",设置为REINTERRUPT。
if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
interruptMode = REINTERRUPT;
if (node.nextWaiter != null) // clean up if cancelled
unlinkCancelledWaiters();
// 如果线程发生过中断则根据THROW_IE或是REINTERRUPT分别抛出异常或者重新中断。
if (interruptMode != 0)
reportInterruptAfterWait(interruptMode);
}
private Node addConditionWaiter() {
Node t = lastWaiter;
/*
* 如果条件队列中最后一个waiter节点状态为取消,
* 则调用unlinkCancelledWaiters清理队列。
*/
if (t != null && t.waitStatus != Node.CONDITION) {
unlinkCancelledWaiters();
// 重读lastWaiter。
t = lastWaiter;
}
Node node = new Node(Thread.currentThread(), Node.CONDITION);
// t如果为null, 初始化firstWaiter为当前节点。
if (t == null)
firstWaiter = node;
else
// 将队尾的next连接到node。
t.nextWaiter = node;
lastWaiter = node;
return node;
}
/**
* 移除队列中所有取消节点。
*/
private void unlinkCancelledWaiters() {
Node t = firstWaiter;
// 记录上一个非取消节点。
Node trail = null;
while (t != null) {
Node next = t.nextWaiter;
if (t.waitStatus != Node.CONDITION) {
// 断开
t.nextWaiter = null;
if (trail == null)
// 如果trail为null,取当前节点的后继作为头节点的值(next可以为null)。
firstWaiter = next;
else
// 否则把trail连接到当前节点的后继。
trail.nextWaiter = next;
// 如果当前节点没有后继了, 更新lastWaiter为trail, 即上一个非取消节点。
if (next == null)
lastWaiter = trail;
}
else
trail = t;
t = next;
}
}
final boolean isOnSyncQueue(Node node) {
/*
* 节点状态为CONDITION一定是在条件队列,
* 或者如果prev为null也一定是在条件队列。
*
* 同步队列里的节点prev为null只可能是获取到锁后调用setHead清为null,
* 新入队的节点prev值是不会为null的。
* 另外,条件队列里节点是用nextWaiter来维护的，不用next和prev。
*/
if (node.waitStatus == Node.CONDITION || node.prev == null)
return false;
/*
* 如果next不为null,一定是在同步队列的。
* 这里值得一提的是在AQS的cancelAcquire方法中,
* 一个节点将自己移除出队列的时候会把自己的next域指向自己。
* 即node.next = node;
*
* 从GC效果上来看node.next = node和node.next = null无异,
* 但是这对此处next不为null一定在同步队列上来说,
* 这样可以将节点在同步队列上被取消的情况与普通情况归一化判断。
*/
if (node.next != null)
return true;
/*
* 有可能node.prev的值不为null,但还没在队列中,因为入队时CAS队列的tail可能失败。
* 这是从tail向前遍历一次,确定是否已经在同步队列上。
*/
return findNodeFromTail(node);
}
/**
* 从队列尾部向前遍历判断节点是否在队列中。
*/
private boolean findNodeFromTail(Node node) {
Node t = tail;
for (;;) {
if (t == node)
return true;
if (t == null)
return false;
t = t.prev;
}
}
private int checkInterruptWhileWaiting(Node node) {
/*
* 1. 线程未中断返回0
* 2. 线程中断且入同步队列成功,返回THROW_IE表示后续要抛出InterruptedException。
* 3. 线程中断且未能入同步队列(由于被signal方法唤醒),则返回REINTERRUPT表示后续重新中断。
*/
return Thread.interrupted() ?
(transferAfterCancelledWait(node) ? THROW_IE : REINTERRUPT) :
0;
}
final boolean transferAfterCancelledWait(Node node) {
if (compareAndSetWaitStatus(node, Node.CONDITION, 0)) {
enq(node);
return true;
}
/*
* 上面CAS失败的原因是signal()方法被调用,状态已经被抢先更新了。
* 这时需要自旋等待节点成功进入同步队列,否则会影响后续的重新获取锁acquireQueued()方法。
* 因为acquireQueued必须在节点成功入队后才可以调用。
*/
while (!isOnSyncQueue(node))
Thread.yield();
return false;
}
/**
* THROW_IE则抛出InterruptedException,
* REINTERRUPT则重新中断当前线程。
*/
private void reportInterruptAfterWait(int interruptMode)
throws InterruptedException {
if (interruptMode == THROW_IE)
throw new InterruptedException();
else if (interruptMode == REINTERRUPT)
selfInterrupt();
}
static void selfInterrupt() {
Thread.currentThread().interrupt();
}
3.3 signal/signalAll方法
public final void signal() {
// 检查互斥锁持有情况。
if (!isHeldExclusively())
throw new IllegalMonitorStateException();
Node first = firstWaiter;
if (first != null)
doSignal(first);
}
private void doSignal(Node first) {
do {
// 将firstWaiter设置为后继节点,如果为null,则置lastWaiter为null。
if ( (firstWaiter = first.nextWaiter) == null)
lastWaiter = null;
// 断开连接。
first.nextWaiter = null;
/*
* 如果转移失败并且下一个节点不为null,则重试。
* 在这里转移失败只可能因为节点被取消。
*/
} while (!transferForSignal(first) &&
(first = firstWaiter) != null);
}
public final void signalAll() {
// 检查互斥锁持有情况。
if (!isHeldExclusively())
throw new IllegalMonitorStateException();
Node first = firstWaiter;
if (first != null)
doSignalAll(first);
}
private void doSignalAll(Node first) {
// 将firstWaiter和lastWaiter先清为null。
lastWaiter = firstWaiter = null;
// 从first开始一直遍历到第一个null节点。
do {
Node next = first.nextWaiter;
first.nextWaiter = null;
transferForSignal(first);
first = next;
} while (first != null);
}
final boolean transferForSignal(Node node) {
// 必须将状态从CONDITION流转到0,如果失败则说明节点被取消,因为这里不会存在signal的竞争。
if (!compareAndSetWaitStatus(node, Node.CONDITION, 0))
return false;
// 插入同步队列。
Node p = enq(node);
int ws = p.waitStatus;
/*
* 如果前驱节点状态为取消或者无法将状态CAS到SIGNAL(比如可能前驱在此期间被取消了),
* 则需要唤醒参数node节点对应的线程,使其能开始尝试争锁。
*
* 如果将前驱状态切到SIGNAL了,则由相应线程在释放锁之后唤醒node节点对应线程。
*/
if (ws > 0 || !compareAndSetWaitStatus(p, ws, Node.SIGNAL))
LockSupport.unpark(node.thread);
return true;
}
4. 思考与总结
至此，已经对ConditionObject的await/signal/signalAll方法源码进行了分析。对于中断不敏感的awaitUninterruptibly, 带有时限的awaitNanos由于大致套路与await无异，不作冗述。
ConditionObject的firstWaiter/lastWaiter以及AQS.Node的nextWaiter都是没有volatile修饰的。这是因为ConditionObject假设在await/signal/signalAll等方法的调用是已经持有互斥锁的。
个人认为ConditionObject这样的设计是有个问题的。即按照Condition接口的定义，在不持锁情况下调用await由子类决定如何处理，通常是抛出InterruptedException。但如果同时有持锁和不持锁的线程调用await方法，可能会对ConditionObject的内部队列造成破坏，后果就是有些成功调用await方法的线程可能永远没有办法被唤醒,因为无法通过队列追溯到它们。也就是非法调用会抛出异常，但仍然会对内部数据结构造成破坏，这其实是有一些不合理的，至少是可以改进的地方。
最简单的处理方式是，对于不持锁的请求抛出异常，不应该依靠await -> fullyRelease这一步来抛出异常，此时按照流程已经调用过addConditionWaiter了。可以在await这类方法前面与signal/signalAll一样预检查一次持锁情况:
if (!isHeldExclusively())
throw new IllegalMonitorStateException();
整体而言,ConditionObject中代码对GC友好，逻辑缜密，读过之后受益匪浅。
