AQS源码深度剖析,大厂面试必看!

AQS(AbstractQueuedSynchronizer)是Java众多锁以及并发工具的基础类,底层采用乐观锁,大量采用CAS操作保证其原子性,并且在并发冲突时,采用自旋方法重试。实现了轻量高效的获取锁。

1. AQS的关注点
ReentrantLock中使用到了AQS高并发组件,用它来维护锁的状态,这样就不需要利用操作系统来维护,减少了上下文切换。AQS中使用了CAS、自旋操作来提高性能。但是在线程过多的时候,还是会和操作系统打交道,挂起线程和唤醒线程两个上下文操作。

AQS在线程交替运行时,只需要借助CAS和自旋就可以完成加锁。

而synchronized在JDK1.6版本前是重量级锁,加锁的操作是涉及到操作系统进行互斥操作,就是会把当前线程挂起,然后操作系统进行互斥操作修改,由mutexLock来完成,之后才唤醒。操作系统来判断线程是否加锁,所以它是一个重量级操作。挂起、唤醒这两个操作进行了两次上下文切换,消耗CPU,降低性能。
总之一句话,重量级锁是需要依靠操作系统来实现互斥锁的,这导致大量上下文切换,消耗大量CPU,影响性能。

1.1 信号量
在AQS中,状态是由volatile state来表示。

private volatile int state;

该属性值表示锁的状态。state为0表示锁未被占用,state为1表示锁被线程持有,而state大于1表示锁被重入。

而本文分析的是独占锁,那么同一时刻,锁只能被一个线程持有。

不仅需要记录锁的状态,还需要记录当前获取锁的线程,实现重入。可以通过来记录。

private transient Thread exclusiveOwnerThread;

1.2. 等待队列
等待队列采用悲观锁的思想,表示当前所等待的资源,状态或条件短时间内可能无法满足,而调用park方法(借助操作系统)来完成线程的阻塞。

在AQS中,队列时一个双端链表,将当前线程包装成某种类型的数据结构扔到等待队列中。

static final class Node {
// 节点所代表的线程
volatile Thread thread;
// 双向链表,每个节点需要保存自己的前驱节点和后继节点的引用
volatile Node prev;
volatile Node next;
// 线程所处的等待锁的状态,初始化时,该值为0。
volatile int waitStatus;
//队列中节点线程被取消
static final int CANCELLED = 1;
//节点将其前驱节点设置为-1,当前驱节点释放锁后,会自动唤醒该节点。
static final int SIGNAL = -1;
//线程被重新包装为Node节点,并存入Condition队列中。
static final int CONDITION = -2;
//共享锁唤醒风暴时,将0->PROPAGATE,表示被传播唤醒
static final int PROPAGATE = -3;
// 该属性用于条件队列或者共享锁 。在Condition队列中,使用其作为指针。
Node nextWaiter;
}

一般在独占锁下,我们需要关注的就是下面几个参数:

thread:当前Node所代表的线程;
waitStatus:表示节点所处的等待状态;
prev next:节点的前驱和后继
1.3. CAS操作
CAS采用乐观锁机制,保证操作的原子性。一般是改变状态或改变指针(引用)指向。

 

1.4 总结
在AQS源码中:

锁属性
//锁的状态
private volatile int state;
//当前持有锁的线程
private transient Thread exclusiveOwnerThread;

sync queue相关的属性
//thread属性为null
private transient volatile Node head;
private transient volatile Node tail; // 队尾,新入队的节点

Node相关属性
// 节点所代表的线程
volatile Thread thread;

// 双向链表,每个节点需要保存自己的前驱节点和后继节点的引用
volatile Node prev;
volatile Node next;

// 线程所处的等待锁的状态,初始化时,该值为0
volatile int waitStatus;
static final int CANCELLED = 1;
static final int SIGNAL = -1;

2. 源码解析
ReentrantLock有公平锁和非公平锁两种实现,默认实现非公平锁。但是可配置为公平锁:

ReentrantLock lock=new ReentrantLock(true);

调用公平锁加锁逻辑:

final void lock() {
//开始加锁,将state修改为1
acquire(1);
}


真正的加锁方法:

public final void acquire(int arg) {
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}


2.1 加锁的逻辑方法
只执行上述方法便可完成整个的加锁逻辑。而该方法中又包含下列四个方法的调用:

1. tryAcquire(arg)
该方法由继承AQS的子类实现,为获取锁的具体逻辑;

2. addWaiter(Node.EXCLUSIVE)
该方法由AQS实现,负责在获取锁失败后调用,将当前请求锁的线程包装成Node并且放到等待队列中,并返回该Node。

3. acquireQueued(addWaiter(Node.EXCLUSIVE), arg)
该方法由AQS实现。针对上面加入到队列的Node不断尝试两种操作之一:

若前驱节点是head节点的时候,尝试获取锁;
调用park将当前线程挂起,线程阻塞。
4. selfInterrupt
该方法由AQS实现。恢复用户行为。

用户在外界调用t1.interrupt()进行中断。

线程在parkAndCheckInterrupt方法被唤醒之后。会调用Thread.interrupted();判断线程的中断标识,而该方法调用完毕会清除中断标识位。

而AQS为了不改变用户标识。再次调用selfInterrupt恢复用户行为。

2.2 如何构建等待队列——addWaiter
我们使用ReentrantLock独占锁时,等待队列是延迟加载的。也就是说若是线程交替执行,那么借助信号量(状态)来保证。若是线程并发执行,就需要将阻塞线程放入到队列中。

//注意这个方法可能存在并发问题,mode为null(独占锁)。
private Node addWaiter(Node mode) {
Node node = new Node(Thread.currentThread(), mode);
Node pred = tail;
//队列已经存在
if (pred != null) {
//新节点的前驱指针指向尾节点(可能造成尾分叉)
node.prev = pred;
//保证原子性,只有一个才能成功
if (compareAndSetTail(pred, node)) {
pred.next = node;
return node;
}
}
//队列不存在&&上面CAS失败的线程会进入enq方法自旋
enq(node);
return node;
}

队列不存在的情况

 

注意,该方法处理CAS操作是原子性的,其他操作都存在并发冲突问题。

private Node enq(final Node node) {
for (;;) {
Node t = tail;
//初始化阻塞队列
if (t == null) { // Must initialize
if (compareAndSetHead(new Node()))
tail = head;
} else {
//自旋处理addWaiter中CAS加锁失败的线程
node.prev = t;
if (compareAndSetTail(t, node)) {
t.next = node;
return t;
}
}
}
}


该方法采用自旋+CAS。CAS是保证同一时刻只有一个线程能成功改变引用的指向。

 

根据上面的流程图,sync queue的创建过程。head节点是new Node()产生的,即其中的属性为默认值。也就是thread属性为null。也就是说正在执行的线程也会在sync queue中占据头节点,但是节点中不会保存线程信息。

 

尾分叉问题:

上面已经说了,该方法是线程不安全的。

//步骤1:可能多个节点的prev指针都指向尾结点,导致尾分叉
node.prev = t;
//步骤2:但同一时刻,tail引用只会执行一个node。
if (compareAndSetTail(t, node)) {
//步骤3:现在环境是线程安全,旧尾结点的后继指针指向新尾结点。
t.next = node;
return t;
}

 

执行完步骤2,但步骤3还未执行时,恰好有线程从头节点开始往后遍历。**此时(旧)尾结点中的next域还为null。**它是遍历不到新加进来的尾结点的。这显然是不合理的。

但此时步骤1是执行成功的,所以若是tail节点往前遍历,实际上是可以遍历到所有节点的,这也是为什么在AQS源码中,有时候常常会出现从尾结点开始逆向遍历链表的情况。

那些“分叉”的节点,肯定会入队失败。那么继续自旋,等待所有的线程节点全部入队成功。

2.3 尝试获取锁——tryAcquire
根据标志位state,来判断锁是否被占用。此时可能锁未被占用,由于是公平锁,于是会去判断sync queue中是否有人在排队。

protected final boolean tryAcquire(int acquires) {
//获取当前线程
final Thread current = Thread.currentThread();
//获取Lock对象的上锁情况,0-表示无线程持有;1-表示被线程持有;大于1-表示锁被重入
int c = getState();
//若此刻无人占有锁
if (c == 0) {
if (!hasQueuedPredecessors() && //判断队列中是否有前辈。若返回false代表没有,开始尝试加锁
compareAndSetState(0, acquires)) { //此刻队列中没有存在前辈,尝试加锁
setExclusiveOwnerThread(current); //将当前线程修改为持有锁的线程(后续判断可重入)
return true;
}
}
//若是当前线程是持有锁的线程
else if (current == getExclusiveOwnerThread()) {
//当前状态+1
int nextc = c + acquires;
if (nextc < 0)
throw new Error("Maximum lock count exceeded");
setState(nextc);
return true;
}
//否则,代表加锁失败
return false;
}


下面的方法返回false才会尝试加锁(该方法不具有原子性,可能会放行多个线程)。

//该方法不具有原子性,可能多个线程都觉得自己不需要排队,最终还是依靠外面
//条件上的CAS来保持其原子性。
public final boolean hasQueuedPredecessors() {
Node t = tail; //尾节点
Node h = head; //头节点
Node s;
return h != t &&
((s = h.next) == null || s.thread != Thread.currentThread());
}


上述方法是判断队列中是否存在元素。可能存在以下几种情况:

此时未维护队列【h和t指向null】,h!=t返回false,即无人排队;
此时队列只有头节点(哑结点)【h和t都指向哑结点】,h!=t返回false,即无人排队;
此时队列中存在2个以上的节点。若线程是头结点的后继节点线程(即处理正在办理业务的线程,进来的线程是第一个排队的线程)。那么s.thread != Thread.currentThread()返回false,即可是尝试加锁。
队列存在2个以上节点,且进来的线程不是第一个排队的线程,那么该线程需要乖乖的排队。
当然该方法不是并发安全的方法,即可能存在多个线程觉得自己无需排队,最终还是依靠CAS来争夺锁。

if (!hasQueuedPredecessors() && compareAndSetState(0, acquires)) {
//线程安全
setExclusiveOwnerThread(current);
return true;
}


同一时刻,只有一个线程可以成功改变state的状态。记录该线程为独占锁线程,一般后续可以重入。

没成功获取锁那么会调用2.2 中的方法,将该线程加入到阻塞队列中。

2.3. 阻塞线程——acquireQueued
若执行到该方法,说明addWaiter方法已经成功将该线程包装为Node节点放到了队尾。
在该方法中依旧尝试获取锁;
再次获取锁失败后,会将其阻塞;
final boolean acquireQueued(final Node node, int arg) {
boolean failed = true;
try {
boolean interrupted = false;
for (;;) {
//获取node的前驱节点
final Node p = node.predecessor();
//若前驱节点在办理业务,那么它将再次获取一次锁。
if (p == head && tryAcquire(arg)) {
//获取锁成功,此处便是线程安全。
//将自己设置为头节点,并将自己设置为哑节点
setHead(node);
p.next = null; // help GC
failed = false;
return interrupted;
}
//获取锁失败,将自己挂起。
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
interrupted = true;
}
} finally {
if (failed)
cancelAcquire(node);
}
}


上述方法时自旋方法,而出口就是获取到锁。若线程获取不到锁,便会将自己阻塞。

//该方法时node线程获取锁成功后执行的,故是线程安全的。
private void setHead(Node node) {
head = node;
node.thread = null;
node.prev = null;
}

 

private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
//上一个节点的waitStatus
int ws = pred.waitStatus;
// Node.SIGNAL==-1
if (ws == Node.SIGNAL)
return true;
//ws大于0,则说明该节点已经被取消了。
if (ws > 0) {
do {
node.prev = pred = pred.prev;
} while (pred.waitStatus > 0);
pred.next = node;
} else {
//CAS变更ws的状态
compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
}
return false;
}


上述方法是加锁失败开始执行的。也就是一个线程决定挂起之前需要执行的操作。这里就用到了节点中的信号量waitStatus。

判断前驱节点waitStatus的值,会做出如下操作:
1.1 前驱节点waitStatus若是-1,直接返回true。
1.2 前驱节点waitStatus若大于0,证明前驱节点已被取消,那么在链表中删除前驱节点,直到node的前驱节点的waitStatus不大于0为止。然后返回false
1.3. 若前驱节点waitStatus等于0,使用CAS尝试改变前驱节点waitStatus状态,由0到-1,然后返回false。

若是返回true,那么去阻塞该节点,若是返回false,那么继续自旋,继续上述过程,直至该方法返回true为止,方法返回true,便会执行下列方法,阻塞线程。

private final boolean parkAndCheckInterrupt() {
//将线程挂起
LockSupport.park(this);
//线程被唤起时,查看线程的中断标识(注意,查看完毕后,中断标识归位)
return Thread.interrupted();
}


需要注意的是:当前节点在阻塞之前,会将前驱节点的waitStatus设置为-1,就可保证前驱节点在适当的时机唤醒自己。

附录
对象的CAS算法

 

开始我认为对象的CAS算法,实际上会是B对象去覆盖堆内存上的A对象,其实不然。比较交换的是引用。

//该方法是获取引用。而非堆上的内存。
static {
try {
valueOffset = unsafe.objectFieldOffset
(AtomicReference.class.getDeclaredField("value"));
} catch (Exception ex) { throw new Error(ex); }
}


3. 加锁总结
因为AQS的等待队列是延迟加载,只有多个线程并发访问时,才会开始维护队列。
因为head节点中不包含thread属性的值,又被称为哑节点。
head是正在办理业务的节点,而他的后继节点是第一个排队节点。
尝试加锁过程
根据status判断当前锁是否被持有,若被持有,直接维护队列;
若未被持有,判断当前队列是否有节点在排队,若有节点排队,直接维护队列;
若无节点排队,则通过CAS修改锁状态标识,修改成功代表线程持有该锁;
使用exclusiveOwnerThread来保存持有锁的线程(解决线程重入);
维护队列过程
最终线程的head节点为哑节点。后续线程被组装成node节点,维护在链表中。

线程阻塞过程
判断node节点是否为head节点的后续节点(第一个排队节点),若是的话,尝试获取锁。若获取到,将其设置为head节点,并将其设置为哑节点;
在阻塞前,会将自己的前驱节点的waitStatus设置为SIGNAL。以便可以唤醒自己。

posted @ 2021-06-16 22:37  hanease  阅读(145)  评论(0编辑  收藏  举报