Java源码中经常看到的CountDownLatch类详解

我们在Java源码里面经常看到CountDownLatch类的存在，它的作用是什么呢，今天我们来揭开它的真面目。
您请往下看

CountDownLatch类的作用是什么

CountDownLatch是JAVA提供在java.util.concurrent包下的一个辅助类，可以把它看成是一个计数器，其内部维护着一个count计数，只不过对这个计数器的操作都是原子操作，同时只能有一个线程去操作这个计数器，CountDownLatch通过构造函数传入一个初始计数值，调用者可以通过调用CounDownLatch对象的cutDown()方法，来使计数减1；如果某一个线程调用CountDownLatch的await()方法，那么调用者就会一直阻塞在这里，直到别人通过cutDown方法，将计数减到0，才可以继续执行

比如下面的代码

package com.one.util;

import java.util.concurrent.CountDownLatch;

public class Hello {
    /**
     * mCountDownLatch默认计数器是2
     */
    private final static CountDownLatch mCountDownLatch = new CountDownLatch(2);

    /**
     * 示例工作线程类
     */
    private static class WorkingThread extends Thread {
        private final String mThreadName;
        private final int mSleepTime;
        public WorkingThread(String name, int sleepTime) {
            mThreadName = name;
            mSleepTime = sleepTime;
        }

        @Override
        public void run() {
            System.out.println("[" + mThreadName + "] started!");
            try {
                Thread.sleep(mSleepTime);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            mCountDownLatch.countDown();
            System.out.println("[" + mThreadName + "] end!");
        }
    }

    /**
     * 示例线程类
     */
    private static class SampleThread extends Thread {

        @Override
        public void run() {
            System.out.println("[SampleThread] started!");
            try {
                // mCountDownLatch会导致SampleThread线程阻塞在这里等待，直到另外的WorkingThread调用countDown()把mCountDownLatch 里的count变为0，然后SampleThread才能够继续往下执行；
                mCountDownLatch.await();
            } catch (InterruptedException e) {

            }
            System.out.println("[SampleThread] end!");
        }
    }

    public static void main(String[] args) throws Exception {
        // 最先run SampleThread
        new SampleThread().start();
        // 运行两个工作线程
        new WorkingThread("WorkingThread1", 2000).start();
        new WorkingThread("WorkingThread2", 2000).start();
    }
}

从下面的结果可以验证上面的说法，此时可以看到SampleThread线程先启动了，然后执行[SampleThread] started!这句话，但是一直没有执行[SampleThread] end!这句话，直到WorkingThread1线程和WorkingThread2线程都调用countDown()把mCountDownLatch的值从2变成0之后，SampleThread线程才开始执行[SampleThread] end!这句话，这就是CountDownLatch类的作用

调用CountDownLatch类的await方法的线程会被阻塞掉

什么意思呢，比如下面的代码，因为在下面的 mCountDownLatch.await()位置调用了await方法，所以System.out.println(“执行不到”)这一句是执行不到的

package com.one.util;

import java.util.concurrent.CountDownLatch;

public class Hello {
    /**
     * mCountDownLatch默认计数器是2
     */
    private final static CountDownLatch mCountDownLatch = new CountDownLatch(2);

    public static void main(String[] args) throws InterruptedException {
        mCountDownLatch.await();
        System.out.println("执行不到");
    }
}

如下所示，此时可以看到线程还在运行，但是一直不能打印"执行不到"这句话