CountDownLatch入门实战

引言

在很多的面经中都看到过提问 CountDownLatch 的问题，正好我们最近也在梳理学习AQS（抽象队列同步器），而CountDownLatch又是其中典型的代表，我们今天就继续来学一下这个同步工具类！

CountDownLatch有何作用？

我们知道AQS是专属于构造锁和同步器的一个抽象工具类，基于它Java构造出了大量的常用同步工具，如ReentrantLock、Semaphore、ReentrantReadWriteLock、SynchronousQueue等等，我们今天的主角CountDownLatch同样如此。

CountDownLatch(倒时器)允许N个线程阻塞在同一个地方，直至所有线程的任务都执行完毕。CountDownLatch 有一个计数器，可以通过countDown()方法对计数器的数目进行减一操作，也可以通过await()方法来阻塞当前线程，直到计数器的值为 0。

CountDownLatch的底层原理

想要迅速了解一个Java类的内部构造，或者使用原理，最快速直接的办法就是看它的源码，这是很多初学者比较抵触的，会觉得很多封装起来的源码都晦涩难懂，诚然很多类内部实现是复杂，但我们作为Java工程师也不能只追求CRUD呀，培养自己看源码的习惯，硬着头皮看段时间，代码能力绝对会提升的！

废话说的有点多了，我们直接进入CountDownLatch内部去看看它的底层原理吧

【源码解析1】

//几乎所有基于AQS构造的同步类，内部都需要一个静态内部类去继承AQS
private static final class Sync extends AbstractQueuedSynchronizer {
     private static final long serialVersionUID = 4982264981922014374L;
     Sync(int count) {
         setState(count);
     }
     int getCount() {
         return getState();
     }
 }
private final Sync sync;
//构造方法中初始化count值
public CountDownLatch(int count) {
    if (count < 0) throw new IllegalArgumentException("count < 0");
    this.sync = new Sync(count);
}

几乎所有基于AQS构造的同步类，内部都需要一个静态内部类去继承AQS，并实现其提供的钩子方法，通过封装AQS中的state为count来确定多个线程的计时器。

countDown()方法

【源码解析2】

//核心方法，内部封装了共享模式下的线程释放
 public void countDown() {
 	//内部类Sync，继承了AQS
     sync.releaseShared(1);
 }
 //AQS内部的实现
 public final boolean releaseShared(int arg) {
  if (tryReleaseShared(arg)) {
  		//唤醒后继节点
         doReleaseShared();
         return true;
     }
     return false;
 }   

在CountDownLatch中通过countDown来减少倒计时数，这是最重要的一个方法，我们继续跟进源码看到它通过releaseShared()方法去释放锁，这个方法是AQS内部的默认实现方法，而在这个方法中再一次的调用了tryReleaseShared(arg)，这是一个AQS的钩子方法，方法内部仅有默认的异常处理，真正的实现由CountDownLatch内部类Sync完成

【源码解析3】 

// 对 state 进行递减，直到 state 变成 0；
// 只有 count 递减到 0 时，countDown 才会返回 true
protected boolean tryReleaseShared(int releases) {
    // 自选检查 state 是否为 0
    for (;;) {
        int c = getState();
        // 如果 state 已经是 0 了，直接返回 false
        if (c == 0)
            return false;
        // 对 state 进行递减
        int nextc = c-1;
        // CAS 操作更新 state 的值
        if (compareAndSetState(c, nextc))
            return nextc == 0;
    }
}

CountDownLatch的使用

由于await的实现步骤和countDown类似，我们就不贴源码了，大家自己跟进去也很容易看明白，我们现在直接来一个小demo感受一下如何使用CountDownLatch做一个倒时器

public class Test {
    public static void main(String[] args) throws InterruptedException {
        // 创建一个倒计数为 3 的 CountDownLatch
        CountDownLatch latch = new CountDownLatch(3);

        Thread service1 = new Thread(new Service("3", 1000, latch));
        Thread service2 = new Thread(new Service("2", 2000, latch));
        Thread service3 = new Thread(new Service("1", 3000, latch));

        service1.start();
        service2.start();
        service3.start();

        // 等待所有服务初始化完成
        latch.await();
        System.out.println("发射");
    }

    static class Service implements Runnable {
        private final String name;
        private final int timeToStart;
        private final CountDownLatch latch;

        public Service(String name, int timeToStart, CountDownLatch latch) {
            this.name = name;
            this.timeToStart = timeToStart;
            this.latch = latch;
        }

        @Override
        public void run() {
            try {
                Thread.sleep(timeToStart);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println(name);
            // 减少倒计数
            latch.countDown();
        }
    }
}