AQS源码解读（五）——从acquireShared探索共享锁实现原理，何为共享？如何共享？

最新推荐文章于 2024-05-16 06:14:40 发布

2401_84408814

最新推荐文章于 2024-05-16 06:14:40 发布

阅读量872

点赞数 14

分类专栏：程序员文章标签： java 开发语言

本文链接：https://blog.csdn.net/2401_84408814/article/details/137852264

版权

程序员专栏收录该内容

308 篇文章 0 订阅

订阅专栏

p不是头结点 or 获取锁失败，判断是否应该被阻塞
前继节点的ws = SIGNAL 时应该被阻塞

if (shouldParkAfterFailedAcquire(p, node) &&

parkAndCheckInterrupt())

interrupted = true;

}

} finally {

if (failed)

cancelAcquire(node);

}

（1）setHeadAndPropagate传播唤醒后继

自旋判断新节点node前驱是head，且尝试获取共享锁成功，则需要传播唤醒共享后继节点。基本流程如下：

node出队列，并成为新的head。
判断是否满足传播条件。
如若node还有后继且后继是共享节点，则执行唤醒操作doReleaseShared。

（共享锁的传播模式比较复杂，如需深入了解，请阅读拙作《AQS源码解读（七）——从PROPAGATE和setHeadAndPropagate()分析共享锁的传播性》）

private void setHeadAndPropagate(Node node, int propagate) {

Node h = head; // Record old head for check below

//设置node为新head

setHead(node);

//一连串判断满足后，唤醒共享后继

if (propagate > 0 || h == null || h.waitStatus < 0 ||

(h = head) == null || h.waitStatus < 0) {

//唤醒后继共享节点

Node s = node.next;

if (s == null || s.isShared())

doReleaseShared();

}

（2）判断是否应该阻塞和阻塞线程

共享锁判断判断是否应该阻塞和阻塞线程的代码和独占锁的完全一样，不过多赘述。shouldParkAfterFailedAcquire主要做了如下几步：

判断node前驱状态是否为SIGNAL，是则直接返回true。
node前驱状态不是SIGNAL，有可能是ws>0，说明前驱取消了，自旋跳过取消的节点，并寻找链接一个正常的前驱。
node前驱状态不是SIGNAL，有可能是0（初始化状态）或PROPAGATE（传播状态），修改node前驱状态为SIGNAL。

private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {

int ws = pred.waitStatus;

if (ws == Node.SIGNAL)

This node has already set status asking a release
to signal it, so it can safely park.
node拿锁失败，前继节点的状态是SIGNAL，node节点可以放心的阻塞，
因为下次会被唤醒

return true;

if (ws > 0) {

Predecessor was cancelled. Skip over predecessors and
indicate retry.
pred节点被取消了，跳过pred

do {

//pred = pred.prev;

//node.pred = pred;

node.prev = pred = pred.prev;

} while (pred.waitStatus > 0);

//跳过取消节点，给node找一个正常的前驱,然后再循环一次

pred.next = node;

} else {

/* 0 -3

waitStatus must be 0 or PROPAGATE. Indicate that we
need a signal, but don’t park yet. Caller will need to
retry to make sure it cannot acquire before parking.

compareAndSetWaitStatus(pred, ws, Node.SIGNAL);

}

return false;

}

shouldParkAfterFailedAcquire判断返回true，则调用parkAndCheckInterrupt阻塞当前线程。

private final boolean parkAndCheckInterrupt() {

LockSupport.park(this);

return Thread.interrupted();

}

三、共享模式可中断获取锁

acquireSharedInterruptibly在代码结构上与doAcquireShared类似，不同之处在于doAcquireShared不响应中断，acquireSharedInterruptibly响应中断。

public final void acquireSharedInterruptibly(int arg)

throws InterruptedException {

if (Thread.interrupted())

//被打断抛出异常

throw new InterruptedException();

if (tryAcquireShared(arg) < 0)

//获取锁失败

doAcquireSharedInterruptibly(arg);

}

doAcquireSharedInterruptibly入队列，自旋，阻塞，响应中断

doAcquireSharedInterruptibly逻辑上也和doAcquireShared类似，但是doAcquireSharedInterruptibly在线程阻塞期间，被中断导致唤醒，将抛出异常InterruptedException，响应中断，此时failed=true，finally块中会执行取消节点的操作，取消过程和独占锁一样。

private void doAcquireSharedInterruptibly(int arg)

throws InterruptedException {

//新建节点入队列

final Node node = addWaiter(Node.SHARED);

boolean failed = true;

try {

for (;😉 {

//node节点的前驱节点是head，获取锁

final Node p = node.predecessor();

if (p == head) {

int r = tryAcquireShared(arg);

if (r >= 0) {

setHeadAndPropagate(node, r);

p.next = null; // help GC

failed = false;

return;

}

if (shouldParkAfterFailedAcquire(p, node) &&

parkAndCheckInterrupt())

//在阻塞期间因为中断而唤醒，抛异常

throw new InterruptedException();

}

} finally {

if (failed)

cancelAcquire(node);

}

四、共享模式可超时获取锁

和独占锁的可超时获取锁类似，共享模式超时获取锁，不仅可以响应中断，还可以将线程阻塞一段时间，自动唤醒。tryAcquireSharedNanos可传入一个纳秒单位的时间nanosTimeout，可超时的逻辑在doAcquireSharedNanos中。

public final boolean tryAcquireSharedNanos(int arg, long nanosTimeout)

throws InterruptedException {

if (Thread.interrupted())

throw new InterruptedException();

return tryAcquireShared(arg) >= 0 ||

doAcquireSharedNanos(arg, nanosTimeout);

}

doAcquireSharedNanos入队列，自旋，阻塞，自动唤醒，响应中断

doAcquireSharedNanos和doAcquireShared逻辑类似，但是不仅可以响应中断，同时还可以让线程阻塞一段时间自动唤醒，如果超时了还没获取锁则返回false。

doAcquireSharedNanos还有一个非常不同之处，就是即使shouldParkAfterFailedAcquire判断应该阻塞了，也有可能不阻塞，还会再自旋一段时间，这个自旋的时长有一个阈值spinForTimeoutThreshold = 1000L，1000纳秒，自旋了1000纳秒后还没有获取锁，且此时也判断应该阻塞了，就让线程休眠一段时间。

线程唤醒，有可能是自动唤醒，有可能是被其他释放锁的线程唤醒，还有可能是线程中断导致唤醒，如果是线程中断唤醒，则需要响应中断，抛出异常InterruptedException；如果没有中断，则继续循环刚才的流程（判断前驱是否是head，判断是否超时，判断是否需要阻塞等）。

private boolean doAcquireSharedNanos(int arg, long nanosTimeout)

throws InterruptedException {

if (nanosTimeout <= 0L)

return false;

final long deadline = System.nanoTime() + nanosTimeout;

final Node node = addWaiter(Node.SHARED);

boolean failed = true;

try {

for (;😉 {

final Node p = node.predecessor();

if (p == head) {

int r = tryAcquireShared(arg);

if (r >= 0) {

setHeadAndPropagate(node, r);

p.next = null; // help GC

failed = false;

return true;

}

nanosTimeout = deadline - System.nanoTime();

if (nanosTimeout <= 0L)

return false;

//判断是否应该阻塞

//在阈值spinForTimeoutThreshold范围自旋一定时长，

if (shouldParkAfterFailedAcquire(p, node) &&

nanosTimeout > spinForTimeoutThreshold)

//阻塞一定时间，时间到可自动唤醒

LockSupport.parkNanos(this, nanosTimeout);

if (Thread.interrupted())

//线程如若发生中断，响应中断抛出异常

throw new InterruptedException();

}

} finally {

if (failed)

cancelAcquire(node);

}

五、共享模式释放锁

共享模式释放锁的逻辑很简单，tryReleaseShared释放锁成功后，调用doReleaseShared唤醒后继节点。tryReleaseShared需要子类自行实现。

public final boolean releaseShared(int arg) {

if (tryReleaseShared(arg)) {

//读锁释放才唤醒后继节点

doReleaseShared();

return true;

}

return false;

}

doReleaseShared唤醒后继

共享锁释放和在AQS同步队列中的共享节点获取锁，传播唤醒时都调用了doReleaseShared，其唤醒后继是一个自旋的过程（for(;;)），其基本流程如下：

获取head，判断队列是否为空。
不为空，则继续判断head的状态是否是SIGNAL，即是否可以唤醒后继。
head的状态为SIGNAL，则将状态置为0。head状态置为0成功，则唤醒后节点线程unparkSuccessor，失败则自旋重试。
若head状态为0，则将其置为PROPAGATE，使得head具有传播唤醒的特性。
如若自旋的过程中head变了，则继续自旋唤醒后继，head没有变，跳出自旋。

private void doReleaseShared() {

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

笔者福利

以下是小编自己针对马上即将到来的金九银十准备的一套“面试宝典”，不管是技术还是HR的问题都有针对性的回答。

有了这个，面试踩雷？不存在的！

回馈粉丝，诚意满满！！！

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
“https://img-blog.csdnimg.cn/img_convert/9f4282ea5cddfccb9e653df70ae8ae70.jpeg” alt=“img” style=“zoom: 33%;” />

笔者福利

以下是小编自己针对马上即将到来的金九银十准备的一套“面试宝典”，不管是技术还是HR的问题都有针对性的回答。

有了这个，面试踩雷？不存在的！

回馈粉丝，诚意满满！！！

[外链图片转存中…(img-wIUPkxLe-1713307458076)]
[外链图片转存中…(img-NpEoNLnz-1713307458076)]
[外链图片转存中…(img-9Awe12Rb-1713307458076)]
[外链图片转存中…(img-1gbp0QKQ-1713307458077)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！