J.U.C--locks--AQS分析

看一下AbstractQueuedSynchronizer(下面简称AQS)的子类就行知道，J.U.C中宣传的封装良好的同步工具类Semaphore、CountDownLatch、ReentrantLock、ReentrantReadWriteLock、FutureTask等尽管各自都有不同特征，可是其内部的实现都与AQS分不开。

所以分析AQS的实现原理对其余显示锁或则同步工具类的理解非常重要。

1.主要属性和内部类

这一篇blog主要分析AQS的实现中的重要属性和内部类。尤其是对于ReentrantLock和ReentrantReadWriteLock。其lock()方法和unlock()方法的实现终于都是由AQS同步器实现的。由此可见分析AQS类的重要性可见一斑。

在AQS中，我们先看属性远比看方法来的更加easy理解这个类的作用。首先看AQS类的主要属性：

//等待队列的头指针
private transient volatile Node head;
//等待队列的尾指针
private transient volatile Node tail;
//同步器的状态位，注意这里state是声明了volatile。保证了可视性
private volatile int state;

凝视事实上已经告诉我们了。Node类型的 head 和 tail 是一个FIFO的wait queue。一个int类型的状态位state。到这里也能猜到AQS对外呈现（或者说声明）的主要行为就是由一个状态位和一个有序队列来配合完毕。

state属性

对于state状态的管理，在AQS中仅仅通过三个方法来实现：

java.util.concurrent.locks.AbstractQueuedSynchronizer.getState();

java.util.concurrent.locks.AbstractQueuedSynchronizer.setState(int);

java.util.concurrent.locks.AbstractQueuedSynchronizer.compareAndSetState(int, int);

前面两个函数事实上就是get和set方法。

第三个函数事实上是通过Unsafe类实现CAS设置状态值，CAS+volatile 保证了state变量的线程安全。

Node结点

前面还提到了同步器的实现还依赖于一个FIFO的队列。队列中的元素Node就是保存着线程引用和线程状态的容器，每一个线程对同步器的訪问。都可以看做是队列中的一个节点。

Node类的源代码不多，我直接所有粘贴出来：

static final class Node {

    static final Node SHARED = new Node();

    static final Node EXCLUSIVE = null;

    static final int CANCELLED =  1;

    static final int SIGNAL    = -1;

    static final int CONDITION = -2;

    static final int PROPAGATE = -3;

    volatile int waitStatus;

    volatile Node prev;

    volatile Node next;

    volatile Thread thread;

    Node nextWaiter;

    final boolean isShared() {
        return nextWaiter == SHARED;
    }

    final Node predecessor() throws NullPointerException {
        Node p = prev;
        if (p == null)
            throw new NullPointerException();
        else
            return p;
    }
    /** 构造器 */
    Node() {    // Used to establish initial head or SHARED marker
    }
    /** 构造器 */
    Node(Thread thread, Node mode) {     // Used by addWaiter
        this.nextWaiter = mode;
        this.thread = thread;
    }
    /** 构造器 */
    Node(Thread thread, int waitStatus) { // Used by Condition
        this.waitStatus = waitStatus;
        this.thread = thread;
    }
}

Node类主要有5个属性：

volatile int waitStatus;//
volatile Node prev;//
volatile Node next;//
volatile Thread thread;//
Node nextWaiter;//

以上五个成员变量主要负责保存该节点的线程引用，同步等待队列（下面简称sync队列）的前驱和后继节点。同一时候也包括了同步状态。

对这5个变量的解释例如以下：

属性名称	描写叙述
int waitStatus	表示节点的状态。当中包括的状态有： 1.CANCELLED，值为1，表示当前的线程被取消。 2.SIGNAL，值为-1。表示当前节点的后继节点包括的线程须要执行，也就是unpark； 3.CONDITION，值为-2。表示当前节点在等待condition，也就是在condition队列中； 4.PROPAGATE，值为-3。表示当前场景下兴许的acquireShared可以得以执行； 5.值为0，表示当前节点在sync队列中。等待着获取锁。
Node prev	前驱节点。比方当前节点被取消，那就须要前驱节点和后继节点来完毕连接。
Node next	后继节点。
Node nextWaiter	存储condition队列中的后继节点。
Thread thread	入队列时的当前线程。

节点成为sync队列和condition队列构建的基础，在同步器中就包括了sync队列。

同步器拥有三个成员变量：sync队列的头结点head、sync队列的尾节点tail和状态state。对于锁的获取。请求形成节点，将其挂载在尾部。而锁资源的转移（释放再获取）是从头部開始向后进行。对于同步器维护的状态state，多个线程对其的获取将会产生一个链式的结构。

2.重要函数的源代码解析

获取锁相关函数

acquire(int arg);//以独占模式获取对象，忽略中断。
acquireInterruptibly(int arg);//以独占模式获取对象。假设被中断则中止。
acquireShared(int arg);//以共享模式获取对象，忽略中断。
acquireSharedInterruptibly(int arg);//以共享模式获取对象，假设被中断则中止。

tryAcquire(int arg);//试图在独占模式下获取对象状态。
tryAcquireNanos(int arg, long nanosTimeout);//试图以独占模式获取对象，假设被中断则中止。假设到了给定超时时间，则会失败。
tryAcquireShared(int arg);//试图在共享模式下获取对象状态。
tryAcquireSharedNanos(int arg, long nanosTimeout);//试图以共享模式获取对象，假设被中断则中止。假设到了给定超时时间。则会失败。

释放锁相关函数

release(int arg);//以独占模式释放对象。
releaseShared(int arg);//以共享模式释放对象

tryRelease(int arg);//试图设置状态来反映独占模式下的一个释放。
tryReleaseShared(int arg);//试图设置状态来反映共享模式下的一个释放。

1）acquire(int arg)函数

首先看看Javadoc的定义：

以独占模式获取对象，忽略中断。

通过至少调用一次 tryAcquire(int) 来实现此方法，并在成功时返回。否则在成功之前，一直调用 tryAcquire(int) 将线程加入队列，线程可能反复被堵塞或不被堵塞。可以使用此方法来实现 Lock.lock() 方法。

可知该函数是以独占模式获取对象而且忽略中断，完毕synchronized语义。

在AQS类中的源代码例如以下：

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

该函数主要完毕的逻辑例如以下：

1）首先调用tryAcquire(arg)函数尝试获取；

尝试更改状态state的值，而且保证原子性。

在tryAcquire方法中使用了同步器提供的对state操作的方法。利用compareAndSet保证仅仅有一个线程可以对状态进行成功改动，而没有成功改动的线程将进入sync队列排队。

值得注意的是这个函数在AQS中并没有实现，而是在其继承子类中实现（比方在ReentrantLock类中的内部类中NonfairSync和FairSync中均实现了这种方法）。

当获取成功时，就会返回true,这时源代码中的if语句就会直接执行if(0),也就是不满足执行条件。

2）假设获取不到，将当前线程构造成节点Node并加入sync队列。

进入队列的每一个线程都是一个节点Node，从而形成了一个双向队列，相似CLH队列，这样做的目的是线程间的通信会被限制在较小规模（也就是两个节点左右）。

3）再次尝试获取。假设没有获取到那么将当前线程从线程调度器上摘下。进入等待状态。

使用LockSupport将当前线程unpark，关于LockSupport兴许会具体介绍。

看看addWaiter()方法的逻辑：

凝视解释的是：通过给定的模式和当前线程创建同步队列结点。

源代码例如以下：

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;
        }
    }
    //
    enq(node);
    return node;
}

//Inserts node into queue, initializing if necessary. See picture above.
private Node enq(final Node node) {
    //死循环直至return
    for (;;) {
        Node t = tail;
        //必须初始化的步骤
        if (t == null) {
            if (compareAndSetHead(new Node()))
                tail = head;
        } else {
            node.prev = t;
            if (compareAndSetTail(t, node)) {
                t.next = node;
                return t;
            }
        }
    }
}

上面的逻辑解释：

1）使用当前线程构造Node；

对于一个新创建的节点须要做的是：将新节点前驱节点指向尾节点（current.prev = tail）。尾节点指向它（tail = current），原有的尾节点的后继节点指向它（t.next = current）而这些操作要求是原子的。上面的操作是利用尾节点的设置来保证的，也就是compareAndSetTail来完毕的。

2）先行尝试在队尾加入；

假设尾节点已经有了。然后做例如以下操作：

(1)分配引用pred指向尾节点。

(2)调用compareAndSetTail(pred, node)将新节点更新为尾节点；

(3)直接return，返回新插入的结点。

3）假设队尾加入失败或者是第一个入队的节点。

假设是第1个节点，也就是sync队列没有初始化，那么会进入到enq这种方法，进入的线程可能有多个，或者说在addWaiter中没有成功入队的线程都将进入enq这种方法。

enq(node)函数的逻辑是确保进入的Node都会有机会顺序的加入到sync队列中，而加入的过程例如以下：

(1)假设尾节点为空，那么原子化的分配一个头节点。并将尾节点指向头节点，这一步是初始化；

(2)然后是反复在addWaiter中做的工作，可是在一个for (;;)的循环中，直到当前节点入队为止。

至此。addWaiter()方法的逻辑分析完毕。接下来就是分析(final Node node, int arg) 方法的逻辑。

进入sync队列之后。接下来就是要进行锁的获取，或者说是訪问控制了。仅仅有一个线程可以在同一时刻继续的执行。而其它的进入等待状态。而每一个线程都是一个独立的个体，它们自省的观察。当条件满足的时候（自己的前驱是头结点而且原子性的获取了状态）。那么这个线程可以继续执行。

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)) {
                setHead(node);
                p.next = null; // help GC
                failed = false;
                return interrupted;
            }
            if (shouldParkAfterFailedAcquire(p, node) &&
                parkAndCheckInterrupt())
                interrupted = true;
        }
    } finally {
        if (failed)
            cancelAcquire(node);
    }
}

上述逻辑主要包括：

1. 获取当前节点的前驱节点；

须要获取当前节点的前驱节点。而头结点所相应的含义是当前站有锁且正在执行。

2. 当前驱节点是头结点而且可以获取状态，代表该当前节点占有锁；

假设满足上述条件，那么代表可以占有锁。根据节点对锁占有的含义，设置头结点为当前节点。

3. 否则进入等待状态。

假设没有轮到当前节点执行。那么将当前线程从线程调度器上摘下。也就是进入等待状态。

这里针对acquire做一下总结：

1. 状态的维护；

须要在锁定时。须要维护一个状态(int类型)。而对状态的操作是原子和非堵塞的，通过同步器提供的对状态訪问的方法对状态进行操纵，而且利用compareAndSet来确保原子性的改动。

2. 状态的获取；

一旦成功的改动了状态，当前线程或者说节点，就被设置为头节点。

3. sync队列的维护。

在获取资源未果的过程中条件不符合的情况下(不该自己，前驱节点不是头节点或者没有获取到资源)进入睡眠状态，停止线程调度器对当前节点线程的调度。

这时引入的一个释放的问题，也就是说使睡眠中的Node或者说线程获得通知的关键，就是前驱节点的通知，而这一个过程就是释放。释放会通知它的后继节点从睡眠中返回准备执行。

下面的流程图基本描写叙述了一次acquire所须要经历的过程：

如上图所看到的，当中的判定退出队列的条件，判定条件是否满足和休眠当前线程就是完毕了自旋spin的过程。

2）release(int arg)

首先看看Javadoc的定义：

以独占模式释放对象。假设 tryRelease(int) 返回 true。则通过消除一个或多个线程的堵塞来实现此方法。可以使用此方法来实现 Lock.unlock() 方法

源代码例如以下：

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

在unlock方法的实现中。使用了同步器的release方法。

相对于在之前的acquire方法中可以得出调用acquire，保证可以获取到锁（成功获取状态），而release则表示将状态设置回去，也就是将资源释放，或者说将锁释放。

上述逻辑主要包括：

1）尝试释放状态。

tryRelease()函数可以保证原子化的将状态设置回去。当然须要使用compareAndSet来保证。假设释放状态成功过之后。将会进入后继节点的唤醒过程。

2. 唤醒当前节点的后继节点所包括的线程。

通过LockSupport的unpark方法将休眠中的线程唤醒，让其继续acquire状态。

private void unparkSuccessor(Node node) {
    // 将状态设置为同步状态
    int ws = node.waitStatus;
    if (ws < 0)
        compareAndSetWaitStatus(node, ws, 0);
    // 获取当前节点的后继节点，假设满足状态，那么进行唤醒操作
    // 假设没有满足状态，从尾部開始找寻符合要求的节点并将其唤醒
    Node s = node.next;
    if (s == null || s.waitStatus &gt; 0) {
        s = null;
        for (Node t = tail; t != null &amp;&amp; t != node; t = t.prev)
            if (t.waitStatus &lt;= 0)
                s = t;
    }
    if (s != null)
        LockSupport.unpark(s.thread);
}

上述逻辑主要包括，该方法取出了当前节点的next引用，然后对其线程(Node)进行了唤醒。这时就仅仅有一个或合理个数的线程被唤醒。被唤醒的线程继续进行对资源的获取与争夺。

回想整个资源的获取和释放过程：

1)在获取时，维护了一个sync队列。每一个节点都是一个线程在进行自旋，而根据就是自己是否是首节点的后继而且可以获取资源；

2)在释放时。仅仅须要将资源还回去。然后通知一下后继节点并将其唤醒。

这里须要注意，队列的维护（首节点的更换）是依靠消费者（获取时）来完毕的，也就是说在满足了自旋退出的条件时的一刻。这个节点就会被设置成为首节点。

至此AQS基本的两个函数分析完毕。这两个函数也是lock()函数和unlock()函数的核心。