Java并发编程之AQS

多线程编程中锁和线程的相互协作的Lock/Semaphore 很类似，都是拿到一个锁或凭证，然后进入临界区，完事儿后释放锁。都需

1. 管理状态的原子性
2. 线程的阻塞与解除
3. 管理队列

如果每个并发工具类都是给自维护这些功能，将有很多重复代码，大佬们当然不会这么干，所以就出现了AQS （AbstractQueuedSynchronizer），作为一个用于构建锁和同步器等线程协作工具类的框架，基于它就可以开发新的协作类。

管理状态的原子性

在信号量中需要一个state，表示剩余许可证的数量。
在CountdownLatch 里需要一个state，表示需要倒数的数量，一直减到0，然后闸门打开。
在ReentrantLock 需要state 表示锁的占有情况，0是没有线程占有，1 表示被一个线程持有，1以上表示线程重入多次持有锁。

既然这么多工具类需要维护state，AQS 就把这个功能提取了。

private volatile int state;
protected final boolean compareAndSetState(int expect, int update) {
    return unsafe.compareAndSwapInt(this, stateOffset, expect, update);
}

管理队列

既然有多个线程争抢，就需要有一个队列来管理没有拿到锁的队列，

获取与释放

多个线程争抢，需要一个获取方法
对于 Lock 类，lock就是获取方法，

 final void lock() {
            if (compareAndSetState(0, 1))
                setExclusiveOwnerThread(Thread.currentThread());
            else
                acquire(1);
        }

对于 Semaphore acquire 方法就是其中一个“获取方法” ，
对于 CountDownLatch 获取方法就是 await 方法（包含重载方法）。
不同的协作方式对获取方法有不同的操作方式，但多和state 相关，通常也会将线程设置为阻塞状态。
相反地，释放方法就是获取方法的对立面。

从CountDownlatch 看看具体怎么用AQS

1. 内部定义Sync类（可能也有NonfairSync），继承AQS
2. 在Sync中，是独占则重写 tryAcquire 和 tryRelease，非独占则重写 tryAcquireShared 和 tryReleaseShared 方法
3. 在工具类中实现获取/释放的逻辑。

具体看源码

public class CountDownLatch {
    /**
     * Synchronization control For CountDownLatch.
     * Uses AQS state to represent count.
     */

    private static final class Sync extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = 4982264981922014374L;
        Sync(int count) {
            setState(count);
        }
        int getCount() {
            return getState();
        }
        protected int tryAcquireShared(int acquires) {
            return (getState() == 0) ? 1 : -1;
        }
        protected boolean tryReleaseShared(int releases) {
            // Decrement count; signal when transition to zero
            for (;;) {
                int c = getState();
                if (c == 0)
                    return false;
                int nextc = c-1;
                if (compareAndSetState(c, nextc))
                    return nextc == 0;
            }
        }
    }
    private final Sync sync;
   //省略其他代码...
}

参考文献

1. http://gee.cs.oswego.edu/dl/papers/aqs.pdf
2. https://javadoop.com/post/AbstractQueuedSynchronizer