CountDownLatch是一种同步辅助工具,允许一个或多个线程等待在其他线程中执行的一组操作完成,可以利用它来实现并发执行某些操作。
CountDownLatch 维护了一个计数器,所有线程调用countDown()方法后计数器减1,主线程等到直到计数器为0,可以用作发令抢 一个线程等待 所有线程将计数器减少为0,模拟并发的场景。可以一个或者多个线程,等待计数器到0。CountDownLatch的实现很简单,基于AQS实现的共享锁。
AQS实现:
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;
}
}
}
countDown方法:
public void countDown() {
sync.releaseShared(1);
}
countDown方法不断释放共享锁,state减1,然后唤醒下一个排队加锁的线程,state所有线程共享,没有像ReentrantReadWriteLock这样用ThreadLocal隔离不同线程的锁次数。
await方法:
public boolean await(long timeout, TimeUnit unit)
throws InterruptedException {
return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
}
await方法,判断state是否达到了0,如果没有到0则会进入排队和阻塞,实现等
待其他线程操作的目的,直到state=0。
Rocketmq扩展:
在Rocketmq的实现中,就对CountDownLatch进行了扩展,添加了重置
功能,使得CountDownLatch也可以像CyclicBarrier一样循环使用。
他将初始化时的state值进行了暂存到startCount变量上,在reset方法的
时候重新将startCount值设置到state上。
发现CountDownLatch的实现很简单,其实是因为AQS的抽象类,AQS已经提
供了很好的同步抽象,使得实现其他辅助的同步工具类非常简单,因此学习好
AQS对于学习并发工具还是有很有必要的。
2022坚持学习,继续加油。