AQS 就是一个抽象类,主要用来构建锁和同步器。使用 AQS 能简单且高效地构造出应用广泛的大量的同步器,比如 ReentrantLock
,Semaphore,
ReentrantReadWriteLock等等。
AQS 的原理是什么?
如果被请求的共享资源空闲,则将当前请求资源的线程设置为有效的工作线程,并且将共享资源设置为锁定状态。将暂时获取不到锁的线程加入到阻塞等待队列中。 AQS 是用 CLH 队列锁 实现的,将每条请求共享资源的线程封装成一个 CLH 锁队列(虚拟双向队列,不存在队列实例)的一个结点(Node)来实现锁的分配。在 CLH 同步队列中,一个节点表示一个线程,它保存着线程的引用(thread)、 当前节点在队列中的状态(waitStatus)、前驱节点(prev)、后继节点(next)。
CLH 队列结构
AQS 核心原理图
其中state
变量由 volatile
修饰,用于展示当前临界资源的获锁情况。
Semaphore
Semaphore
是共享锁的一种实现,它默认构造 AQS 的 state
值为 permits(资源数量),
通常用于那些资源有明确访问数量限制的场景(仅限单机模式)。当初始的资源个数为 1 的时候,Semaphore
退化为独占锁。
调用semaphore.acquire()
,线程尝试获取许可证,如果 state >= 0
的话,则表示可以获取成功。如果获取成功的话,使用 CAS 操作去修改 state
的值 state=state-1
。如果 state<0
的话,则表示许可证数量不足。此时会创建一个 Node 节点加入阻塞队列,挂起当前线程。
调用semaphore.release();
,线程尝试释放许可证,并使用 CAS 操作去修改 state
的值 state=state+1
。释放许可证成功之后,同时会唤醒同步队列中的一个线程。被唤醒的线程会重新尝试去修改 state
的值 state=state-1
,如果 state>=0
则获取令牌成功,否则重新进入阻塞队列,挂起线程。
ReentrantLock
ReentrantLock
的state
初始值为 0,表示未锁定状态。A 线程 lock()
时,会调用 tryAcquire()
独占该锁并将 state+1
。此后,其他线程再 tryAcquire()
时就会失败,直到 A 线程 unlock()
到 state=
0(即释放锁)为止,A 线程获取多少次锁就要释放多少次。
CountDownLatch
CountDownLatch
将任务分为 N 个子线程去执行,state
也初始化为 N( N 与线程个数一致)。这 N 个子线程是并行执行的,每个子线程执行完后countDown()
一次,state 会 CAS(Compare and Swap) 减 1。等到所有子线程都执行完后(即 state=0
),会 unpark()
主调用线程,然后主调用线程就会从 await()
函数返回,继续后余动作。CountDownLatch
允许 count 个线程阻塞在一个地方,直至所有线程的任务都执行完毕在执行后续的任务。
注意:CountDownLatch
是一次性的,计数器的值只能在构造方法中初始化一次,之后没有任何机制再次对其设置值。
CyclicBarrier
CyclicBarrier
和 CountDownLatch
非常类似,但是它的功能比 CountDownLatch
更加复杂和强大。
CyclicBarrier和CountDownLatch区别
1. CountDownLatch的await()线程会等待计数器减为0,而执行CyclicBarrier的await()方法会使所有线程进入阻塞等待其他线程到达障点。
2. CountDownLatch计数器不能重置,CyclicBarrier可以重置循环利用。
3. CountDownLatch是基于AQS的共享模式实现的,CyclicBarrier是基于ReentrantLock和Condition实现的。
4. CountDownLatch不会让子线程进入阻塞,CyclicBarrier会使所有子线程进入阻塞。