ReentrantLock和AQS原理机制浅探（待续）

title: AQS原理机制浅探
date: 2019-11-18 16:14:00
tags:

• AQS

AQS原理机制（with源码）

源码探AQS原理机制

AQS(AbstractQueuedSynchronizer)是concurrent包下用来构建锁或其他同步组件（信号量、事件等）的基础框架类。 JDK中许多并发工具类的内部实现都依赖于AQS，如ReentrantLock, Semaphore, CountDownLatch等等。

自定义同步器在实现时只需要实现共享资源state的获取与释放方式即可，至于具体线程等待队列的维护（如获取资源失败入队/唤醒出队等），AQS已经在顶层实现好了。自定义同步器实现时主要实现以下几种方法：

• isHeldExclusively()：该线程是否正在独占资源。只有用到condition才需要去实现它。
• tryAcquire(int)：独占方式。尝试获取资源，成功则返回true，失败则返回false。
• tryRelease(int)：独占方式。尝试释放资源，成功则返回true，失败则返回false。
• tryAcquireShared(int)：共享方式。尝试获取资源。负数表示失败；0表示成功，但没有剩余可用资源；正数表示成功，且有剩余资源。
• tryReleaseShared(int)：共享方式。尝试释放资源，如果释放后允许唤醒后续等待结点返回true，否则返回false。

todo:按调用顺序分方法拷贝出await()、signal()等源码添加阅读中文注释，包括方法头上 1. 2. 3.步总结做了什么操作，然后在方法内关键代码行上也添加阅读注释，参考如下

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UCHysqbt-1580121420273)(https://i.imgur.com/v4SH0J6.png)]

Condition原理：

await():

signal():

确认持有独占锁后调用enq()将等待队列中的Node节点移动到同步器同步队列中，并且通过LockSupport.unpark(node.thread)唤醒节点中的线程参与同步队列中获取同步状态(即锁)的竞争中，同时await()中因不再满足判断不在同步队列中条件，退出while循环，acquireQueued()成功获取到同步状态（锁）后继续往下执行，从await()方法返回,原线程不再阻塞继续往下执行

        public final void await() throws InterruptedException {
            if (Thread.interrupted())
                throw new InterruptedException();
            Node node = addConditionWaiter();
            int savedState = fullyRelease(node);
            int interruptMode = 0;
            //signal()移动到同步队列后不满足条件退出while循环
            while (!isOnSyncQueue(node)) {
                LockSupport.park(this);
                if ((interruptMode = checkInterruptWhileWaiting(node)) != 0)
                    break;
            }
            //acquireQueued()成功获取到同步状态（锁）后继续往下执行，从await()方法返回,原线程不再阻塞继续往下执行
            if (acquireQueued(node, savedState) && interruptMode != THROW_IE)
                interruptMode = REINTERRUPT;
            if (node.nextWaiter != null) // clean up if cancelled
                unlinkCancelledWaiters();
            if (interruptMode != 0)
                reportInterruptAfterWait(interruptMode);
        }

CLH自旋锁能确保无饥饿性，提供先来先服务的公平性。互斥锁资源被占用时会引起调用者睡眠，而自旋锁不会 会一直循环获取。

列举JDK中几种常见使用了AQS的同步组件：

• ReentrantLock: 使用了AQS的独占获取和释放,用state变量记录某个线程获取独占锁的次数,获取锁时+1，释放锁时-1，在获取时会校验线程是否可以获取锁。
• Semaphore: 使用了AQS的共享获取和释放，用state变量作为计数器，只有在大于0时允许线程进入。获取锁时-1，释放锁时+1。用作控制并发的数量，如数据库连接池等需要限制数量的场景，如大小为10的线程池限制DB连接数最大为3private static Semaphore semaphore = new Semaphore(3);。
• CountDownLatch: 使用了AQS的共享获取和释放，用state变量作为计数器，在初始化时指定。只要state还大于0，获取共享锁会因为失败而阻塞，直到计数器的值为0时，共享锁才允许获取，所有等待线程会被逐一唤醒。

ReentrantLock独占模式获取锁过程：

tryAcquire()尝试进行加锁

先getState判断状态，如果状态为0即为无锁状态。

​ 如果为公平锁，则先判断是否需要排队，判断依据为head节点的下一个非取消（waitStatus<=0）节点不为null且节点中的thread线程不为null。如果不需要排队返回false，if (!hasQueuedPredecessors() && compareAndSetState(0, acquires))，取反true则直接尝试CAS变更state状态值加锁，如果加锁成功则设置独占线程为当前线程。并且tryAcquire()返回true表示加锁成功，不需要再进行下面的入队排队等

需要排队时不用进行&&后面的CAS操作，不执行ifelse中代码而是继续往下判断state状态

如果状态不为0，即为加锁状态，则判断当前拥有独占锁的线程是否为当前线程

​ 如果是，则将state值再加1（因为此时持有锁，所以直接 setState(c) 不用CAS）。这也是为什么ReentrantLock可重入，相应的重入锁释放锁过程每次将state减1，减为0时完全释放锁。

​ 如果不是，则直接返回false表示加锁失败，进行入队排队。

    /**
     * Creates and enqueues node for current thread and given mode.
     *
     * @param mode Node.EXCLUSIVE for exclusive, Node.SHARED for shared
     * @return the new node
     */
    private Node addWaiter(Node mode) {
        Node node = new Node(mode);

        for (;;) {
            Node oldTail = tail;
            if (oldTail != null) {
                //设置node的前节点为tail
                node.setPrevRelaxed(oldTail);
                if (compareAndSetTail(oldTail, node)) {
                    //CAS替换tail为当前node，成功之后将旧tail节点的next指针指向当前节点，注意这两步之间可能有其他线程成为新的tail节点，此时next指针还未加上，所以比如释放锁时要从tail往前prev遍历才不会漏掉节点，enq()方法中同样如此
                    oldTail.next = node;
                    return node;
                }
            } else {
                initializeSyncQueue();
            }
        }
    }

/**
 * Initializes head and tail fields on first contention.
 * 当第一次争夺锁时初始化同步队列，也就是说，当第一个线程进来时队列未初始化，也不会初始化，因为此时没有竞争，
 * 看代码料想第一个之后的后续直接获取到锁的线程也是不会初始化同步队列
 */
private final void initializeSyncQueue() {
    Node h;
    if (HEAD.compareAndSet(this, null, (h = new Node())))
        tail = h;
}

参考文章：

AbstractQueuedSynchronizer源码解读

AbstractQueuedSynchronizer 原理分析 - 独占/共享模式

Java并发之AQS详解