AQS详解

什么是AQS

AQS(AbstractQueuedSynchronizer),即队列同步器,它是构建锁或者其他同步组件的基础框架,如ReentrantLock、ReentrantReadWriteLock、Semaphore,CountDownLatch等。

AQS是一个抽象类,主要是通过继承方式使用,本身没有实现任何接口,仅仅是定义了同步状态的获取和释放的方法。AQS中 维护了一个volatile int state(代表共享资源)和一个FIFO线程等待队列(多线程竞争资源被阻塞时会进入此队列)。

这里volatile能够保证多线程下的可见性,当state=1则代表当前对象锁已经被占有,其他线程来加锁时则会失败,加锁失败的线程会被放入一个FIFO的等待队列中,同时会被UNSAFE.park()操作挂起,等待其他获取锁的线程释放锁才能够被唤醒。

另外state的操作都是通过CAS来保证其并发修改的安全性。

AQS中关于state的定义有三种

  • getState():返回同步状态的当前值;
  • setState(int newState),设置当前同步状态
  • compareAndSetState(int expect,int update),使用CAS设置当前状态,该方法能够保证状态设置的原执行

自定义同步器实现时主要实现以下几种方法

第一类:子类实现的方法,AQS不作处理(模板方法)

  • tryAcquire(int arg):独占获取同步状态,获取同步状态成功后,其他线程需要等待该线程释放同步状态才能获取同步状态。
  • tryRelease(int arg) 独占式释放同步状态
  • tryAcquireShared(int arg) 共享式获取同步状态,返回值大于等于0则表示获取成功,否则获取失败。
  • tryReleaseShared(int arg) 共享式释放同步状态。
  • isHeldExclusively:当前同步器是否在独占模式下被线程占用,一般该方法表示同步器是否被当前线程独占。

第二类:AQS本身的实现的方法,定义给子类通用实现的方法

  • aquire(int arg):独占式的获取锁的操作,独占式获取同步状态都调用者方法,通过子类的tryAquire方法判断是否获取到锁。
  • acquireShared(int arg) 共享式的获取锁的操作,在读写锁中用到,通过tryAquireShared方法判断是否获取同步状态
  • release(int args) 独占式的释放同步状态,通过tryRelease方法判断是否释放了独占式同步状态
  • releaseShared(int arg):共享式的释放同步状态,通过tryReleaseShared方法判断是否已经释放了共享同步状态。

AQS的两种功能

从使用层面来说,AQS功能分为两种:独占和共享

  • 独占锁,每次只能一个线程持有锁,比如ReentrantLock就是独占锁
  • 共享锁,允许多个线程持有锁,并发访问共享资源,比如ReentrantReadWriteLock
  • 共享锁和独占锁的释放有一定区别,前面部分是一致的,先判断头结点是不是signal状态,如果是则唤醒头节点的下一个节点,并将该节点设置为头结点。而共享锁不一样,某个节点被设置为head之后,如果它的后继节点是shared状态,那么会尝试使用doReleaseShared方法尝试唤醒节点,实现共享状态的传播。

AQS内部实现

AQS是依赖内部的同步队列实现,也就是FIFO双向队列,如果当前线程竞争锁失败,那么AQS会把当前线程以及等待状态封装成一个Node节点加入到同步队列中,同时阻塞该线程,当同步状态释放时,会把首节点唤醒,使其再次尝试获取同步状态。

AQS队列内部维护的是一个双向链表,这种结构每个数据都有两个指针,分别指向直接的的前驱节点和后继节点,当线程抢占锁失败时候,会封装成Node加入到AQS中去。

在同步队列中,一个节点表示一个线程,他保存这线程的引用ThreadId,状态(watiStatus),前驱结点(pre),后继节点(next),其数据结构如下:

节点状态waitStatus

每个节点包含了线程的的等待状态,是否被阻塞,是否等待唤醒,是否被取消。变量waitStatus则表示当前Node节点的等待状态,共有5中取值,cancelled,signal,condition,propagate,0

  • cancelled(1):表示当前节点已取消调度。当timeout或者中断情况下,会触发变更为此状态,进入该状态后的节点不再变化
  • signal(-1),表示当前节点释放锁的时候,需要唤醒下一个节点。或者说后继节点在等待当前节点唤醒,后继节点入队时候,会将前驱节点更新给signal
  • condition(-2),当其他线程调用了condition的signal方法后,condition状态的节点会从等待队列转移到同步队列中,等待获取同步锁。
  • propagate(-3),共享模式下,前驱节点不仅会唤醒其后继节点,同时也可能唤醒后继的后继节点。
  • 0 ,新节点入队时候的默认状态。

添加节点addWaiter

入队操作就是tail指向新节点,新节点的前驱节点pre指向之前当前最后的节点,当前最后的节点的next指向新节点,相关操作在addWaiter方法里

ini 复制代码
private Node addWaiter(Node mode) {
        //根据给定的模式(独占或者共享)新建Node
        Node node = new Node(Thread.currentThread(), mode);
        //快速尝试添加尾节点
        Node pred = tail;
        if (pred != null) {
            node.prev = pred;
            //CAS设置尾节点
            if (compareAndSetTail(pred, node)) {
                pred.next = node;
                return node;
            }
        }
        //多次尝试
        enq(node);
        return node;
    }

addWaiter(Node node)先通过快速尝试设置尾节点,如果失败,则调用enq(Node node)方法设置尾节点

ini 复制代码
private Node enq(final Node node) {
        //多次尝试,直到成功为止
        for (;;) {
            Node t = tail;
            //tail不存在,设置为首节点
            if (t == null) {
                if (compareAndSetHead(new Node()))
                    tail = head;
            } else {
                //设置为尾节点
                node.prev = t;
                if (compareAndSetTail(t, node)) {
                    t.next = node;
                    return t;
                }
            }
        }
    }

此方法用于将Node加入队尾,核心就是通过CAS自旋的方式设置尾结点。假如有两个线程t1,t2,同时进入enq方法 ,t==null表示队列是首次使用,需要先初始化,另一个线程cas失败,则进入下次循环,通过cas操作将node添加到队尾。

同步状态的获取和释放

独占式同步状态获取

acquire(int arg): 独占式的获取锁,此方法不响应中断,在这个过程中中断,线程不会从同步队列中移除,也不会立马中断,在整个过程结束后再自我中断

scss 复制代码
public final void acquire(int arg){
              if(!tryAcquire(arg) && acquireQueued(addWaiter(Node.EXCLUSIVE))){
                         selfInterrupt();//如果这个过程中出现中断,在整个过程结束后再自我中断
              }
         }
  • tryAcquire:去尝试获取锁,获取成功后设置锁的状态并返回true,否则返回false。该方法由自定的同步组件实现,该方法的实现必须保证线程安全的获取同步状态。
  • addWaiter:如果tryAcquire返回false(获取锁失败),则调用该方法将当前线程封装成节点加入到同步队列尾部,并标记为独占模式。addWaiter(Node.EXCLUSIVE)),Node.EXCLUSIV既为独占模式。
  • acquiredQueued:当前线程获取锁失败后,就进入了一个自旋的状态,如果在等待过程中被中断过(如timeout),就返回true(接着自我中断),否则返回false。也就是说当前线程进入同步队列后进入自旋状态,每个节点会自我观察,当条件满足时,获取到同步状态后,就可以从这个自旋过程中退出,否则一直执行下去。
  • 如果线程在等待过程中被中断过,它是不响应的。只是获取资源后才再进行自我中断selfInterrupt(),将中断补上。因为在等待队列中自旋状态的线程是不会响应中断的,它会把中断记录下来,如果在自旋时发生过中断,就返回true。然后就会执行selfInterrupt()方法,而这个方法就是简单的中断当前线程Thread.currentThread().interrupt();其作用就是补上在自旋时没有响应的中断。

tryAcquire在NonfairSync的实现

java 复制代码
final boolean nonfairTryAcquire(int acquires) {
            final Thread current = Thread.currentThread();
            // 获取锁同步状态
            int c = getState();
            if (c == 0) {// 0表示无锁状态
            // cas竞争锁,替换state的值改为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;
        }

acquiredQueued的实现

前线程进入同步队列后进入自旋状态,每个节点会自我观察,当条件满足时,获取到同步状态后,就可以从这个自旋过程中退出(返回interrupted为false),否则一直执行下去。也不是每个节点都有获取锁的资格,因为是FIFO的先进先出队列,acquireQueued方法保证了只有头部节点的后继节点才有资格去获取同步状态

ini 复制代码
final boolean acquireQueued(final Node node, int arg) {
       /* 标记是否成功拿到资源 */
       boolean failed = true;
        try {
            /* 中断标志*/
            boolean interrupted = false;
            /*  自旋,一个死循环 */
            for (;;) {
                /* 获取前线程的前驱节点*/
                final Node p = node.predecessor();
                /*当前线程的前驱节点是头结点,即该节点是第二个节点,且获取锁成功*/
                if (p == head && tryAcquire(arg)) {
                    /*将head指向该节点*/
                    setHead(node);
                   /* 方便GC回收垃圾 */
                    p.next = null; 
                    failed = false;
                   /*返回等待过程中是否被中断过*/
                    return interrupted;
                }
                /*获取失败,就需要阻塞了,线程就进入waiting状态,直到被unpark()*/
                // shouldParkAfterFailedAcquire ---> 检查上一个节点的状态,如果是 SIGNAL 就阻塞,否则就改成 SIGNAL。
                if (shouldParkAfterFailedAcquire(p, node) &&parkAndCheckInterrupt())
                    /*如果等待过程中被中断过一次,就标记为true*/
                    interrupted = true;
            }
        } finally {
            if (failed)
                cancelAcquire(node);
        }
    }

shouldParkAfterFailedAcquire(Node node,Node node):判断一个线程是否阻塞

arduino 复制代码
private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {
    int ws = pred.waitStatus;//拿到前驱节点的状态
    if (ws == Node.SIGNAL)
        //状态为SIGNAL,如果前驱节点处于等待状态,直接返回true
        return true;
    if (ws > 0) {
        /*
         * 如果前驱节点放弃了,那就一直往前找,直到找到最近一个正常等待的状态,并排在它的后边。
         * 注意:那些放弃的结点,由于被自己"加塞"到它们前边,它们相当于形成一个无引用链,稍后就会被GC回收,这个操作实际是把队列中的cancelled节点剔除掉
         */
        do {
            node.prev = pred = pred.prev;
        } while (pred.waitStatus > 0);
        pred.next = node;
    } else {
         //如果前驱节点正常,那就把前驱的状态通过CAS的方式设置成SIGNAL
        compareAndSetWaitStatus(pred, ws, Node.SIGNAL);
    }
    return false;
}
  • 1.如果当前线程的前驱节点状态为SINNAL,则表明当前线程需要被阻塞,调用unpark()方法唤醒,直接返回true,当前线程阻塞
  • 2.如果当前线程的前驱节点状态为CANCELLED(ws > 0),则表明该线程的前驱节点已经等待超时或者被中断了,则需要从CLH队列中将该前驱节点删除掉,直到回溯到前驱节点状态 <= 0 ,返回false
  • 3.如果前驱节点非SINNAL,非CANCELLED,则通过CAS的方式将其前驱节点设置为SINNAL,返回false

整个流程中,如果前驱结点的状态不是SIGNAL,那么自己就不能安心去休息,需要去找个安心的休息点(前驱节点状态 <= 0 ),也就是只有当前驱节点为SIGNAL时这个线程才可以进入等待状态。shouldParkAfterFailedAcquire ---> 检查上一个节点的状态,如果是 SIGNAL 就阻塞,否则就改成 SIGNAL。

上面如果 shouldParkAfterFailedAcquire(Node pred, Node node) 方法返回true,则调用parkAndCheckInterrupt()方法阻塞当前线程:

parkAndCheckInterrupt()阻塞当前线程

前面的方法是判断是否阻塞,而这个方法就是真正的执行阻塞的方法同时返回中断状态

arduino 复制代码
private final boolean parkAndCheckInterrupt() {
        //调用park()使线程进入waiting状态
          LockSupport.park(this); 
          //如果被唤醒,查看自己是不是被中断的
          return Thread.interrupted();
 }

parkAndCheckInterrupt() 方法主要是把当前线程挂起,从而阻塞住线程的调用栈,同时返回当前线程的中断状态。

release(int arg):独占式的释放锁

java 复制代码
public final boolean release(int arg){
            if(tryRelease(arg)){//子类自定义实现
                Node h = head;
                if(h != null && h.waitStatus != 0){
                     unparkSuccessor(h);//唤醒下一个节点
                }
                return true;
            }
           return false;
}

释放锁的流程很简单,首先子类自定义的方法如果释放了同步状态,如果头节点不为空并且头节点的等待状态不为0就唤醒其后继节点。主要依赖的就是子类自定义实现的释放操作。

unparkSuccessor(Node node):唤醒后继节点获取同步状态

scss 复制代码
private void unparkSuccessor(Node node){
           //获取头节点的状态
           int ws = node.waitStatus;
           if(ws < 0){
              compareAndSetWaitStatus(node,ws,0);//通过CAS将头节点的状态设置为初始状态
           }
           Node s = node.next;//后继节点
           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);//真正的唤醒操作
           }
        }

唤醒操作,通过判断后继节点是否存在,如果不存在就寻找等待时间最长的适合的节点将其唤醒唤醒操作通过LockSupport中的unpark方法唤醒底层也就是unsafe类的操作。

锁释放移除节点

head节点表示获取锁成功的节点,当头结点在释放同步状态时,会唤醒后继节点,如果后继节点获得锁成功,会把自己设置为头结点,节点变化过程如下:

CAS设置为尾结点

这个过程涉及到2个变化

  • 设置头结点指向下一个获得锁的节点
  • 新的获得锁的节点,将pre指针指向null
  • 将原来head节点的next设置为null,即断开原head节点的next引用

acquire方法流程总结

  • 首先通过子类判断是否获取了锁,如果获取了就什么也不干。tryAcquire

  • 如果没有获取锁、通过线程创建节点加入同步队列的队尾。addWaiter

  • 当线程在同步队列中不断的通过自旋去获取同步状态,如果获取了锁,就把其设为同步队列中的头节点,否则在同步队列中不停的自旋等待获取同步状态 acquireQueued,shouldParkAfterFailedAcquire(Node pre,Node node),parkAndCheckInterrupt()

  • 如果在获取同步状态的过程中被中断过最后自行调用interrupted方法进行中断操作

为什么 AQS 需要一个虚拟 head 节点

每个节点都必须设置前置节点的 ws 状态为 SIGNAL(-1),因为每个节点在休眠前,都需要将前置节点的 ws 设置成 SIGNAL。否则自己永远无法被唤醒,所以必须要一个前置节点,而这个前置节点,实际上就是当前持有锁的节点。

由于第一个节点他是没有前置节点的,就创建一个假的。

总结下来就是:每个节点都需要设置前置节点的 ws 状态(这个状态为是为了保证数据一致性),而第一个节点是没有前置节点的,所以需要创建一个虚拟节点。

LockSupport 在AQS上的用处

LockSupport是Java中的一个工具类,提供了一些用于线程操作和等待/唤醒操作的方法。在AQS(AbstractQueuedSynchronizer)中,LockSupport主要用于以下两个方面:

线程等待/唤醒操作: AQS中的节点(Node)通常包含一个线程对象和一个运行状态(state)等信息。当线程获取不到锁时,它会将自己插入到等待队列中,通过调用LockSupport.park()方法使线程进入等待状态。而当有其他线程释放了锁时,AQS会通过调用LockSupport.unpark()方法来唤醒等待队列中的某个线程,使其重新进入运行状态。

线程上下文切换的优化: LockSupport类提供了一些方法,如unpark()和park()等,可以用于线程的等待/唤醒操作。这些方法相对于直接的线程睡眠(Thread.sleep()或Object.wait())有更好的上下文切换性能。在AQS中,当线程在等待或唤醒时,会使用这些方法来减少上下文切换的开销,从而提高性能。

总之,LockSupport类在AQS中的主要用处是协助实现线程的等待/唤醒操作,并提供优化的上下文切换机制,以实现更高性能的同步和并发控制。

场景分析

如果同时有三个线程 并发抢占锁,此时线程一 抢占锁成功,线程二线程三抢占锁失败,具体执行流程如下:

此时AQS内部数据为:

线程二线程三加锁失败:

如图可以看出,等待队列中的节点Node是一个双向链表,这里SIGNAL是Node中waitStatus属性,Node中还有一个nextWaiter属性,这个并未在图中画出来,这个到后面Condition会具体讲解的。

非公平锁执行流程:

这里我们还是用之前的线程模型来举例子,当线程二 释放锁的时候,唤醒被挂起的线程三线程三 执行tryAcquire()方法使用CAS操作来尝试修改state值,如果此时又来了一个线程四也来执行加锁操作,同样会执行tryAcquire()方法。

这种情况就会出现竞争,线程四 如果获取锁成功,线程三 仍然需要待在等待队列中被挂起。这就是所谓的非公平锁线程三 辛辛苦苦排队等到自己获取锁,却眼巴巴的看到线程四插队获取到了锁。

公平锁执行流程:

公平锁在加锁的时候,会先判断AQS等待队列中是存在节点,如果存在节点则会直接入队等待

Condition 实现原理

Condition 简介

上面已经介绍了AQS所提供的核心功能,当然它还有很多其他的特性,这里我们来继续说下Condition这个组件。

javascript 复制代码
Condition`是在`java 1.5`中才出现的,它用来替代传统的`Object`的`wait()`、`notify()`实现线程间的协作,相比使用`Object`的`wait()`、`notify()`,使用`Condition`中的`await()`、`signal()`这种方式实现线程间协作更加安全和高效。因此通常来说比较推荐使用`Condition

其中AbstractQueueSynchronizer中实现了Condition中的方法,主要对外提供await()和signal()调用。

csharp 复制代码
public class ReentrantLockDemo {
    static ReentrantLock lock = new ReentrantLock();

    public static void main(String[] args) {
        Condition condition = lock.newCondition();

        new Thread(() -> {
            lock.lock();
            try {
                System.out.println("线程一加锁成功");
                System.out.println("线程一执行 await 被挂起");
                condition.await();
                System.out.println("线程一被唤醒成功");
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                lock.unlock();
                System.out.println("线程一释放锁成功");
            }
        }).start();

        new Thread(() -> {
            lock.lock();
            try {
                System.out.println("线程二加锁成功");
                condition.signal();
                System.out.println("线程二唤醒线程一");
            } finally {
                lock.unlock();
                System.out.println("线程二释放锁成功");
            }
        }).start();
    }
}

代码执行流程图:

同步阻塞队列存放的都是竞争锁失败的线程,主要表征的是线程之间的竞争、互斥,而条件等待队列中存储的是因为某一个条件不满足而需要阻塞的线程,通常需要被其他线程主动唤醒,主要表征的是线程协作。

总结

总结的来说:线程获取锁,如果获取了锁就 保存当前获得锁的线程,如果没获取就创造一个节点通过compareAndSetTail(CAS操作)操作的方式将创建的节点加入同步队列的尾部,在同步队列中的节点通过自旋的操作不断去获取同步状态【当然由于FIFO先进先出的特性】等待时间越长就越先被唤醒。当头节点释放同步状态的时候,首先查看是否存在后继节点,如果存在就唤醒自己的后继节点,如果不存在就获取等待时间最长的符合条件的线程。

相关推荐
Estar.Lee3 小时前
查手机号归属地免费API接口教程
android·网络·后端·网络协议·tcp/ip·oneapi
2401_857610034 小时前
SpringBoot社团管理:安全与维护
spring boot·后端·安全
凌冰_5 小时前
IDEA2023 SpringBoot整合MyBatis(三)
spring boot·后端·mybatis
码农飞飞5 小时前
深入理解Rust的模式匹配
开发语言·后端·rust·模式匹配·解构·结构体和枚举
一个小坑货5 小时前
Rust 的简介
开发语言·后端·rust
monkey_meng6 小时前
【遵守孤儿规则的External trait pattern】
开发语言·后端·rust
Estar.Lee6 小时前
时间操作[计算时间差]免费API接口教程
android·网络·后端·网络协议·tcp/ip
新知图书6 小时前
Rust编程与项目实战-模块std::thread(之一)
开发语言·后端·rust
盛夏绽放7 小时前
Node.js 和 Socket.IO 实现实时通信
前端·后端·websocket·node.js
Ares-Wang7 小时前
Asp.net Core Hosted Service(托管服务) Timer (定时任务)
后端·asp.net