经过前面几篇文章的铺垫,今天我们终于要看看AQS
的庐山真面目了,建议第一次看AbstractQueuedSynchronizer
类源码的朋友可以先看下我前面几篇文章:
分析源码是非常枯燥乏味的一件事,其实代码本身其实就是最好的说明了,因此基本都是贴出一些代码加上一些注释, 因为AbstractQueuedSynchronizer
上千行代码不可能也不需要每行都要分析,所以只捡一些关键的地方或 比较难理解的地方做说明,有一些地方可能我理解的有出入,欢迎大家指正。 详细的注释我都放在了GitHub上。
前面提到AQS
是基于CLH lock queue的,AbstractQueuedSynchronizer
是通过一个内部类Node
实现了一个变种。 前面基本说明了Node的主要内容,但这个类还有一些其他重要的字段:
//标记当前结点是共享模式 static final Node SHARED = new Node(); //标记当前结点是独占模式 static final Node EXCLUSIVE = null; //结点的等待状态。 volatile int waitStatus; //拥有当前结点的线程。 volatile Thread thread;
其中waitStatus
很重要,用来控制线程的阻塞/唤醒,以及避免不必要的调用LockSupport的park/unpark方法。 它主要有以下几个取值:
//代表线程已经被取消 static final int CANCELLED = 1; //代表后续节点需要唤醒 static final int SIGNAL = -1; //代表线程在condition queue中,等待某一条件 static final int CONDITION = -2; //代表后续结点会传播唤醒的操作,共享模式下起作用 static final int PROPAGATE = -3;
出队操作
只要设置新的head
结点就可以了。
private void setHead(Node node) { head = node; node.thread = null; node.prev = null; }
入队操作
private Node addWaiter(Node mode) { Node node = new Node(Thread.currentThread(), mode); // Try the fast path of enq; backup to full enq on failure Node pred = tail; // 这个if分支其实是一种优化:CAS操作失败的话才进入enq中的循环。 if (pred != null) { node.prev = pred; if (compareAndSetTail(pred, node)) { pred.next = node; return node; } } enq(node); return node; } private Node enq(final Node node) { for (;;) { Node t = tail; if (t == null) { // Must initialize if (compareAndSetHead(new Node())) tail = head; } else { node.prev = t; if (compareAndSetTail(t, node)) { t.next = node; return t; } } } }
独占模式获取
public final void acquire(int arg) { // tryAcquire 由子类实现本身不会阻塞线程,如果返回 true,则线程继续, // 如果返回 false 那么就 加入阻塞队列阻塞线程,并等待前继结点释放锁。 if (!tryAcquire(arg) && acquireQueued(addWaiter(Node.EXCLUSIVE), arg)) // acquireQueued返回true,说明当前线程被中断唤醒后获取到锁, // 重置其interrupt status为true。 selfInterrupt(); }
一旦tryAcquire成功则立即返回,否则线程会加入队列 线程可能会反复的被阻塞和唤醒直到tryAcquire成功,这是因为线程可能被中断, 而acquireQueued方法中会保证忽视中断,只有tryAcquire成功了才返回。中断版本的独占获取是acquireInterruptibly
这个方法, doAcquireInterruptibly
这个方法中如果线程被中断则acquireInterruptibly
会抛出InterruptedException
异常。
addWaiter
方法只是入队操作,acquireQueued
方法是主要逻辑,需要重点理解。
final boolean acquireQueued(final Node node, int arg) { boolean failed = true; try { boolean interrupted = false; // 等待前继结点释放锁 // 自旋re-check for (;;) { // 获取前继 final Node p = node.predecessor(); // 前继是head,说明next就是node了,则尝试获取锁。 if (p == head && tryAcquire(arg)) { // 前继出队,node成为head setHead(node); p.next = null; // help GC failed = false; return interrupted; } // p != head 或者 p == head但是tryAcquire失败了,那么 // 应该阻塞当前线程等待前继唤醒。阻塞之前会再重试一次,还需要设置前继的waitStaus为SIGNAL。 // 线程会阻塞在parkAndCheckInterrupt方法中。 // parkAndCheckInterrupt返回可能是前继unpark或线程被中断。 if (shouldParkAfterFailedAcquire(p, node) && parkAndCheckInterrupt()) // 说明当前线程是被中断唤醒的。 // 注意:线程被中断之后会继续走到if处去判断,也就是会忽视中断。 // 除非碰巧线程中断后acquire成功了,那么根据Java的最佳实践, // 需要重新设置线程的中断状态(acquire.selfInterrupt)。 interrupted = true; } } finally { // 出现异常 if (failed) cancelAcquire(node); } }
基本每行都有注释,但得结合shouldParkAfterFailedAcquire
和parkAndCheckInterrupt
这两个方法来一起理解会更 容易些。shouldParkAfterFailedAcquire方法的作用是:
- 确定后继是否需要park;
- 跳过被取消的结点;
- 设置前继的waitStatus为SIGNAL.
int ws = pred.waitStatus; if (ws == Node.SIGNAL)// 前继结点已经准备好unpark其后继了,所以后继可以安全的park /* * This node has already set status asking a release to signal it, * so it can safely park. */ return true; if (ws > 0) {// CANCELLED // 跳过被取消的结点。 do { node.prev = pred = pred.prev; } while (pred.waitStatus > 0); pred.next = node; } else {// 0 或 PROPAGATE (CONDITION用在ConditonObject,这里不会是这个值) /** * waitStatus 等于0(初始化)或PROPAGATE。说明线程还没有park,会先重试 * 确定无法acquire到再park。 */ // 更新pred结点waitStatus为SIGNAL compareAndSetWaitStatus(pred, ws, Node.SIGNAL); } return false;
parkAndCheckInterrupt就是用LockSupport来阻塞当前线程,很简单:
private final boolean parkAndCheckInterrupt() { LockSupport.park(this); return Thread.interrupted(); }
线程被唤醒只可能是:被unpark
,被中断或伪唤醒。被中断会设置interrupted
,acquire方法返回前会 selfInterrupt
重置下线程的中断状态,如果是伪唤醒的话会for循环re-check。
独占模式释放
比较简单只要直接唤醒后续结点就可以了,后续结点会从parkAndCheckInterrupt
方法中返回。
public final boolean release(int arg) { // tryReease由子类实现,通过设置state值来达到同步的效果。 if (tryRelease(arg)) { Node h = head; // waitStatus为0说明是初始化的空队列 if (h != null && h.waitStatus != 0) // 唤醒后续的结点 unparkSuccessor(h); return true; } return false; }
共享模式获取
acquireShared
方法是用来共享模式获取。
public final void acquireShared(int arg) { //如果没有许可了则入队等待 if (tryAcquireShared(arg) < 0) doAcquireShared(arg); } private void doAcquireShared(int arg) { // 添加队列 final Node node = addWaiter(Node.SHARED); boolean failed = true; try { boolean interrupted = false; // 等待前继释放并传递 for (;;) { final Node p = node.predecessor(); if (p == head) { int r = tryAcquireShared(arg);// 尝试获取 if (r >= 0) { // 获取成功则前继出队,跟独占不同的是 // 会往后面结点传播唤醒的操作,保证剩下等待的线程能够尽快 获取到剩下的许可。 setHeadAndPropagate(node, r); p.next = null; // help GC if (interrupted) selfInterrupt(); failed = false; return; } } // p != head || r < 0 if (shouldParkAfterFailedAcquire(p, node) && parkAndCheckInterrupt()) interrupted = true; } } finally { if (failed) cancelAcquire(node); } }
核心是这个doAcquireShared
方法,跟独占模式的acquireQueued
很像,主要区别在setHeadAndPropagate
方法中, 这个方法会将node设置为head。如果当前结点acquire到了之后发现还有许可可以被获取,则继续释放自己的后继, 后继会将这个操作传递下去。这就是PROPAGATE
状态的含义。
private void setHeadAndPropagate(Node node, int propagate) { Node h = head; // Record old head for check below setHead(node); /* * 尝试唤醒后继的结点:<br /> * propagate > 0说明许可还有能够继续被线程acquire;<br /> * 或者 之前的head被设置为PROPAGATE(PROPAGATE可以被转换为SIGNAL)说明需要往后传递;<br /> * 或者为null,我们还不确定什么情况。 <br /> * 并且 后继结点是共享模式或者为如上为null。 * <p> * 上面的检查有点保守,在有多个线程竞争获取/释放的时候可能会导致不必要的唤醒。<br /> * */ if (propagate > 0 || h == null || h.waitStatus < 0) { Node s = node.next; // 后继结是共享模式或者s == null(不知道什么情况) // 如果后继是独占模式,那么即使剩下的许可大于0也不会继续往后传递唤醒操作 // 即使后面有结点是共享模式。 if (s == null || s.isShared()) // 唤醒后继结点 doReleaseShared(); } } private void doReleaseShared() { for (;;) { Node h = head; // 队列不为空且有后继结点 if (h != null && h != tail) { int ws = h.waitStatus; // 不管是共享还是独占只有结点状态为SIGNAL才尝试唤醒后继结点 if (ws == Node.SIGNAL) { // 将waitStatus设置为0 if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0)) continue; // loop to recheck cases unparkSuccessor(h);// 唤醒后继结点 // 如果状态为0则更新状态为PROPAGATE,更新失败则重试 } else if (ws == 0 && !compareAndSetWaitStatus(h, 0, Node.PROPAGATE)) continue; // loop on failed CAS } // 如果过程中head被修改了则重试。 if (h == head) // loop if head changed break; } }
共享模式释放
主要逻辑也就会doReleaseShared
。
public final boolean releaseShared(int arg) { if (tryReleaseShared(arg)) { doReleaseShared(); return true; } return false; }
独占和共享模式除了对应的中断版本,还有超时版本,整体代码相差不大,具体再赘述了。提前前面文章 中提到的自旋
,好像目前整个AQS中都没用到这个功能,accquire中for循环主要作用不是为了自旋,那么 它用在什么地方呢?AQS中有一个变量:
static final long spinForTimeoutThreshold = 1000L;
这个变量用在doAcquireNanos
方法,也就是支持超时的获取版本。
private boolean doAcquireNanos(int arg, long nanosTimeout) throws InterruptedException { long lastTime = System.nanoTime(); final Node node = addWaiter(Node.EXCLUSIVE); boolean failed = true; try { for (;;) { final Node p = node.predecessor(); if (p == head && tryAcquire(arg)) { setHead(node); p.next = null; // help GC failed = false; return true; } if (nanosTimeout <= 0)// 超时 return false; // nanosTimeout > spinForTimeoutThreshold // 如果超时时间很短的话,自旋效率会更高。 if (shouldParkAfterFailedAcquire(p, node) && nanosTimeout > spinForTimeoutThreshold) LockSupport.parkNanos(this, nanosTimeout); long now = System.nanoTime(); nanosTimeout -= now - lastTime; lastTime = now; if (Thread.interrupted()) throw new InterruptedException(); } } finally { if (failed) cancelAcquire(node); } }
AQS的的主要内容其实差不多看完了,但是上面的逻辑中waitStatus中有一个状态还没涉及到那就是CONDITION
, 下一篇博客《Java并发包源码学习之AQS框架(五)ConditionObject源码分析》中会介绍它。