国内chatgpt入口: https://chatgptmirror.com?share=AHO1O5
CountDownLatch相当于一个计数锁存器,当一定的操作结束后,某个线程才能执行。每执行一个操作计数减1,计数到0时,才执行某个操作。
实现原理:
CountDownLatch实现了一个继承AQS的共享锁Sync,里面重写的tryAcquireShared()方法,只有state为0的时候锁才可以被获取。在构造对象的时候,初始化一个计数值N,把N赋给sync的state变量,每执行一个操作,N就会减1,N减到0时,此时state也为0,这时锁可以被获取,可以执行特定的操作。
执行countDown()方法的线程让N减1,执行await()方法的线程等待,直到其他线程让N减到0。
主要方法
//只有一个构造方法
public CountDownLatch(int count) {
if (count < 0) throw new IllegalArgumentException("count < 0");
this.sync = new Sync(count);
}
//执行这个方法的线程等待,直到计数值为0
public void await() throws InterruptedException {
sync.acquireSharedInterruptibly(1);
}
//执行这个方法的线程让计数值减1
public void countDown() {
sync.releaseShared(1);
}
//如果state值为0,返回1,正值代表获取锁成功,负值代表获取锁失败
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;
}
}
}