CountDownLatch是一个常用的共享锁，其功能相当于一个多线程环境下的倒数门闩。CountDownLatch可以指定一个计数值，在并发环境下由线程进行减一操作，当计数值变为0之后，被await方法阻塞的线程将会唤醒。通过CountDownLatch可以实现线程间的计数同步。
为什么说CountDownLatch是一种共享锁？因为CountDownLatch提供了一种“计数器”机制，允许N个线程在CountDownLatch的协调下同时运行，计数器归零才会触发阻塞的线程继续执行后续代码。
可能还是有人无法明白“共享”两个字，举个例子：如果计数器初始值是10，那开始的时候会有十个线程自动获取到共享锁，由于计数器的值是10，所以业务上如果有更多的线程想要获取锁，就要等待（主线程await）；十个线程调用countDown方法之后锁才会释放，这时候阻塞的主线程才能获取到共享锁。
一、基本使用
CountDownLatch的使用步骤如下所示：
（1）创建倒数闩，初始化CountDownLatch时设置倒数的总次数，比如为100。
（2）等待线程调用倒数闩的await()方法阻塞自己，等待倒数闩的计数器数值为0（倒数线程全部执行结束）。
（3）倒数线程执行完，调用CountDownLatch.countDown()方法将计数器数值减一。
现在我们定义10个线程，十个线程都运行任务结束后，打印“Hello，World”。
import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; import java.util.concurrent.CountDownLatch; import java.util.concurrent.TimeUnit; /** * @author kdyzm * @date 2024/12/28 */ @Slf4j public class CountdownLatchDemo { private static final CountDownLatch countDownLatch = new CountDownLatch(10); public static void main(String[] args) throws InterruptedException { for (int i = 0; i < 10; i++) { new Thread(new Task(), "" + i).start(); } countDownLatch.await(); log.info("Hello,World"); } static class Task implements Runnable { @SneakyThrows @Override public void run() { log.info("我是线程【{}】，即将完成任务", Thread.currentThread().getName()); TimeUnit.SECONDS.sleep(1); countDownLatch.countDown(); } } } 代码运行结果如下：
CountDownLatch很神奇，竟然能同步多个线程的执行，让多个线程都执行完才进行下一步行动，那么它实现的原理是什么呢？
搂一眼CountDownLatch类：
这代码可太熟悉了，和ReentrantLock非常相似啊。。。内部也是定义了Sync类继承了AQS类，大体上可以明白了，CountDownLatch类也是基于AQS类实现的。
CountDownLatch有两个重要的方法：await方法和countDown方法。
二、CountDownLatch实例的创建
CountDownLatch构造方法需要传递一个整数
public CountDownLatch(int count) { if (count < 0) throw new IllegalArgumentException("count < 0"); this.sync = new Sync(count); } 在构造方法中初始化了Sync类的实例。
private static final class Sync extends AbstractQueuedSynchronizer { private static final long serialVersionUID = 4982264981922014374L; Sync(int count) { //使用传递的count参数初始化state数值 setState(count); } ...... } 可以看到，在Sync类的构造方法中，会使用传入的count参数初始化state参数。在ReetrantLock类中，state参数代表着重入锁的重入次数，在CountDownLatch中将其设置成count数值有什么意义呢？
实际上在CountDownLatch中state参数表示可以执行countDown方法的次数。
二、await方法
一般而言，await方法是先开始执行的方法，然后迅速进入阻塞状态，等待其它线程计数到0，就会触发唤醒，然后主线程执行await方法之后的代码。
public void await() throws InterruptedException { sync.acquireSharedInterruptibly(1); } await方法很简单，它的逻辑都委托给Sync类实例来实现了，但是实际上这里的acquireSharedInterruptibly方法调用是AQS的模板方法。
1、可中断获取共享锁：acquireSharedInterruptibly
public final void acquireSharedInterruptibly(int arg) throws InterruptedException { if (Thread.interrupted()) throw new InterruptedException(); //查看计数器是否归零 if (tryAcquireShared(arg) < 0) //如果没归零，就开始执行可中断获取共享锁流程 doAcquireSharedInterruptibly(arg); } 这个方法非常简单，这个方法是一个可中断的方法，可中断意味着它可以抛出InterruptedException异常。所以上来它先检查了当前线程是否发生了中断，如果发生了中断，就立刻抛出InterruptedException异常。
接下来调用了tryAcquireShared方法查看计数器是否归零
2、查看计数器是否归零：tryAcquireShared
tryAcquireShared方法是AQS的钩子方法，被CountDownLatch实现，用于检查计数器是否变成了0。
protected int tryAcquireShared(int acquires) { return (getState() == 0) ? 1 : -1; } 可以看到，传过来的acquires参数并没有被用到，实际上该方法就是检查了state参数是否变成了0。
这里其实有个疑问，根据ReentrantLock的经验，tryAcquireShared是尝试获取锁的方法，但是它仅仅只是判断了state的值是否为0，这是为何？
3、执行获取共享锁：doAcquireSharedInterruptibly
private void doAcquireSharedInterruptibly(int arg) throws InterruptedException { //将节点标记为共享类型，并加入AQS同步队列。 final Node node = addWaiter(Node.SHARED); boolean failed = true; try { for (;;) { //获取节点的前置节点 final Node p = node.predecessor(); //判断是否是头结点，是头结点就表示自己可以尝试获取锁了。 if (p == head) { //尝试获取共享锁，实际上是检查了state参数是否为0 int r = tryAcquireShared(arg); //r>=0的条件只有一个，那就是r为1，表示state值已经变成了0 if (r >= 0) { //执行获取锁成功的流程。 setHeadAndPropagate(node, r); p.next = null; // help GC failed = false; return; } } //检查是否应该在失败后挂起。 if (shouldParkAfterFailedAcquire(p, node) && //挂起线程 parkAndCheckInterrupt()) //如果发生了中断，就抛出IE异常。 throw new InterruptedException(); } } finally { if (failed) cancelAcquire(node); } } 在这里，AQS队列的规则和ReentrantLock中AQS队列一样，头部节点表示已经获取锁的节点，一旦被唤醒，头部节点的后续节点就可以抢锁了。
CountDownLatch的抢锁逻辑比较简单，只是检查下state参数是否已经变成了0，变成了0，就表示其余线程都执行完了countDown方法，计数器已经变成了0，当前线程需要释放了；没变成0，就表示不是所有线程都执行了countDown方法，还不满足获取锁的条件。
不满足获取锁的条件的话，就开始判定是否需要挂起线程，执行shouldParkAfterFailedAcquire方法，该方法会被执行两次，然后执行parkAndCheckInterrupt方法，将当前主线程挂起。
主线程挂起之后，就开始等待其余线程执行执行countDown方法之后唤醒它。
4、获取锁成功之后：setHeadAndPropagate
执行到setHeadAndPropagate方法，就表示当前线程已经获取到了锁，执行到该方法的时候，计数器必定已经变成了0，parkAndCheckInterrupt()方法原本是被阻塞的现在已经被释放。
private void setHeadAndPropagate(Node node, int propagate) { Node h = head; setHead(node); //propagate值为1，满足if条件 if (propagate > 0 || h == null || h.waitStatus < 0 || (h = head) == null || h.waitStatus < 0) { Node s = node.next; //AQS队列中只有node一个等待节点，所以next必定为null，满足if条件 if (s == null || s.isShared()) //后续没有等待节点了，执行锁释放流程 doReleaseShared(); } } 该方法中有两处if，有很多判断根本用不到，大概是给其它共享锁使用的。总之CountDownLatch的await方法执行到这里之后会执行doReleaseShared方法执行锁释放，因为后续已经没有节点了。
5、锁释放：doReleaseShared
await方法中刚获取到锁就开始执行锁释放了，这是因为AQS中等待队列中就一个节点，该节点后续没有了等待节点，就需要释放锁。
rivate void doReleaseShared() { for (;;) { Node h = head; //此时AQS队列就一个head节点，head == tail，不满足if条件 if (h != null && h != tail) { int ws = h.waitStatus; if (ws == Node.SIGNAL) { if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0)) continue; unparkSuccessor(h); } else if (ws == 0 && !compareAndSetWaitStatus(h, 0, Node.PROPAGATE)) continue; } //满足if条件，结束循环 if (h == head) break; } } CountDownLatch的await方法中执行到doReleaseShared的时候是有附件条件的：AQS队列中只有一个头结点，没有等待节点了，所以第一个if条件没有满足条件，第二个if条件满足了，所以结束了循环。
doReleaseShared方法什么都没做。
这样await方法结束后，主线程继续执行await方法后的代码。
三、countDown方法
通常情况下，await方法会先执行，并在doAcquireSharedInterruptibly方法中的parkAndCheckInterrupt处阻塞，countDown方法在其它线程中被调用，并在计数器归零时唤醒被阻塞在await方法的线程。
public void countDown() { sync.releaseShared(1); } countDown方法依然调用了sync的方法：releaseShared方法是AQS的模板方法。
1、释放共享锁：releaseShared
public final boolean releaseShared(int arg) { //尝试释放锁，计数器减一 if (tryReleaseShared(arg)) { //释放共享锁 doReleaseShared(); return true; } return false; } releaseShared是AQS的模板方法，该方法会调用钩子方法tryReleaseShared方法执行计数器减一，如果满足了释放共享锁的条件，就执行doRelease方法释放共享锁。
比如计数器初始值是10，则前九个线程调用countDown方法都不会满足if条件，第十个线程调用countDown方法时，调用tryReleaseShared方法之后会满足if条件，然后执行doReleaseShared方法。
2、计数器减一：tryReleaseShared
tryReleaseShared是AQS的钩子方法，在CountDownLatch中，被Sync实现了该方法。
protected boolean tryReleaseShared(int releases) { //循环防止之后的CAS失败 for (;;) { int c = getState(); //如果计数器的值已经是0，就返回失败，这里是为了防止出现超出计数器初始值的线程数调用countDown方法 if (c == 0) return false; //计数器减一 int nextc = c-1; //CAS设置state值 if (compareAndSetState(c, nextc)) /** * 如果设置成功了，再次判断计数器是否为0： * 如果是0表示当前操作使得计数器归零，应当触发释放共享锁的操作 * 如果不是0表示计数器还未归零，不到释放共享锁的时机 */ return nextc == 0; } } 由于多个线程可能会同时执行countDown方法，可能会并行修改state值，所以该方法使用了for循环+CAS的方式保证了线程安全性。
需要注意的唯一一点就是
if (c == 0) return false; 这个判断语句，该判断语句用于防止出现超出计数器初始值的线程数调用countDown方法，当超出计数器初始值的线程调用countDown方法，会因为计数器已经归零而导致tryReleaseShared方法返回false。
3、释放锁：doReleaseShared
在众多执行countDown方法的线程中，只有将计数器置为零的那个线程有执行doReleaseShared方法的机会。
doReleaseShared方法在之前await方法中已经讲过一部分，在await方法中如果当前节点是最后一个节点，那最后将调用doRelease方法释放锁。为什么这里会释放锁？因为执行countDown方法的线程都是获取到锁的线程，执行完毕之后要调用countDown方法释放锁，当然还是那句话，只有将计数器置为零的那个线程有执行doReleaseShared方法的机会。
和await方法中调用doReleaseShared方法什么都没做不一样，在countDown方法中调用doReleaseShared执行逻辑就变得不一样了。一般来说，由于await方法会先于countDown方法执行，所以在执行countDown方法的时候，AQS队列中会有两个节点，一个是头结点（AQS初始化创建的虚拟节点），一个是等待队列中的主线程节点。
private void doReleaseShared() { for (;;) { //获取头结点，并暂存到h变量 Node h = head; //当前AQS队列中有两个节点，满足if条件 if (h != null && h != tail) { //获取到头结点状态 int ws = h.waitStatus; //由于await方法已经将头结点状态更改成了SIGNAL,所以这里会满足if条件 if (ws == Node.SIGNAL) { //将头结点状态改为0 if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0)) continue; //唤醒后继结点 unparkSuccessor(h); } else if (ws == 0 && !compareAndSetWaitStatus(h, 0, Node.PROPAGATE)) continue; // loop on failed CAS } //一般情况下，执行countDown方法的线程执行速度比较快，会满足该if条件并结束循环 if (h == head) break; } } 可以看到将计数器归零的countDown方法调用的线程，最后会在doReleaseShared方法中调用unparkSuccessor方法唤醒后继线程。后继线程实际上就是await方法被阻塞的主线程，它将会在doAcquireSharedInterruptibly方法中的parkAndCheckInterrupt方法处被唤醒后继续执行后续代码。
四、扩展问题
1、countDown方法先于await方法执行
一般情况下await方法会先执行，执行后线程被挂起阻塞；countDown方法后执行，最后将计数器归零的线程负责将调用await方法被挂起的线程唤醒。想一个问题：如果cuontDown方法先都执行完了，最后再执行await方法，会发生什么呢？
答案是：什么都不会发生，因为这就是正确的逻辑啊。await方法和countDown方法均为此做了防范措施。
countDown方法：
最后将计数器清零的线程会将被阻塞的线程唤醒，但是如果说await方法没有被调用，则AQS队列中就没有需要被唤醒的线程，doReleaseShared方法会如何做呢？
private void doReleaseShared() { for (;;) { //由于AQS队列为空，所以head为null Node h = head; //head为null，不满足if条件 if (h != null && h != tail) { int ws = h.waitStatus; if (ws == Node.SIGNAL) { if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0)) continue; unparkSuccessor(h); } else if (ws == 0 && !compareAndSetWaitStatus(h, 0, Node.PROPAGATE)) continue; } //h == head == null，满足if条件，结束循环 if (h == head) break; } } doReleaseShared方法此时什么都没做，由于没有满足触发条件，因此都没有机会执行unparkSuccessor(h);唤醒后继节点。
await方法：
await方法中调用了acquireShardInterruptibly方法，该方法在执行的过程中会先判定当前计数器的值，如果是0，就不会再继续执行doAcquireSharedInterruptibly方法，避免了线程被挂起无人唤醒的尴尬局面。
public final void acquireSharedInterruptibly(int arg) throws InterruptedException { if (Thread.interrupted()) throw new InterruptedException(); //由于计数器的值已经归零，所以这里没有满足if条件 if (tryAcquireShared(arg) < 0) doAcquireSharedInterruptibly(arg); } 2、超出计数器初始值的线程数执行countDown方法
如下代码所示
@Slf4j public class CountdownLatchDemo { //计数器初始值是1 private static final CountDownLatch countDownLatch = new CountDownLatch(1); public static void main(String[] args) throws InterruptedException { //10个线程执行了countDown方法 for (int i = 0; i < 10; i++) { new Thread(new Task(), "" + i).start(); } countDownLatch.await(); log.info("Hello,World"); } static class Task implements Runnable { @SneakyThrows @Override public void run() { log.info("我是线程【{}】，即将完成任务", Thread.currentThread().getName()); //等待一秒钟模拟业务执行，确保await方法先执行 TimeUnit.SECONDS.sleep(1); countDownLatch.countDown(); } } } CountDownLatch对多余执行countDown方法的线程的处理策略就是：忽略它。看看countDown方法中调用的releaseShared方法
//该方法为AQS的模板方法 public final boolean releaseShared(int arg) { //tryReleaseShared方法是关键 if (tryReleaseShared(arg)) { doReleaseShared(); return true; } return false; } //该方法文为CountDownLatch中实现AQS的钩子方法 protected boolean tryReleaseShared(int releases) { for (;;) { int c = getState(); //如果计数器值已经变成了0，就直接返回false，防止超量的线程执行countDown方法。 if (c == 0) return false; int nextc = c-1; if (compareAndSetState(c, nextc)) return nextc == 0; } } releaseShared方法仅仅是通过判断一次
if (c == 0) return false; 就解决了超量线程调用countDown方法的问题。
3、CountDownLatch使用进阶
本篇文章通篇均是以以下代码示例分析的：
import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; import java.util.concurrent.CountDownLatch; import java.util.concurrent.TimeUnit; /** * @author kdyzm * @date 2024/12/28 */ @Slf4j public class CountdownLatchDemo { private static final CountDownLatch countDownLatch = new CountDownLatch(10); public static void main(String[] args) throws InterruptedException { for (int i = 0; i < 10; i++) { new Thread(new Task(), "" + i).start(); } countDownLatch.await(); log.info("Hello,World"); } static class Task implements Runnable { @SneakyThrows @Override public void run() { log.info("我是线程【{}】，即将完成任务", Thread.currentThread().getName()); TimeUnit.SECONDS.sleep(1); countDownLatch.countDown(); } } } 这种CountDownLatch的使用方式比较简单，实际上CountDownLatch还有另外一种经典的使用方式，其源码示例如下所示
import java.util.concurrent.CountDownLatch; class Driver { // ... void main() throws InterruptedException { CountDownLatch startSignal = new CountDownLatch(1); CountDownLatch doneSignal = new CountDownLatch(N); for (int i = 0; i < N; ++i) // create and start threads new Thread(new Worker(startSignal, doneSignal)).start();//线程刚开始执行全部进入阻塞状态 doSomethingElse(); // 先执行一些动作 startSignal.countDown(); // 让所有阻塞的线程开始执行 doSomethingElse(); doneSignal.await(); // 等待所有线程执行完毕 } } class Worker implements Runnable { private final CountDownLatch startSignal; private final CountDownLatch doneSignal; Worker(CountDownLatch startSignal, CountDownLatch doneSignal) { this.startSignal = startSignal; this.doneSignal = doneSignal; } public void run() { try { startSignal.await(); doWork(); doneSignal.countDown(); } catch (InterruptedException ex) { } // return; } void doWork() { ...} } 这种模式巧妙的使用了双CountDownLatch实例精准控制线程的开始运行时间，并在开始执行前、执行中间、执行之后插入一些自定义的方法执行。实际上使用了CountDownLatch的两种使用模式：
模式一：N次countDown方法调用，一次await方法调用 ：这种模式用于等待所有线程执行完毕之后再执行await方法之后的代码
模式二：N次await方法调用，一次countDown方法调用 ：这种模式用于需要在所有线程开始执行前插入一些自定义操作。
本篇文章一直围绕模式一分析，模式二一直没有提及，因为模式二用的实在太少，在实际开发中没见到过。。。
现在将模式二的核心代码提炼出来，看看到底它做了什么事情：
import lombok.extern.slf4j.Slf4j; import java.util.concurrent.CountDownLatch; @Slf4j public class Driver { public static void main(String[] args) throws InterruptedException { CountDownLatch startSignal = new CountDownLatch(1); for (int i = 0; i < 10; ++i) { new Thread(new Worker(startSignal)).start(); } log.info("在所有线程执行前先执行一些任务"); startSignal.countDown(); } } @Slf4j class Worker implements Runnable { private final CountDownLatch startSignal; Worker(CountDownLatch startSignal) { this.startSignal = startSignal; } @Override public void run() { try { startSignal.await(); doWork(); } catch (InterruptedException ex) { log.error("", ex); } } void doWork() { log.info("正在执行任务。。。"); } } 我们分析下这个过程，由于10个线程运行时上来就调用了await方法，所以它们会在AQS队列中排队，其数据结构如下所示
此时AQS队列中有11个节点，第一个头部节点是队列初始化的时候创建的虚拟节点，剩余十个节点，前9个节点状态都是SIGNAL（-1），因为它们在抢锁失败后会调用shouldParkAfterFailedAcquire方法，该方法会将当前节点的前置节点修改成SIGNAL（-1），最后一个节点入队后默认状态是0，由于没有后续节点入队了，所以其状态没有变成-1，而是0。
接下来主线程调用了countDown方法，由于计数器的初始值是1，所以CountDownLatch方法一次调用就会触发唤醒后继结点。接着，head节点后的节点将会在parkAndCheckInterrupt方法处继续执行后续代码：
private void doAcquireSharedInterruptibly(int arg) throws InterruptedException { //将节点标记为共享类型，并加入AQS同步队列。 final Node node = addWaiter(Node.SHARED); boolean failed = true; try { for (;;) { //获取节点的前置节点 final Node p = node.predecessor(); //判断是否是头结点，是头结点就表示自己可以尝试获取锁了。 if (p == head) { //尝试获取共享锁，实际上是检查了state参数是否为0 int r = tryAcquireShared(arg); //r>=0的条件只有一个，那就是r为1，表示state值已经变成了0 if (r >= 0) { //执行获取锁成功的流程。 setHeadAndPropagate(node, r); p.next = null; // help GC failed = false; return; } } //检查是否应该在失败后挂起。 if (shouldParkAfterFailedAcquire(p, node) && //挂起线程 parkAndCheckInterrupt()) //如果发生了中断，就抛出IE异常。 throw new InterruptedException(); } } finally { if (failed) cancelAcquire(node); } } 在该方法中，会执行setHeadAndPropagate方法：
private void setHeadAndPropagate(Node node, int propagate) { Node h = head; setHead(node); //propagate值为1，满足if条件 if (propagate > 0 || h == null || h.waitStatus < 0 || (h = head) == null || h.waitStatus < 0) { Node s = node.next; //此时AQS队列中有很多节点，节点类型都是SHARED类型,所以依然满足if条件 if (s == null || s.isShared()) //执行锁释放，并唤醒后续节点 doReleaseShared(); } } 接着执行doReleaseShared方法：在N次countDown，一次await模式中，该方法在await方法中的调用什么都没做就结束了循环；在N次await调用，一次countDown调用的模式中，await方法调用将唤醒后继节点
private void doReleaseShared() { for (;;) { //head节点已经变成当前执行线程节点 Node h = head; //满足if条件 if (h != null && h != tail) { //当前节点状态是SIGNAL int ws = h.waitStatus; //满足if条件 if (ws == Node.SIGNAL) { //唤醒后继结点前，先将头部节点状态更改为初始状态0 if (!compareAndSetWaitStatus(h, Node.SIGNAL, 0)) continue; // loop to recheck cases //唤醒后继结点 unparkSuccessor(h); } else if (ws == 0 && !compareAndSetWaitStatus(h, 0, Node.PROPAGATE)) continue; // loop on failed CAS } if (h == head) // loop if head changed break; } } 可以看到，AQS队列中的节点一个一个唤醒后继结点，然后去做自己的事情，直到最后一个节点；最后一个节点由于不满足h != tail的条件，会什么都不做就结束循环。
最后考虑一个问题，如果countDown方法先被执行了，那await方法执行的时候会怎样？
答案是什么事情都不会发生，因为await方法执行前需要先判断计数器的值，如果等于0，就不会有后续的获取锁、释放锁流程了。
public final void acquireSharedInterruptibly(int arg) throws InterruptedException { if (Thread.interrupted()) throw new InterruptedException(); //countDown计数器为0的时候不满足该if条件。 if (tryAcquireShared(arg) < 0) doAcquireSharedInterruptibly(arg); } 总结一下，ReentrantLock中lock方法和unlock方法中间是临界区代码，临界区代码是抢锁线程依次轮流进入执行的，所以ReetrantLock是“独占锁”；而在CountDownLatchd的“N次await方法调用，一次countDown方法调用”的模式中，中，所有线程排队进入AQS队列之后，一旦被唤醒，被唤醒的节点还会自动唤醒后边节点，然后各自做各自的事情，所有线程都是并行运行的，所以CountDownLatch是“共享锁”。
最后的最后，欢迎关注我的个人博客呀：
END.
