Java并发之AQS详解

2401_84048179

于 2024-04-02 17:29:14 发布

阅读量402

点赞数 3

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

本文链接：https://blog.csdn.net/2401_84048179/article/details/137278297

版权

程序员专栏收录该内容

266 篇文章 0 订阅

订阅专栏

// 如获取资源成功，直接返回

if (!tryAcquire(arg) &&

// 如获取资源失败，将线程包装为Node添加到队列中阻塞等待

acquireQueued(addWaiter(Node.EXCLUSIVE), arg))

// 如阻塞线程被打断

selfInterrupt();

}

复制代码

acquire核心为tryAcquire、addWaiter和acquireQueued三个函数，其中tryAcquire需具体类实现。每当线程调用acquire时都首先会调用tryAcquire，失败后才会挂载到队列，因此acquire实现默认为非公平锁。

addWaiter将线程包装为独占节点，尾插式加入到队列中，如队列为空，则会添加一个空的头节点。值得注意的是addWaiter中的enq方法，通过CAS+自旋的方式处理尾节点添加冲突。

acquireQueue在线程节点加入队列后判断是否可再次尝试获取资源，如不能获取则将其前驱节点标志为SIGNAL状态(表示其需要被unpark唤醒)后，则通过park进入阻塞状态。

参照流程图，acquireQueued方法核心逻辑为for(;;)和shouldParkAfterFailedAcquire。tail节点默认初始状态为0，当新节点被挂载到队列后，将其前驱即原tail节点状态设为SIGNAL，表示该节点需要被唤醒，返回true后即被park陷入阻塞。for循环直到节点前驱为head后才尝试进行资源获取。

release

release流程较为简单，尝试释放成功后，即从头结点开始唤醒其后继节点，如后继节点被取消，则转为从尾部开始找阻塞的节点将其唤醒。阻塞节点被唤醒后，即进入acquireQueued中的for(;;)循环开始新一轮的资源竞争。

共享锁分析

acquireShared & releaseShared

public final void acquireShared(int arg) {

// 负数表示获取共享锁失败，不同于tryAcquire的bool返回

if (tryAcquireShared(arg) < 0)

doAcquireShared(arg);

}

public final boolean releaseShared(int arg) {

if (tryReleaseShared(arg)) {

doReleaseShared();

return true;

}

return false;

}

复制代码

acquireShared和releaseShared整体流程与独占锁类似，tryAcquireShared获取失败后以Node.SHARED挂载到队尾阻塞，直到队头节点将其唤醒。在doAcquireShared与独占锁不同的是，由于共享锁是可以被多个线程获取的，因此在首个阻塞节点被唤醒后，会通过setHeadAndPropagate传递唤醒后续的阻塞节点。

// doAcquireShared核心代码

final Node node = addWaiter(Node.SHARED);

…

for (;😉 {

final Node p = node.predecessor();

if (p == head) {

int r = tryAcquireShared(arg);

if (r >= 0) {

// r>=0 表示获取锁成功，调整头结点并传递唤醒

setHeadAndPropagate(node, r);

}

…

}

复制代码

setHeadAndPropagate和doReleaseShared构成共享锁唤醒的核心逻辑。

这两方法的逻辑较为简单，不再进行展开，主要对setheadAndPropagate的多节点唤醒判断逻辑做出分析。

进入setHeadAndPropagate，首先需要明确的是，该函数的传入参数propagate一定是非负数，接下来其唤醒主要为两个判断逻辑：

如果propagate > 0，表示存在多个共享锁可以获取，可直接进行doReleaseShared唤醒阻塞节点。
如果propagate = 0，表示仅当前节点可被唤醒，则有两种情况：
h == null || h.waitStatus < 0，通常情况下h != null，现给出h.waitStatus < 0的场景。

(h = head) == null || h.waitStatus < 0的场景执行序列如下：

独占锁共享锁小结

1、独占锁共享锁默认都是非公平获取策略，可能被插队。

2、独占锁只有一个线程可获取，其他线程均被阻塞在队列中；共享锁可以有多个线程获取。

3、独占锁释放仅唤醒一个阻塞节点，共享锁可以根据可用数量，一次唤醒多个阻塞节点

ConditionObject

AQS中Node除了组成阻塞队列外，还在ConditionObject中得到应用，ConditionObject的核心定义为：

public class ConditionObject implements Condition, java.io.Serializable {

…

private transient Node firstWaiter;

private transient Node lastWaiter;

…

}

复制代码

ConditionObject通过Node也构成了一个FIFO的队列，那么ConditionObject为AQS提供了怎样的功能呢？

public interface Condition {

…

void await() throws InterruptedException;

void signal();

void signalAll();

…

}

复制代码

查看Condition接口的定义，可以看到其定义的方法与Object类的wait/notify/notifyAll功能是一致的。

在Synchronized详解中笔者曾对ObjectMonitor做过简单介绍，其中ObjectMonitor包含_WaitSet和_EntryList两个队列，分别用于存储wait调用和sychronized锁竞争时挂起的线程，而AQS通过ConditionObject同样也提供了wait/notify机制的阻塞队列。

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