前言
AQS在jdk17之后提取了一个cleanQueue()方法,这个方法写得真是匠心独运,我们一起来欣赏吧,
本文基于jdk-17.0.3.1版本,相对于jdk8,jdk11,jdk17对AQS做了重构优化,代码更内聚。
cleanQueue做了什么
看看方法描述:
Possibly repeatedly traverses from tail, unsplicing cancelled nodes until none are found. Unparks nodes that may have been relinked to be next eligible acquirer.
翻译出来其实干了2件事情:
- 循环地从同步队列中移除已经被取消的node
- unpark那些relinked的nodes
AQS队列结构
AQS队列是Node节点的双向循环链表,Node包含prev,next,status,waiter属性。
cleanQueue有几个特点:
- AQS队列Node节点status为负数就是被取消
- Node节点status可能被其他线程改变
- 节点取消和节点添加会改变队列,并且可能并发发生的
- 取消的节点可能是head节点,可能是tail节点,也可能是中间的节点
- 是从head向tail循环还是从tail向head循环?
咱们下面来看看jdk17的实现,主要是使用Usafe类原子操作和双重循环
源码分析
private void cleanQueue() {
for (;;) { // restart point
//1.内循环break之后程序的位置
for (Node q = tail, s = null, p, n;;) { // (p, q, s) triples
//2.内循环下次循环程序的位置
if (q == null || (p = q.prev) == null)
return; // end of list
if (s == null ? tail != q : (s.prev != q || s.status < 0))
break; // inconsistent
if (q.status < 0) { // cancelled
if ((s == null ? casTail(q, p) : s.casPrev(q, p)) &&
q.prev == p) {
p.casNext(q, s); // OK if fails
if (p.prev == null)
signalNext(p);
}
break;
}
if ((n = p.next) != q) { // help finish
if (n != null && q.prev == p) {
p.casNext(n, q);
if (p.prev == null)
signalNext(p);
}
break;
}
s = q;
q = q.prev;
}
}
}
cleanQueue选择的是从tail向head循环,因为前面的节点可能马上获取执行机会。
方法开始前AQS队列如下:
(1)
执行q = tail,此时q指向了tail
(2)
如果此时q==null或者q.prev为空,这时候代表前面没有节点了,执行return 程序退出,这是程序出口
如果继续向下执行,此时条件一定是执行 q!= null && (p = q.prev) == null,此时p指向q的prev
(3)
判断s==null ?此时s为空 , 判断 tail != q ?
如果不等,代表q已经不指向tail了,即其他线程已经并发修改了tail节点,执行break,跳到1.重新内循环
如果相等,代表没有线程修改tail,可以继续向下执行
(4)
q.status < 0 ?
如果为false,表示q未被取消,程序继续向下执行。
如果为true,表示q已被取消,将q移除队列,修改tail为p。无论if为真为假, 都执行break;跳到1.重新内循环
将q移除队列:casTail(q, p),此方法通过Unsafe原子处理:在队列tail为q的条件下修改tail为p
判断q.prev == p,为true,才执行p.casNext(q, s),即将p的next指向s(null),
执行if (p.prev == null) signalNext§< 即判断如果p没有prev,如果p的prev为null,唤醒p.next
此时和第一次进入内循环是状态一样
(5)
执行n = p.next
n != q ?有的小伙伴可能疑惑了,这不是相等的吗?
其实如果是单线程没有并发,结果为真。但是AQS是多线程并发,有可能tail已经被修改过了,n和q不相等了,如同就是下面指向:
此时需要变为cleanQueue需要的状态,如下:
变化步骤如下:
如果n == q ,继续向下执行,
如果n != q && n!=null && q.prev == p, 既n不为空,且q的prev为p的条件下,执行p.casNext(n, q),原子地改变p的next为q。如果p.prev == null ,唤醒p.next。 最后break; 跳到1.重新内循环
(6)
执行s = q
执行q = q.prev
继续下一次内循环 跳到2. 也就是说本次循环没有找到取消的节点,往head方向继续
(7)
如果此时q==null或者q.prev为空,这时候代表前面没有节点了,执行return 程序退出,这是程序出口
如果继续向下执行,此时条件一定是执行 q!= null && (p = q.prev) == null,此时p指向q的prev
(8)
此时s!=null 判断 s.prev != q || s.status < 0 ?
如果true,代表s的prev变了或者s被取消,执行break,跳到1.重新内循环
如果false,代表s的prev没变,并且s未被取消,可以继续向下执行
(9)
q.status < 0 ?
如果为false,表示q未被取消,程序继续向下执行。
如果为true,表示q已被取消,将q移除队列。无论if为真为假, 都执行break;跳到1.重新内循环
将q移除队列:判断s == null? false,执行s.casPrev(q, p),在q.prev == p && s的prev为q的条件 下原子地设置s的prev为p,并修改p的next为s,修改完成如下图:
(10)
执行n = p.next
此流程和前面分析一样
(11)
执行s = q
执行q = q.prev
此时就和执行完(6)之后一样了,剩下的就继续内循环,直到程序提出。
总结
可以看到jdk17使用了一个双循环,直到没有取消的节点为止,大量使用了Unsafe的原子类操作,
如果Unsafe修改失败,会再次循环。
在修改tail,prev,next节点是都使用了Unsafe,如果中间发现节点指向变了,会再次循环。