多线程编程中锁和线程的相互协作的Lock/Semaphore 很类似,都是拿到一个锁或凭证,然后进入临界区,完事儿后释放锁。都需
- 管理状态的原子性
- 线程的阻塞与解除
- 管理队列
如果每个并发工具类都是给自维护这些功能,将有很多重复代码,大佬们当然不会这么干,所以就出现了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
- 内部定义Sync类(可能也有NonfairSync),继承AQS
- 在Sync中,是独占则重写 tryAcquire 和 tryRelease,非独占则重写 tryAcquireShared 和 tryReleaseShared 方法
- 在工具类中实现获取/释放的逻辑。
具体看源码
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;
//省略其他代码...
}
参考文献
- http://gee.cs.oswego.edu/dl/papers/aqs.pdf
- https://javadoop.com/post/AbstractQueuedSynchronizer