AQS

java.util.concurrent.locks.AbstractQueuedSynchronizer

AQS （抽象队列同步器）： AbstractQueuedSynchronizer 是什么

• 来自jdk1.5，是用来实现锁或者其他同步器组件的公共基础部分的抽象实现，是重量级基础框架以及JUC的基石，主要用于解决锁分配给谁的问题
• 整体是通过一个抽象的FIFO队列来完成资源获取线程的排队工作，并通过一个int变量表示持有锁的状态
• 锁是面向开发人员的，而同步器是JDK统一规范并简化了锁的实现，并抽象出来的公共基础部分，屏蔽了同步状态、同步队列的管理，和线程排队、通知、唤醒等机制

和AQS相关的类：

• ReentranLock
• CountDownLatch
• ReentrantReadWriteLock
• Semaphore
• 等

AQS 原理

• 整体是通过一个抽象的FIFO队列来完成资源获取线程的排队工作，并通过一个int变量表示持有锁的状态
• 如果共享资源被占用，就需要阻塞唤醒机制来保证锁的分配，这个机制主要是通过CLH队列的变体实现的，将暂时获取锁失败的线程，以及自身的等待状态封装成队列的节点对象node，放入队列中
• 通过CAS、自旋等维护共享资源的状态，达到并发效果
• 内部结构：
  • 队列的 头指针、尾指针
  • int 类型的同步状态的标识 state ，默认值0代表没有被占用，大于等于1代表被占用
  • 内部类node，将暂时获取锁失败的线程，以及自身的等待状态封装成队列的节点对象node
    • int类型变量 waitStatus 当前节点再队列中的等待状态，默认为0
      • 1表示线程被取消
      • -1表示后继线程需要被唤醒
      • -2表示等待conditon唤醒
      • -3表示共享式（锁分为共享和独占）同步状态获取将无条件地传播下去
    • 前一个节点的指针和后一个节点的指针
    • 请求线程

ReentrantLock

• 是lock的实现类，构造器可以传入一个boolean值，true创建的就是公平锁，false为非公平锁，默认为非公平锁
• 内部有一个静态内部类sync，继承了 AbstractQueuedSynchronizer ，用于锁的各种操作

lock

• 当调用 lock 方法加锁时

  • 非公平锁，会先尝试通过cas 比较并交换的操作把 states 的状态值从 0更新为1，如果更新成功，就把持有锁的线程设置为自己
  • 更新失败就和公平锁一样，执行 AQS 的 acquire方法
  • 除此之外，公平和非公平锁的区别就是，再获取同步状态时，公平锁需要判断等待队列中再自己之前是否存在有效节点，如果有公平锁就需要排队
  • 因为公平锁讲究先到先得，线程再获取锁时，如果这个锁的等待队列已经有线程再等待，当前线程就会直接进入等待队列
  • 而非公平锁，不管是否有队列，如果可以获取锁，就会立刻占有锁的对象，所以第一个在队列里排队的线程苏醒后，仍然需要去竞争锁，且不一定能竞争到锁

acquire 方法源码：

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

acquire

• 调用lock方法加锁，除非是非公平锁能直接拿到锁，其他情况下都是在调用acquire 方法

• acquire 方法分为三种情况：

• 调用 tryAcquire 方法尝试加锁；

• 加锁失败，调用 addWaite方法，进入等待队列；

• 进入队列之后，调用acquireQueued 方法，线程进入阻塞状态，等待唤醒后才能继续执行

非公平锁的 tryAcquire 方法源码

• AQS类的tryAcquire方法只是做了规范，方法内直接抛出异常，所以这个方法需要由子类去实现

• 非公平锁的tryAcquire 方法会先判断锁的状态state是否为0，为0说明没有被其他线程占用，就立即使用cas操作变更state为1，变更成功就把持有锁的线程设置为自己，变更失败就表示加锁失败

• 如果锁的状态为1，说明锁已经被占用，在比较当前线程和持有锁的线程是否一致，不一致就加锁失败

• tryAcquire 方法公平和非公平锁的区别是

  • 再获取同步状态时，公平锁需要判断等待队列中再自己之前是否存在有效节点，如果有公平锁就需要排队
  • 非公平锁，不管是否有队列，如果可以获取锁，就会立刻占有锁的对象

• 如果 tryAcquire 方法抢锁失败，就需要调用 addWaiter加入到等待队列

        final boolean nonfairTryAcquire(int acquires) {final Thread current = Thread.currentThread();int c = getState();//先判断锁的状态state是否为0，为0说明没有被其他线程占用if (c == 0) {//为0说明没有被其他线程占用，使用cas操作变更state为1if (compareAndSetState(0, acquires)) {//变更成功就把持有锁的线程设置为自己，变更失败就表示加锁失败setExclusiveOwnerThread(current);return true;}}//如果锁的状态为1，说明锁已经被占用，在比较当前线程和持有锁的线程是否一致else if (current == getExclusiveOwnerThread()) {int nextc = c + acquires;if (nextc < 0) // overflowthrow new Error("Maximum lock count exceeded");setState(nextc);return true;}//加锁失败return false;}

addWaiter(Node.EXCLUSIVE) 加入等待队列 源码：

• Node.EXCLUSIVE 代表的是独占的节点，也就是排他锁
• acquire 方法的 addWaiter 方法创建的是独占的node节点，节点中封装的是当前线程
• addWaiter方法首先要判断 链表的尾指针是否为空
  • 如果为空，就需要初始化链表，首先new一个空的哨兵节点，这个节点并不存储信息，只是作为占位使用，然后设置哨兵节点为头节点，然后把头节点赋值给尾节点
  • 当链表初始化完成后，或者链表中已经由其他节点时，就用CAS操作把新节点加入到链表尾部，如果节点加入链表失败就进行下一次循环，直到把节点加入成功为止
  • 如果不为空，直接用CAS操作把新节点加入到链表尾部，同样如果节点加入链表失败就进行循环，直到把节点加入成功为止
• 节点成功入队后，需要调用acquireQueued 方法

private Node addWaiter(Node mode) {//node节点中封装的是当前线程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;}}enq(node);return node;
}private Node enq(final Node node) {for (;;) {//尾节点Node t = tail;//如果尾节点为nullif (t == null) { // Must initialize//就需要new一个node节点，并且设置为头节点，然后把头节点赋值给尾节点if (compareAndSetHead(new Node()))tail = head;} else {//当链表初始化完成后，或者链表中已经由其他节点时//把要加入链表的新节点的前指针设置为尾节点node.prev = t;//并且把新加入的节点设置为尾节点if (compareAndSetTail(t, node)) {//设置成功就之前尾节点的后指针指向新节点，这样新节点就变成了新的尾节点，如果设置失败，就继续循环，直到把新节点加入到链表尾部为止t.next = node;return t;}}}}

acquireQueued 源码

• 首先获取当前节点的前置节点，如果前置节点是头节点，就尝试去获取锁
• 如果获取锁成功，就把自己设为头节点，就把锁的state改为1，设置当前线程为持有锁的线程
• 如果前置节点不是头节点，或者获取锁失败
  • 就需要判断前置节点的waitStatus状态值，waitStatus值默认为0，第一次进入循环，会把前置节点的waitStatus的值改为-1后，继续下一次循环后，会调用 LockSupport.park 方法阻塞当前线程，需要等待其他线程释放锁后，再唤醒阻塞的线程
  • 当持有锁的线程释放锁，且调用LockSupport.unpark 唤醒该线程后才能继续执行，LockSupport.unpark 唤醒的是头节点的下一个节点
  • 线程被唤醒后,检查线程是否被中断，如果线程没有被中断，就继续进行循环
  • 继续尝试去加锁，因为是非公平锁，所以有可能会加锁失败
    • 如果加锁成功，就把锁的state改为1，设置当前线程为持有锁的线程，并且把当前线程的节点设置为链表的头节点，原本的头节点会从链表中剔除
    • 因为每次唤醒的都是头节点的下一个节点，所以成功抢到到锁后，被唤醒的节点会成为新的头节点，后续会唤醒链表的下一个节点
• 如果线程在等待过程中取消，没有获取到锁就跳出了循环，failed值为默认的true，就会执行cancelAcquire方法，取消正在排队的节点
  • 首先设置当前节点的线程为null，然后获取上一个没有取消的前置节点，
  • 把当前节点的 waitStatus 设置为1（1就是要取消的节点）
  • 如果当前节点是尾节点，就把上一个有效的节点设置为尾节点
  • 如果不是尾节点，并且满足出队条件，就变更链表中相关节点的前置和后置引用，剔除要取消的节点

	//arg为1，独占锁final boolean acquireQueued(final Node node, int arg) {boolean failed = true;try {boolean interrupted = false;for (;;) {//获得node节点的前置节点final Node p = node.predecessor();//node节点的前置节点是否为头节点，如果是就尝试去获取锁if (p == head && tryAcquire(arg)) {setHead(node);p.next = null; // help GCfailed = false;return interrupted;}//判断node节点的前置节点的waitStatus状态，默认情况下都是0，在第二次循环的时候，就会改成-1，然后执行parkAndCheckInterrupt方法//parkAndCheckInterrupt方法会阻塞当前线程//也就是后面的节点会把前面节点的 waitStatus 改为-1if (shouldParkAfterFailedAcquire(p, node) &&parkAndCheckInterrupt())interrupted = true;}} finally {if (failed)cancelAcquire(node);}}/*** waitStatus 当前节点再队列中的等待状态默认为01表示线程获取锁的请求被取消-1表示线程已经准备好了-2表示节点在等待队列中，等待唤醒-3表示共享式（锁分为共享和独占）同步状态获取将无条件地传播下去*/private static boolean shouldParkAfterFailedAcquire(Node pred, Node node) {//前置节点的状态int ws = pred.waitStatus;// SIGNAL= -1 ， 当线程再次进行循环的时候，前一个节点的waitStatus已经被设置为-1，就返回trueif (ws == Node.SIGNAL)return true;//线程被取消if (ws > 0) {do {node.prev = pred = pred.prev;} while (pred.waitStatus > 0);pred.next = node;} else {//如果前置节点的 waitStatus不等于-1也不大于0，就把waitStatus的值改为-1后，返回falsecompareAndSetWaitStatus(pred, ws, Node.SIGNAL);}return false;}//阻塞当前线程private final boolean parkAndCheckInterrupt() {//验证当前线程的通行证，阻塞当前线程LockSupport.park(this);//被唤醒后,检查线程是否被中断，如果线程没有被中断，就返回 falsereturn Thread.interrupted();}

取消正在进行的获取尝试

	// node 为需要取消的节点private void cancelAcquire(Node node) {// Ignore if node doesn't existif (node == null)return;//设置当前节点的线程为nullnode.thread = null;//获取上一个节点Node pred = node.prev;//waitStatus > 0 ，表示上一个节点也要取消while (pred.waitStatus > 0)//那么就一直向上找，直到找到没有取消的前置节点node.prev = pred = pred.prev;//获取不会取消的前置节点的下一个节点Node predNext = pred.next;//把当前节点的 waitStatus 设置为1，1就是要取消的节点node.waitStatus = Node.CANCELLED;//如果当前节点是尾节点，就把上一个还有效的节点设置为尾节点if (node == tail && compareAndSetTail(node, pred)) {//设置成功，就把上一个节点的后置节点设置为null,这样上一个还有效的节点就成为了尾节点compareAndSetNext(pred, predNext, null);} else {//否则int ws;//前置节点不能是头节点，因为头节点只是占位节点，并且满足出队条件if (pred != head &&((ws = pred.waitStatus) == Node.SIGNAL ||(ws <= 0 && compareAndSetWaitStatus(pred, ws, Node.SIGNAL))) &&pred.thread != null) {//变更链表中相关节点的前置和后置引用，剔除要取消的节点Node next = node.next;if (next != null && next.waitStatus <= 0)compareAndSetNext(pred, predNext, next);} else {unparkSuccessor(node);}node.next = node; // help GC}}

unlock

unlock 源码，其实是再调用release方法

    public void unlock() {sync.release(1);}

release方法会首先尝试释放锁

• tryRelease 会把持有锁的线程为null，并且把锁的state设置为0
• 如果链表被初始化过，有在等待的线程节点，头节点就不为空，且waitStatus值为-1
• 接下来会把头节点的waitStatus的改为0，如果头节点的下一个节点不为null，就调用LockSupport.unpark 方法，唤醒头节点的下一个节点

    public final boolean release(int arg) {if (tryRelease(arg)) {Node h = head;if (h != null && h.waitStatus != 0)unparkSuccessor(h);return true;}return false;}

AQS的tryRelease方法，同样没有做实现，需要子类自己去实现，下面是ReentrantLock的实现

		protected final boolean tryRelease(int releases) {//传入的releases为1，持有锁的线程State为1，所以C为0int c = getState() - releases;//如果当前线程不等于持有锁的线程会抛出异常，这种情况一般不会出现if (Thread.currentThread() != getExclusiveOwnerThread())throw new IllegalMonitorStateException();boolean free = false;//c等于0，就设置持有锁的线程为null,并且把state设置为0，返回trueif (c == 0) {free = true;setExclusiveOwnerThread(null);}setState(c);return free;}

unparkSuccessor

    private void unparkSuccessor(Node node) {int ws = node.waitStatus;if (ws < 0)//重新把头节点的waitStatus值改为0compareAndSetWaitStatus(node, ws, 0);//头节点的下一个节点Node s = node.next;//如果链表被初始化过，有在等待的线程节点，头节点的后置节点就不为null//如果链表后面还有其他节点，那么头节点的后置节点waitStatus值就为-1if (s == null || s.waitStatus > 0) {s = null;for (Node t = tail; t != null && t != node; t = t.prev)if (t.waitStatus <= 0)s = t;}//如果头节点的下一个节点不为null，就直接调用 LockSupport.unpark 方法，唤醒头节点的下一个节点if (s != null)LockSupport.unpark(s.thread);}