AQS 是什么?
AQS 全称为 AbstractQueueSynchronizer,顾名思义——抽象队列同步器,是 ReentrantLock、ReentrantReadWriteLock等锁的底层实现。
抽象队列同步器的工作原理是将工作线程封装为Node节点,当线程竞争锁时,通过CAS方式来设置state数值,当state数值大于等于1时,工作线程无法争抢锁,进入双向队列等待;当工作线程工作完毕时,通过CAS操作将state减一处理,若state等于0,根据公平/非公平锁激活工作线程。
AQS 三大基石
AQS是通过state、Node和CHL双向同步队列构建起来框架。
AQS中还大量使用CAS乐观锁操作,通过传递旧值和新值两个参数进行调用(当目标变量当前值与旧值相等时,可以进行赋值操作)。
- state:信号量,被 volatile 修饰,当 state=0 时,线程可以竞争锁,当 state>=1 时,线程无法竞争锁。
- Node:封装工作线程,保存前驱、后继节点的引用,最后会进入 CHL 队列中。
- CHL 同步队列:是一个双向队列,存放未竞争到锁的线程。
AQS 源码之路
通过银行办理业务的案例对 AbstractQueuedSynchronizer 类进行讲解,可以通过全局搜索进入该类。
// 自旋获取锁操作
ReentrantLock lock = new ReentrantLock();
// 银行办理业务案例模拟AQS
// A顾客是第一个顾客,此时受理窗口无任何人
new Thread(() -> {
lock.lock();
try {
System.out.println("----------A thread come in");
TimeUnit.MINUTES.sleep(3);
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
lock.unlock();
}
},"A").start();
// B顾客是第二个顾客 ---> 由于窗口只有一个,等待
new Thread(() -> {
lock.lock();
try {
System.out.println("----------B thread come in");
} finally {
lock.unlock();
}
},"B").start();
// C顾客是第三个顾客 ---> 由于窗口只有一个,等待
new Thread(() -> {
lock.lock();
try {
System.out.println("----------C thread come in");
} finally {
lock.unlock();
}
},"C").start();
final void lock() {
/**
* A线程进入:
* 进行CAS操作,当state为0时,将state设置为1
* 将 exclusiveOwnerThread 线程设置为当前线程
*
* B线程进入:
* 进行CAS操作,发现state不是0,无法进行赋值,进入acquire方法
*/
if (compareAndSetState(0, 1))
setExclusiveOwnerThread(Thread.currentThread());
else
acquire(1);
}
public final void acquire(int arg) {
if (!tryAcquire(arg) &&
acquireQueued(addWaiter(Node.EXCLUSIVE), arg))
selfInterrupt();
}
protected final boolean tryAcquire(int acquires) {
return nonfairTryAcquire(acquires);
}
final boolean nonfairTryAcquire(int acquires) {
final Thread current = Thread.currentThread(); // 获取到B线程
int c = getState(); // 获取信号量数值
if (c == 0) { // 如果此时信号量等于0
// 对state进行CAS操作,当state=0时,将state更新为 acquires
// 此时的 acquires 为1
if (compareAndSetState(0, acquires)) {
setExclusiveOwnerThread(current); // 设置争取到锁的线程为当前线程
return true;
}
}
// 若当前线程与获取到锁的线程是同一线程时
else if (current == getExclusiveOwnerThread()) {
int nextc = c + acquires; // 将信号量数值增加
if (nextc < 0) // overflow
throw new Error("Maximum lock count exceeded");
setState(nextc); // 为信号量设置新值
return true;
}
return false;
}
// 进入当前方法,可得出已经有线程争抢到锁
private Node addWaiter(Node mode) {
// 将当前线程封装为Node节点
Node node = new Node(Thread.currentThread(), mode);
// Try the fast path of enq; backup to full enq on failure
// 获取到链式队列中的尾节点
Node pred = tail;
if (pred != null) {
node.prev = pred;
if (compareAndSetTail(pred, node)) {
pred.next = node;
return node;
}
}
// 第一次进入addWaiter方法时,队列为NULL,直接执行enq
// enq方法是初始化出链式队列
enq(node);
return node;
}
// 初始化尾插链式队列
private Node enq(final Node node) {
for (;;) { // 自旋
Node t = tail; // 获取到尾节点
// 当尾节点为NULL时,将head头节点初始化为 Thread=NULL,waitStatus=0
if (t == null) { // Must initialize
if (compareAndSetHead(new Node()))
tail = head; // 头尾指针指向一个Node节点,即刚刚创建的NULL节点
} else { // 因为是自旋,所以当if执行后,下次循环进入else
node.prev = t; // 设置node的前驱指针为 尾节点地址
if (compareAndSetTail(t, node)) { // 尾节点指向node,但下文中t仍指向head
t.next = node; // 此时t代表的是头节点,头节点的后继指针指向node
return t;
}
}
}
}
// 自旋获取锁,若一直获取不到锁就阻塞。
final boolean acquireQueued(final Node node, int arg) {
boolean failed = true;
try {
boolean interrupted = false;
for (;;) { // 自旋
final Node p = node.predecessor(); // 获得node的前驱节点
// p为头节点
// 尝试获取锁成功时执行
if (p == head && tryAcquire(arg)) { // A线程执行完毕后,B线程执行为true 进行出队操作
setHead(node); // 将node设置为头节点
p.next = null; // help GC
failed = false;
return interrupted;
}
if (shouldParkAfterFailedAcquire(p, node) &&
parkAndCheckInterrupt())
interrupted = true;
}
} finally {
if (failed)
cancelAcquire(node);
}
}
// 第二次进入该方法时,前驱节点waitStatus设置为-1。
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
// 第一次进入获得node前驱节点的waitStatus,此时前驱节点的waitStatus为0
// 第二次进入获得node前驱节点的waitStatus,此时前驱节点的waitStatus为-1
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.
*/
return true;
if (ws > 0) {
/*
* Predecessor was cancelled. Skip over predecessors and
* indicate retry.
*/
do {
node.prev = pred = pred.prev;
} while (pred.waitStatus > 0);
pred.next = node;
} else {
/*
* 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); // 将前驱节点的waitStatus设置为 -1
}
return false;
}
// 将当前线程 park 阻塞,然后返回中断位。
private final boolean parkAndCheckInterrupt() {
LockSupport.park(this); // 阻塞 后 被激活
return Thread.interrupted();
}
// 进行中断
static void selfInterrupt() {
Thread.currentThread().interrupt();
}
// 释放锁操作
public final boolean release(int arg) {
if (tryRelease(arg)) {
Node h = head;
if (h != null && h.waitStatus != 0)
unparkSuccessor(h);// 将阻塞的线程发放许可
return true;
}
return false;
}
// 激活后续线程
private void unparkSuccessor(Node node) {
/*
* If status is negative (i.e., possibly needing signal) try
* to clear in anticipation of signalling. It is OK if this
* fails or if status is changed by waiting thread.
*/
int ws = node.waitStatus;
if (ws < 0)
compareAndSetWaitStatus(node, ws, 0);
/*
* Thread to unpark is held in successor, which is normally
* just the next node. But if cancelled or apparently null,
* traverse backwards from tail to find the actual
* non-cancelled successor.
*/
Node s = node.next; // 获取到线程B
if (s == null || s.waitStatus > 0) {
s = null;
for (Node t = tail; t != null && t != node; t = t.prev)
if (t.waitStatus <= 0)
s = t;
}
if (s != null)
LockSupport.unpark(s.thread); // 发放许可,激活线程
}