最近一直在研究AQS的源码,希望可以更深刻的理解AQS的实现原理。虽然网上有很多关于AQS的源码分析,但是看完以后感觉还是一知半解。于是,我将自己的整个理解过程记录下来了,希望对大家有所帮助。

基本原理

AQS是Java中锁的基础,主要由两个队列组成。一个队列是同步队列,另一个是条件队列

同步队列的原理

  • 同步队列的队列头部是head,队列尾部是tail节点,head节点是一个空节点,同步队列是一个双向链表,通过nextprev连接所有节点
  • 所有的线程在竞争锁的时候都会创建一个Node节点,线程与节点绑定在一起,(如果是同步锁和排他锁不同之处是通过nextWaiter来区分的)并且添加到同步队列的尾部
  • head的第一个节点获取锁,其余节点都需要等待被唤醒
  • 同步队列中的节点会存在取消和null的情况(如:线程超时中断、线程更新节点的中间态),被取消和null的节点不能被唤醒,将会被视为无效节点
  • 一个线程只能被有效的前驱节点(取消和null的节点除外)唤醒
  • 持有锁的线程只能是有一个,其他有效节点对应的线程都会被挂起

条件队列的原理

  • 一个同步队列可以对应多个条件队列
  • 条件队列是一个单向链表,通过nextWaiter来连接起来,条件队列的头节点是firstWaiter,尾节点是lastWaiter
  • 某个条件队列中满足条件的节点(被signalsignalAll方法唤醒的节点)才会被转移到同步队列
  • 条件队列中的被转移到同步队列的节点是从头节点开始,条件队列中被阻塞的线程会添加到队列的尾部

同步队列的实现

首先,了解以下同步队列中队列的节点Node的数据结构

  1. static final class Node {
  2. /** 共享锁的标识 */
  3. static final Node SHARED = new Node();
  4. /** 排他锁的标识 */
  5. static final Node EXCLUSIVE = null;
  6. /** 线程取消 */
  7. static final int CANCELLED = 1;
  8. /** 持有锁的线程的后继线程被挂起 */
  9. static final int SIGNAL = -1;
  10. /** 条件队列标识 */
  11. static final int CONDITION = -2;
  12. /**
  13. * 共享锁情况下,通知所有其他节点
  14. */
  15. static final int PROPAGATE = -3;
  16. /**
  17. * waitStatus的取值如下:
  18. * SIGNAL(-1): 当前节点的后继节点应该被挂起
  19. * CANCELLED(1): 当前节点被取消
  20. * CONDITION(-2): 当前节点在条件队列
  21. * PROPAGATE(-3): 释放共享锁时需要通知所有节点
  22. * 0: 初始值
  23. *
  24. */
  25. volatile int waitStatus;
  26. /**
  27. * 前驱节点
  28. */
  29. volatile Node prev;
  30. /**
  31. * 后继节点
  32. */
  33. volatile Node next;
  34. /**
  35. * 节点对应的线程
  36. */
  37. volatile Thread thread;
  38. /**
  39. * 在共享锁的情况下,该节点的值为SHARED
  40. * 在排他锁的情况下,该节点的值为EXCLUSIVE
  41. * 在条件队列的情况下,链接的是下一个等待条件的线程
  42. */
  43. Node nextWaiter;
  44. }

其次,我们来看一下同步队列的链表结构

接着,我们根据同步队列的原理来分析以下acquirerelease需要做哪些事情:

实现acquire功能需要做的事情

  1. 创建一个Node节点node(该节点可能是排他锁,也可以能是共享锁)
  2. node添加到同步队列尾部,如果同步队列为空(初始情况下),需要先创建一个空的头节点,然后再添加到队列的尾部
  3. 如果node的前驱节点是head,说明node是第一个节点,能够获取锁,需要将head修改成node,释放前驱节点的资源
  4. 如果node的前驱节点不是head,说明获取锁失败,需要检测是否需要将node绑定的线程挂起,分以下几种情况:
    • 如果nodewaitStatus已经被设置为SIGNAL 表示需要被挂起
    • 如果nodewaitStatus设置为CANCEL表示该节点已经被取消,需要被去掉,并修改 nodeprev,直到链接上一个有效的节点为止
    • 否则将nodewaitStatus设置为SIGNAL,表示即将要被挂起
  5. 如果需要将node绑定的线程挂起,则让出CPU,直到当前驱节点来唤起node才会开始继续从步骤3开始执行

与acquire功能相关的代码

  • acquire方法:获取排他锁
  1. public final void acquire(int arg) {
  2. if (!tryAcquire(arg) &&
  3. acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
  4. selfInterrupt();
  5. }
  1. tryAcquire(arg):对外提供的一个扩展方法,常用的锁都要实现这个方法,具体实现与锁相关

  2. addWaiter(Node.EXCLUSIVE): 创建一个排他锁节点,并将该节点添加到同步队列尾部,代码如下:

  1. private Node addWaiter(Node mode) {
  2. // 创建一个node,EXCLUSIVE类型
  3. Node node = new Node(mode);
  4. for (;;) {
  5. // 获取尾节点
  6. Node oldTail = tail;
  7. if (oldTail != null) {
  8. // 设置即将成为尾节点的前驱
  9. node.setPrevRelaxed(oldTail);
  10. // CAS操作设置尾节点
  11. if (compareAndSetTail(oldTail, node)) {
  12. // 将新尾节点的前驱节点与新的尾节点关联起来
  13. oldTail.next = node;
  14. // 返回添加的节点
  15. // 这个节点现在不一定是尾节点,因为如果有多个线程调用这个方法时,
  16. // 可能还有节点添加在这个节点后面
  17. return node;
  18. }
  19. } else {
  20. // 如果队列为空,初始化头节点
  21. initializeSyncQueue();
  22. }
  23. }
  24. }
  1. acquireQueued同步队列中的节点获取排他锁
  1. final boolean acquireQueued(final Node node, int arg) {
  2. try {
  3. // 线程是否中断
  4. boolean interrupted = false;
  5. for (;;) {
  6. // 获取前驱节点
  7. final Node p = node.predecessor();
  8. // 如果前驱节点是头节点,获取锁
  9. if (p == head && tryAcquire(arg)) {
  10. // 修改头节点
  11. setHead(node);
  12. // 释放头节点的资源
  13. p.next = null; // help GC
  14. // 返回线程中断的状态
  15. // 这也是该方法唯一的返回值
  16. // 没有获取锁的线程会一直执行该方法直到获取锁以后再返回
  17. return interrupted;
  18. }
  19. // 获取锁失败后是否需要将线程挂起
  20. if (shouldParkAfterFailedAcquire(p, node) &&
  21. parkAndCheckInterrupt()) // 线程挂起并返回是否被中断
  22. interrupted = true;
  23. }
  24. } catch (Throwable t) {
  25. // 取消该节点
  26. cancelAcquire(node);
  27. throw t;
  28. }
  29. }
  1. shouldParkAfterFailedAcquire:检测线程获取锁失败以后是否需要被挂起
  1. private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
  2. // 前驱节点的状态
  3. int ws = pred.waitStatus;
  4. if (ws == Node.SIGNAL)
  5. /*
  6. * 状态已经设置成SIGNAL,可以直接挂起该节点
  7. */
  8. return true;
  9. // 节点被取消
  10. if (ws > 0) {
  11. /*
  12. * 找到pred第一个有效的前驱节点
  13. */
  14. do {
  15. node.prev = pred = pred.prev;
  16. } while (pred.waitStatus > 0);
  17. // pred可能是一个新的节点,需要将pred的next重写设置为node
  18. pred.next = node;
  19. } else {
  20. /*
  21. * CAS操作将pred节点的状态设置为SIGNAL
  22. */
  23. pred.compareAndSetWaitStatus(ws, Node.SIGNAL);
  24. }
  25. // 只有当pred节点的waitStatus已经是SIGNAL状态时,才可以安全的挂起线程
  26. // 否则需要不能被挂起
  27. return false;
  28. }
  1. parkAndCheckInterrupt:将当前线程挂起,并检测当前线程是否中断
  1. private final boolean parkAndCheckInterrupt() {
  2. // 线程挂起
  3. LockSupport.park(this);
  4. // 检测线程是否中断
  5. return Thread.interrupted();
  6. }
  1. cancelAcquire:取消节点
  1. private void cancelAcquire(Node node) {
  2. // 如果节点为空,什么都不做
  3. if (node == null)
  4. return;
  5. // 释放线程
  6. node.thread = null;
  7. // 从后往前过滤掉所有的被取消的节点
  8. Node pred = node.prev;
  9. while (pred.waitStatus > 0)
  10. node.prev = pred = pred.prev;
  11. // 有效前驱节点的nex节点
  12. Node predNext = pred.next;
  13. // 将node设置为CANCELLED
  14. node.waitStatus = Node.CANCELLED;
  15. // 如果是尾节点,设置新的尾节点
  16. if (node == tail && compareAndSetTail(node, pred)) {
  17. // 将新的尾节点的后续设置为null
  18. pred.compareAndSetNext(predNext, null);
  19. } else {
  20. // If successor needs signal, try to set pred's next-link
  21. // so it will get one. Otherwise wake it up to propagate.
  22. int ws;
  23. // 如果前驱节点的线程不为null并且waitStatus为SIGNAL
  24. if (pred != head &&
  25. ((ws = pred.waitStatus) == Node.SIGNAL ||
  26. (ws <= 0 && pred.compareAndSetWaitStatus(ws, Node.SIGNAL))) &&
  27. pred.thread != null) {
  28. Node next = node.next;
  29. // 将node设置成pred的后继节点
  30. if (next != null && next.waitStatus <= 0)
  31. pred.compareAndSetNext(predNext, next);
  32. } else {
  33. // 唤起node节点的后继节点
  34. // 因为node节点已经释放锁了
  35. unparkSuccessor(node);
  36. }
  37. node.next = node; // help GC
  38. }
  39. }
  1. unparkSuccessor:唤醒后继节点
  1. private void unparkSuccessor(Node node) {
  2. /*
  3. * 获取node节点的waitStatus
  4. */
  5. int ws = node.waitStatus;
  6. // 用CSA操作将waitStatus设置成初始状态
  7. // 不管设置是否成功,都无所谓,因为该节点即将被销毁
  8. if (ws < 0)
  9. node.compareAndSetWaitStatus(ws, 0);
  10. /*
  11. * 获取node的后继节点
  12. */
  13. Node s = node.next;
  14. // 如果后继节点为null或者被取消,
  15. // 通过从同步队列的尾节点开始一直往前找到一个有效的后继节点
  16. if (s == null || s.waitStatus > 0) {
  17. s = null;
  18. for (Node p = tail; p != node && p != null; p = p.prev)
  19. if (p.waitStatus <= 0)
  20. s = p;
  21. }
  22. // 如果后继节点不为空
  23. if (s != null)
  24. LockSupport.unpark(s.thread);// 唤醒后继节点的线程
  25. }

acquire方法类似的还有acquireInterruptiblytryAcquireNanosacquireSharedacquireSharedInterruptiblytryAcquireSharedNanos,我们都一一分析以下

  • acquireInterruptibly方法:获取可中断的排他锁
  1. public final void acquireInterruptibly(int arg)
  2. throws InterruptedException {
  3. if (Thread.interrupted()) // 如果线程中断,直接返回
  4. throw new InterruptedException();
  5. if (!tryAcquire(arg))
  6. doAcquireInterruptibly(arg); // 中断式的获取锁
  7. }
  1. doAcquireInterruptibly:可中断式的获取锁
  1. private void doAcquireInterruptibly(int arg)
  2. throws InterruptedException {
  3. // 创建一个排他节点加入同步队列
  4. final Node node = addWaiter(Node.EXCLUSIVE);
  5. try {
  6. for (;;) {
  7. // 获取前驱节点
  8. final Node p = node.predecessor();
  9. // 如果前驱节点是头节点,说明已经获取的锁
  10. if (p == head && tryAcquire(arg)) {
  11. // 修改头节点
  12. setHead(node);
  13. p.next = null; // help GC
  14. return;
  15. }
  16. // 如果没有获取锁,检测是否需要挂起
  17. if (shouldParkAfterFailedAcquire(p, node) &&
  18. parkAndCheckInterrupt())
  19. throw new InterruptedException(); // 如果发现线程已经被中断,需要抛出异常
  20. }
  21. } catch (Throwable t) {
  22. // 发生异常取消节点
  23. cancelAcquire(node);
  24. throw t;
  25. }
  26. }
  • tryAcquireNanos方法:超时中断获取排他锁
  1. public final boolean tryAcquireNanos(int arg, long nanosTimeout)
  2. throws InterruptedException {
  3. if (Thread.interrupted())
  4. throw new InterruptedException(); // 线程中断直接返回
  5. return tryAcquire(arg) ||
  6. doAcquireNanos(arg, nanosTimeout); // 超时获取排他锁
  7. }
  1. doAcquireNanos:超时获取排他锁
  1. private boolean doAcquireNanos(int arg, long nanosTimeout)
  2. throws InterruptedException {
  3. // 如果超时直接返回
  4. if (nanosTimeout <= 0L)
  5. return false;
  6. // 获取超时时长
  7. final long deadline = System.nanoTime() + nanosTimeout;
  8. // 添加一个排他节点到同步队列尾部
  9. final Node node = addWaiter(Node.EXCLUSIVE);
  10. try {
  11. for (;;) {
  12. // 获取前驱节点
  13. final Node p = node.predecessor();
  14. // 已经获取锁
  15. if (p == head && tryAcquire(arg)) {
  16. setHead(node);
  17. p.next = null; // help GC
  18. return true;
  19. }
  20. nanosTimeout = deadline - System.nanoTime();
  21. // 如果超时了就取消
  22. if (nanosTimeout <= 0L) {
  23. cancelAcquire(node);
  24. return false;
  25. }
  26. // 检测节点是否需要被挂起
  27. if (shouldParkAfterFailedAcquire(p, node) &&
  28. nanosTimeout > SPIN_FOR_TIMEOUT_THRESHOLD)
  29. // 如果需要挂起,且超时时长大于SPIN_FOR_TIMEOUT_THRESHOLD
  30. // 线程挂起nanosTimeout时间
  31. LockSupport.parkNanos(this, nanosTimeout);
  32. if (Thread.interrupted())
  33. throw new InterruptedException();
  34. }
  35. } catch (Throwable t) {
  36. // 发生异常取消节点
  37. cancelAcquire(node);
  38. throw t;
  39. }
  40. }
  • acquireShared方法:获取共享锁
  1. public final void acquireShared(int arg) {
  2. // 对外提供的一个扩展方法,常用的锁都要实现这个方法,
  3. // 该方法的实现与锁的用途有关
  4. if (tryAcquireShared(arg) < 0)
  5. doAcquireShared(arg); // 获取共享锁
  6. }
  1. doAcquireShared:获取共享锁
  1. private void doAcquireShared(int arg) {
  2. // 添加一个共享节点到同步队列尾部
  3. final Node node = addWaiter(Node.SHARED);
  4. try {
  5. boolean interrupted = false;
  6. for (;;) {
  7. // 获取前驱节点
  8. final Node p = node.predecessor();
  9. if (p == head) {
  10. // 返回结果大于等于0表示获取共享锁
  11. int r = tryAcquireShared(arg);
  12. if (r >= 0) {
  13. // 设置头节点并广播通知其他获取共享锁的节点
  14. setHeadAndPropagate(node, r);
  15. p.next = null; // help GC
  16. // 如果线程被中断,将该线程中断
  17. // 共享锁会被多个线程获取,如果需要中断
  18. // 所有获取共享锁的线程都要被中断
  19. if (interrupted)
  20. selfInterrupt();
  21. return;
  22. }
  23. }
  24. // 检测是否需要挂起
  25. if (shouldParkAfterFailedAcquire(p, node) &&
  26. parkAndCheckInterrupt()) // 挂起并中断
  27. interrupted = true;
  28. }
  29. } catch (Throwable t) {
  30. // 发生异常取消节点
  31. cancelAcquire(node);
  32. throw t;
  33. }
  34. }
  1. setHeadAndPropagate:设置头节点并广播其他节点来获取锁
  1. private void setHeadAndPropagate(Node node, int propagate) {
  2. Node h = head; // 记录旧的头节点
  3. setHead(node);// 设置新的头节点
  4. /*
  5. * 如果头节点为null或者是不是取消状态,尝试唤醒后继节点
  6. */
  7. if (propagate > 0 || h == null || h.waitStatus < 0 ||
  8. (h = head) == null || h.waitStatus < 0) {
  9. Node s = node.next;
  10. // node节点的next是SHARED,即共享锁
  11. if (s == null || s.isShared())
  12. // 唤起获取共享锁的线程
  13. doReleaseShared();
  14. }
  15. }
  1. doReleaseShared:唤醒等待共享锁的节点
  1. private void doReleaseShared() {
  2. /*
  3. * 唤醒时是从头节点开始先唤醒第一个共享节点,
  4. * 第一个共享节点被唤醒后会在doAcquireShared方法里继续执行(之前就是在这个方法里被挂起的)
  5. * 第一个共享节点如果获取锁会调用setHeadAndPropagate方法修改头节点,然后再调用doReleaseShared方法
  6. * 唤醒第二个共享节点,以此类推,最后把所有的共享节点都唤醒
  7. */
  8. for (;;) {
  9. Node h = head;
  10. if (h != null && h != tail) {
  11. // 获取头节点的状态
  12. int ws = h.waitStatus;
  13. // 如果头节点是SIGNAL,需要将状态设置为0,表示已经即将被唤醒
  14. if (ws == Node.SIGNAL) {
  15. if (!h.compareAndSetWaitStatus(Node.SIGNAL, 0))
  16. continue; // 如果失败了说明有其他线程在修改头节点,需要继续重试
  17. unparkSuccessor(h); // 唤醒头节点的后继节点
  18. }
  19. else if (ws == 0 &&
  20. !h.compareAndSetWaitStatus(0, Node.PROPAGATE))
  21. continue; // 将头节点状态从0设置成PROPAGATE,如果失败了继续,因为也有其他获取共享锁的线程在更改头节点
  22. }
  23. // 如果头节点未改变(因为没有后继节点需要等待共享锁),跳出循环
  24. if (h == head)
  25. break;
  26. }
  27. }
  1. selfInterrupt:中断当前线程
  1. static void selfInterrupt() {
  2. Thread.currentThread().interrupt();
  3. }
  • acquireSharedInterruptibly方法:可中断的获取共享锁
  1. public final void acquireSharedInterruptibly(int arg)
  2. throws InterruptedException {
  3. if (Thread.interrupted())
  4. throw new InterruptedException(); // 如果线程被中断抛出异常
  5. if (tryAcquireShared(arg) < 0)
  6. doAcquireSharedInterruptibly(arg); // 可中断的方式获取共享锁
  7. }
  1. doAcquireSharedInterruptibly:可中断的方式后去共享锁
  1. private void doAcquireSharedInterruptibly(int arg)
  2. throws InterruptedException {
  3. // 添加共享锁节点到同步队列尾部
  4. final Node node = addWaiter(Node.SHARED);
  5. try {
  6. for (;;) {
  7. // 获取前驱节点
  8. final Node p = node.predecessor();
  9. if (p == head) {
  10. int r = tryAcquireShared(arg);
  11. if (r >= 0) {
  12. // 获取共享锁以后修改头节点,通知其他等待共享锁的节点
  13. setHeadAndPropagate(node, r);
  14. p.next = null; // help GC
  15. return;
  16. }
  17. }
  18. // 线程获取共享锁失败后需要挂起,并且发现线程被中断,所以抛出异常
  19. if (shouldParkAfterFailedAcquire(p, node) &&
  20. parkAndCheckInterrupt())
  21. throw new InterruptedException();
  22. }
  23. } catch (Throwable t) {
  24. // 发生异常取消节点
  25. cancelAcquire(node);
  26. throw t;
  27. }
  28. }
  • tryAcquireSharedNanos方法:超时中断获取共享锁
  1. public final boolean tryAcquireSharedNanos(int arg, long nanosTimeout)
  2. throws InterruptedException {
  3. if (Thread.interrupted()) // 线程如果中断了,直接抛出异常
  4. throw new InterruptedException();
  5. return tryAcquireShared(arg) >= 0 ||
  6. doAcquireSharedNanos(arg, nanosTimeout); // 超时获取共享锁
  7. }
  1. doAcquireSharedNanos:超时的方式获取中断锁
  1. private boolean doAcquireSharedNanos(int arg, long nanosTimeout)
  2. throws InterruptedException {
  3. // 超时直接返回
  4. if (nanosTimeout <= 0L)
  5. return false;
  6. final long deadline = System.nanoTime() + nanosTimeout;
  7. // 添加共享节点到同步队列尾部
  8. final Node node = addWaiter(Node.SHARED);
  9. try {
  10. for (;;) {
  11. // 获取前驱节点
  12. final Node p = node.predecessor();
  13. if (p == head) {
  14. int r = tryAcquireShared(arg);
  15. if (r >= 0) {
  16. // 获取锁,修改头节点,通知所有其他等待共享锁的节点
  17. setHeadAndPropagate(node, r);
  18. p.next = null; // help GC
  19. return true;
  20. }
  21. }
  22. nanosTimeout = deadline - System.nanoTime();
  23. if (nanosTimeout <= 0L) {
  24. // 超时取消节点
  25. cancelAcquire(node);
  26. return false;
  27. }
  28. if (shouldParkAfterFailedAcquire(p, node) &&
  29. nanosTimeout > SPIN_FOR_TIMEOUT_THRESHOLD)
  30. // 如果需要挂起,且超时时长大于SPIN_FOR_TIMEOUT_THRESHOLD
  31. // 线程挂起nanosTimeout时间
  32. LockSupport.parkNanos(this, nanosTimeout);
  33. if (Thread.interrupted())
  34. throw new InterruptedException(); // 中断了抛出异常
  35. }
  36. } catch (Throwable t) {
  37. // 发生异常取消节点
  38. cancelAcquire(node);
  39. throw t;
  40. }
  41. }

实现release功能需要做的事情

  1. 释放当前获取锁的线程持有的资源
  2. 唤醒有效的一个后继节点

与release功能相关的代码

  • release方法:释放排他锁
  1. public final boolean release(int arg) {
  2. if (tryRelease(arg)) {
  3. Node h = head;
  4. // 头节点不能是一个中间态
  5. if (h != null && h.waitStatus != 0)
  6. // 唤醒后继节点
  7. unparkSuccessor(h);
  8. return true;
  9. }
  10. return false;
  11. }
  • release方法:释放共享锁
  1. public final boolean releaseShared(int arg) {
  2. if (tryReleaseShared(arg)) {
  3. // 释放共享锁,从头节点开始一个一个的释放
  4. // 如果存在多个共享节点在同步队列时,doReleaseShared方式其实是递归调用
  5. doReleaseShared();
  6. return true;
  7. }
  8. return false;
  9. }

至此,将所有获取锁和释放锁的方法相关的源码全部分析完

条件队列的实现

我们来看一下条件队列的链表结构

实现await功能需要做的事情

  1. 创建一个CONDITION类型的节点,将该节点添加到条件队列
  2. 释放已经获取的锁(因为只有当前线程先获取了锁才可能再调用Condition.await()方法)
  3. 如果无法获取锁,线程挂起

与await功能相关的代码

  • await方法:等待条件
  1. public final void await() throws InterruptedException {
  2. if (Thread.interrupted())
  3. throw new InterruptedException(); // 如果线程中断,直接抛出异常
  4. // 创建一个CONDITION类型的节点,将该节点添加到条件队列尾部
  5. Node node = addConditionWaiter();
  6. // 释放锁
  7. // 在调用await方法之前都会调用lock方法,这个时候已经获取锁了
  8. // 有时候锁还是可重入的,所以需要将所有的资源都释放掉
  9. int savedState = fullyRelease(node);
  10. int interruptMode = 0;
  11. // 如果节点不再同步队列,全部都要挂起
  12. while (!isOnSyncQueue(node)) {
  13. LockSupport.park(this);
  14. // 如果在等待期间发生过中断(不管是调用signal之前还是之后),直接退出
  15. if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
  16. break;
  17. }
  18. // 让线程尝试去获取锁,如果无法获取锁就挂起
  19. if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
  20. interruptMode = REINTERRUPT;
  21. // 清除所有在条件队列中是取消状态的节点
  22. if (node.nextWaiter != null) // clean up if cancelled
  23. unlinkCancelledWaiters();
  24. // 发生中断,上报中断情况
  25. if (interruptMode != 0)
  26. reportInterruptAfterWait(interruptMode);
  27. }
  1. addConditionWaiter:在条件队列中添加一个节点
  1. private Node addConditionWaiter() {
  2. Node t = lastWaiter;
  3. // 清除条件队列中无效的节点
  4. if (t != null && t.waitStatus != Node.CONDITION) {
  5. unlinkCancelledWaiters();
  6. t = lastWaiter;
  7. }
  8. // 创建一个节点
  9. Node node = new Node(Node.CONDITION);
  10. // 添加到条件队列尾部
  11. if (t == null)
  12. firstWaiter = node;
  13. else
  14. t.nextWaiter = node;
  15. lastWaiter = node;
  16. return node;
  17. }
  1. unlinkCancelledWaiters:清除在条件队列中被取消的节点
  1. private void unlinkCancelledWaiters() {
  2. Node t = firstWaiter;
  3. Node trail = null;
  4. // 遍历条件队列将所有不是CONDITION状态的节点全部清除掉
  5. // 这些节点都是取消状态的节点
  6. while (t != null) {
  7. Node next = t.nextWaiter;
  8. if (t.waitStatus != Node.CONDITION) {
  9. t.nextWaiter = null;
  10. if (trail == null)
  11. firstWaiter = next;
  12. else
  13. trail.nextWaiter = next;
  14. if (next == null)
  15. lastWaiter = trail;
  16. }
  17. else
  18. trail = t;
  19. t = next;
  20. }
  21. }
  1. fullyRelease:释放线程持有的所有锁资源
  1. final int fullyRelease(Node node) {
  2. try {
  3. int savedState = getState();
  4. // 释放所有的资源
  5. // 如果是可重入锁,savedState就是重入的次数
  6. if (release(savedState))
  7. return savedState;
  8. throw new IllegalMonitorStateException();
  9. } catch (Throwable t) {
  10. // 发生异常就取消该节点
  11. node.waitStatus = Node.CANCELLED;
  12. throw t;
  13. }
  14. }
  1. isOnSyncQueue:判断节点是否在同步队列
  1. final boolean isOnSyncQueue(Node node) {
  2. // waitStatus是CONDITION或者node没有前驱节点,说明node不在同步队列
  3. if (node.waitStatus == Node.CONDITION || node.prev == null)
  4. return false;
  5. if (node.next != null) // 有后继节点一定在同步队列
  6. return true;
  7. /*
  8. * 在同步队列中查找node,看是否在同步队列中
  9. */
  10. return findNodeFromTail(node);
  11. }
  1. findNodeFromTail:在同步队列中查找节点
  1. private boolean findNodeFromTail(Node node) {
  2. // 从尾节点开始查找
  3. for (Node p = tail;;) {
  4. if (p == node) // 找到了
  5. return true;
  6. if (p == null) // 找到头了还没找到
  7. return false;
  8. p = p.prev;
  9. }
  10. }
  1. checkInterruptWhileWaiting:检测中断的情况
  1. private int checkInterruptWhileWaiting(Node node) {
  2. // 没有发生中断返回0
  3. // 调用signal之前发生中断返回THROW_IE
  4. // 调用signal之后发生中断返回REINTERRUPT
  5. return Thread.interrupted() ?
  6. (transferAfterCancelledWait(node) ? THROW_IE : REINTERRUPT) :
  7. 0;
  8. }
  1. transferAfterCancelledWait:清除在条件队列中被取消的节点
  1. // 只有线程处于中断状态,才会调用此方法
  2. // 如果需要的话,将这个已经取消等待的节点转移到阻塞队列
  3. // 返回 true,如果此线程在 signal 之前被取消,否则返回false
  4. final boolean transferAfterCancelledWait(Node node) {
  5. // 用 CAS 将节点状态设置为 0
  6. // 如果这步 CAS 成功,说明是 signal 方法之前发生的中断,
  7. // 因为如果 signal 先发生的话,signal 中会将 waitStatus 设置为 0
  8. if (node.compareAndSetWaitStatus(Node.CONDITION, 0)) {
  9. enq(node); // 将节点放入阻塞队列
  10. return true;
  11. }
  12. // 到这里是因为 CAS 失败,肯定是因为 signal 方法已经将 waitStatus 设置为了 0
  13. // signal 方法会将节点转移到阻塞队列,但是可能还没完成,这边自旋等待其完成
  14. // 当然,这种事情还是比较少的吧:signal 调用之后,没完成转移之前,发生了中断
  15. while (!isOnSyncQueue(node))
  16. Thread.yield();
  17. return false;
  18. }
  1. enq:把节点添加到同步队列
  1. private Node enq(Node node) {
  2. // 无限循环,将节点添加到同步队列尾部
  3. for (;;) {
  4. Node oldTail = tail;
  5. if (oldTail != null) {
  6. node.setPrevRelaxed(oldTail);
  7. if (compareAndSetTail(oldTail, node)) {
  8. oldTail.next = node;
  9. return oldTail;
  10. }
  11. } else {
  12. // 如果同步队列为空,初始化
  13. initializeSyncQueue();
  14. }
  15. }
  16. }
  1. reportInterruptAfterWait:中断处理
  1. private void reportInterruptAfterWait(int interruptMode)
  2. throws InterruptedException {
  3. // 如果是THROW_IE状态,抛异常
  4. if (interruptMode == THROW_IE)
  5. throw new InterruptedException();
  6. else if (interruptMode == REINTERRUPT) // 再次中断,因为中断状态被使用过一次
  7. selfInterrupt();
  8. }

awaitNanosawaitUntilawait(long time, TimeUnit unit)这几个方法的整体逻辑是一样的,就不再分析了

实现signal功能需要做的事情

  1. 将条件队列中的节点加入同步队列
  2. 唤醒线程

与signal功能相关的代码

  • signal方法:唤醒等待条件的节点
  1. public final void signal() {
  2. if (!isHeldExclusively())
  3. throw new IllegalMonitorStateException();
  4. // 获取条件队列中的第一个节点
  5. Node first = firstWaiter;
  6. if (first != null)
  7. // 唤醒等待条件的节点
  8. doSignal(first);
  9. }
  1. doSignal:唤醒等待条件的节点
  1. private void doSignal(Node first) {
  2. do {
  3. // 去掉无效的节点
  4. if ( (firstWaiter = first.nextWaiter) == null)
  5. lastWaiter = null;
  6. first.nextWaiter = null;
  7. } while (!transferForSignal(first) && // 将节点转移到同步队列
  8. (first = firstWaiter) != null);
  9. }
  1. transferForSignal:将节点转移到同步队列
  1. final boolean transferForSignal(Node node) {
  2. /*
  3. * 取消的节点不需要转移
  4. */
  5. if (!node.compareAndSetWaitStatus(Node.CONDITION, 0))
  6. return false;
  7. /*
  8. * 将节点加入同步队列尾部
  9. */
  10. Node p = enq(node);
  11. int ws = p.waitStatus;
  12. // ws > 0 说明 node 在阻塞队列中的前驱节点取消了等待锁,直接唤醒 node 对应的线程
  13. // 如果 ws <= 0, 那么 compareAndSetWaitStatus 将会被调用
  14. // 节点入队后,需要把前驱节点的状态设为SIGNAL
  15. if (ws > 0 || !p.compareAndSetWaitStatus(ws, Node.SIGNAL))
  16. // 如果前驱节点取消或者 CAS 失败,会进到这里唤醒线程
  17. LockSupport.unpark(node.thread);
  18. return true;
  19. }
  • signalAlll方法:唤醒所有等待条件的节点
  1. public final void signalAll() {
  2. // 如果是当前线程
  3. if (!isHeldExclusively())
  4. throw new IllegalMonitorStateException();
  5. Node first = firstWaiter;
  6. if (first != null)
  7. // 唤醒所有等待条件的节点
  8. doSignalAll(first);
  9. }
  1. doSignalAll:唤醒所有等待条件的节点
  1. // 将所有的节点都转移到同步队列
  2. private void doSignalAll(Node first) {
  3. lastWaiter = firstWaiter = null;
  4. do {
  5. Node next = first.nextWaiter;
  6. first.nextWaiter = null;
  7. transferForSignal(first);
  8. first = next;
  9. } while (first != null);
  10. }

现在将与AQS相关的核心代码都整理了一遍,里面如果有描述不清晰或者不准确的地方希望大家可以帮忙指出!

深入浅出AQS源码解析的更多相关文章

  1. 深入浅出ReentrantLock源码解析

    ReentrantLock不但是可重入锁,而且还是公平或非公平锁,在工作中会经常使用到,将自己对这两种锁的理解记录下来,希望对大家有帮助. 前提条件 在理解ReentrantLock时需要具备一些基本 ...

  2. 深入浅出ReentrantReadWriteLock源码解析

    读写锁实现逻辑相对比较复杂,但是却是一个经常使用到的功能,希望将我对ReentrantReadWriteLock的源码的理解记录下来,可以对大家有帮助 前提条件 在理解ReentrantReadWri ...

  3. 深入浅出Semaphore源码解析

    Semaphore通过permits的值来限制线程访问临界资源的总数,属于有限制次数的共享锁,不支持重入. 前提条件 在理解Semaphore时需要具备一些基本的知识: 理解AQS的实现原理 之前有写 ...

  4. AbstractQueuedSynchronizer(AQS)源码解析

          关于AQS的源码解析,本来是没有打算特意写一篇文章来介绍的.不过在写本学期课程作业中,有一门写了关于AQS的,而且也画了一些相关的图,所以直接拿过来分享一下,如有错误欢迎指正.       ...

  5. AQS源码解析(一)-AtomicBoolean源码解析

    基本类: AtomicInteger AtomicLong AtomicBoolean 数组类型: AtomicIntegerArray AtomicLongArray AtomicReference ...

  6. 源码解析之AQS源码解析

    要理解Lock首先要理解AQS,而要理解并发类最好的方法是先理解其并发控制量不同值的含义以及该类运作流程,然后配合一步步看源码.该类有一个重要的控制量是WaitStates,节点的状态值. /** w ...

  7. AQS源码解析

    文大篇幅引用自HongJie大佬的一行一行源码分析清楚AbstractQueuedSynchronizer,这只是一篇简单的个人整理思路和总结(倒垃圾),如果觉得有些难懂的话,不要犹豫也不要疑惑,很明 ...

  8. 并发编程之:AQS源码解析

    大家好,我是小黑,一个在互联网苟且偷生的农民工. 在Java并发编程中,经常会用到锁,除了Synchronized这个JDK关键字以外,还有Lock接口下面的各种锁实现,如重入锁ReentrantLo ...

  9. AbstractQueuedSynchronizer源码解析

    1.简介 AbstractQueuedSynchronizer队列同步器,用来实现锁或者其他同步组件的基础框架 AbstractQueuedSynchronizer使用int类型的volatile变量 ...

随机推荐

  1. QTabWidget 中 关于Tab 关闭和添加的基本教程!

    QTabWidget是PyQt5 中使用较为广泛的容器之一,经常会在日常使用的软件中用到它:QTabwidget是由几个标签组成,每个标签可以当作一个界面,下面就是应用Qtabwidget的一个简单例 ...

  2. 千万不要更新网易云音乐UWP!!!!!

    网易云音乐UWP没了!!! 现在 Micrsoft Store 里面的是垃圾 Win32 转置版!!!! 万不可更新!!! 若已经更新,还有救回来的办法:下载 https://lanzous.com/ ...

  3. 备份、恢复数据库(Dos命令提示符下)_数据库安装工具_连载_1

    Dos命令提示符下: 备份.恢复数据库,是不是很简单啊,是的,当你20年不碰MS SQL,是不是又忘记了呢,答案也许也是吧,^_^虽然在程序中执行SQL代码时,很讨厌那个Go,正如MySQL中那个分号 ...

  4. 线上服务的FGC问题排查,看这篇就够了!

    线上服务的GC问题,是Java程序非常典型的一类问题,非常考验工程师排查问题的能力.同时,几乎是面试必考题,但是能真正答好此题的人并不多,要么原理没吃透,要么缺乏实战经验. 过去半年时间里,我们的广告 ...

  5. MySQL的LIKE模糊查询优化

    原文链接:https://www.cnblogs.com/whyat/p/10512797.html %xxx%这种方式对于数据量少的时候,我们倒可以随意用,但是数据量大的时候,我们就体验到了查询性能 ...

  6. Python实现二分法和黄金分割法

    运筹学课上,首先介绍了非线性规划算法中的无约束规划算法.二分法和黄金分割法是属于无约束规划算法的一维搜索法中的代表. 二分法:$$x_{1}^{(k+1)}=\frac{1}{2}(x_{R}^{(k ...

  7. laravel查询常用的方式含义.

    find($id) 传值并返回一个模型.如果不存在匹配的模型,则返回null.findOrFail($id) 传值并返回一个模型.如果不存在匹配的模型, 它会抛出异常.first() 返回在数据库中找 ...

  8. uni-app之实现分页

    一.下载库 官方文档地址为:https://ext.dcloud.net.cn/plugin?id=32 点击下载zip压缩包即可,下载完毕后解压到放置前端相关组件目录,即components目录. ...

  9. 多语言工作者の十日冲刺<5/10>

    这个作业属于哪个课程 软件工程 (福州大学至诚学院 - 计算机工程系) 这个作业要求在哪里 团队作业第五次--Alpha冲刺 这个作业的目标 团队进行Alpha冲刺--第五天(05.04) 作业正文 ...

  10. springboot的jar为何能独立运行

    欢迎访问我的GitHub https://github.com/zq2599/blog_demos 内容:所有原创文章分类汇总及配套源码,涉及Java.Docker.Kubernetes.DevOPS ...