大白话说java并发工具类-CountDownLatch，CyclicBarrier

原创文章&经验总结&从校招到 A 厂一路阳光一路沧桑

详情请戳www.codercc.com

1. 倒计时器 CountDownLatch

在多线程协做完成业务功能时，有时候须要等待其余多个线程完成任务以后，主线程才能继续往下执行业务功能，在这种的业务场景下，一般可使用 Thread 类的 join 方法，让主线程等待被 join 的线程执行完以后，主线程才能继续往下执行。固然，使用线程间消息通讯机制也能够完成。其实，java 并发工具类中为咱们提供了相似“倒计时”这样的工具类，能够十分方便的完成所说的这种业务场景。

为了可以理解 CountDownLatch，举一个很通俗的例子，运动员进行跑步比赛时，假设有 6 个运动员参与比赛，裁判员在终点会为这 6 个运动员分别计时，能够想象没当一个运动员到达终点的时候，对于裁判员来讲就少了一个计时任务。直到全部运动员都到达终点了，裁判员的任务也才完成。这 6 个运动员能够类比成 6 个线程，当线程调用 CountDownLatch.countDown 方法时就会对计数器的值减一，直到计数器的值为 0 的时候，裁判员（调用 await 方法的线程）才能继续往下执行。

下面来看些 CountDownLatch 的一些重要方法。

先从 CountDownLatch 的构造方法看起：

public CountDownLatch(int count)
复制代码

构造方法会传入一个整型数 N，以后调用 CountDownLatch 的countDown方法会对 N 减一，知道 N 减到 0 的时候，当前调用await方法的线程继续执行。

CountDownLatch 的方法不是不少，将它们一个个列举出来：

1. await() throws InterruptedException：调用该方法的线程等到构造方法传入的 N 减到 0 的时候，才能继续往下执行；
2. await(long timeout, TimeUnit unit)：与上面的 await 方法功能一致，只不过这里有了时间限制，调用该方法的线程等到指定的 timeout 时间后，无论 N 是否减至为 0，都会继续往下执行；
3. countDown()：使 CountDownLatch 初始值 N 减 1；
4. long getCount()：获取当前 CountDownLatch 维护的值；

下面用一个具体的例子来讲明 CountDownLatch 的具体用法:

public class CountDownLatchDemo {
private static CountDownLatch startSignal = new CountDownLatch(1);
//用来表示裁判员须要维护的是6个运动员
private static CountDownLatch endSignal = new CountDownLatch(6);
public static void main(String[] args) throws InterruptedException { ExecutorService executorService = Executors.newFixedThreadPool(6); for (int i = 0; i < 6; i++) { executorService.execute(() -> { try { System.out.println(Thread.currentThread().getName() + " 运动员等待裁判员响哨！！！"); startSignal.await(); System.out.println(Thread.currentThread().getName() + "正在全力冲刺"); endSignal.countDown(); System.out.println(Thread.currentThread().getName() + " 到达终点"); } catch (InterruptedException e) { e.printStackTrace(); } }); } System.out.println("裁判员发号施令啦！！！"); startSignal.countDown(); endSignal.await(); System.out.println("全部运动员到达终点，比赛结束！"); executorService.shutdown(); } } 输出结果：
复制代码pool-1-thread-2 运动员等待裁判员响哨！！！ pool-1-thread-3 运动员等待裁判员响哨！！！ pool-1-thread-1 运动员等待裁判员响哨！！！ pool-1-thread-4 运动员等待裁判员响哨！！！ pool-1-thread-5 运动员等待裁判员响哨！！！ pool-1-thread-6 运动员等待裁判员响哨！！！ 裁判员发号施令啦！！！ pool-1-thread-2正在全力冲刺 pool-1-thread-2 到达终点 pool-1-thread-3正在全力冲刺 pool-1-thread-3 到达终点 pool-1-thread-1正在全力冲刺 pool-1-thread-1 到达终点 pool-1-thread-4正在全力冲刺 pool-1-thread-4 到达终点 pool-1-thread-5正在全力冲刺 pool-1-thread-5 到达终点 pool-1-thread-6正在全力冲刺 pool-1-thread-6 到达终点 全部运动员到达终点，比赛结束！

该示例代码中设置了两个 CountDownLatch，第一个endSignal用于控制让 main 线程（裁判员）必须等到其余线程（运动员）让 CountDownLatch 维护的数值 N 减到 0 为止。另外一个startSignal用于让 main 线程对其余线程进行“发号施令”，startSignal 引用的 CountDownLatch 初始值为 1，而其余线程执行的 run 方法中都会先经过 startSignal.await()让这些线程都被阻塞，直到 main 线程经过调用startSignal.countDown();，将值 N 减 1，CountDownLatch 维护的数值 N 为 0 后，其余线程才能往下执行，而且，每一个线程执行的 run 方法中都会经过endSignal.countDown();对endSignal维护的数值进行减一，因为往线程池提交了 6 个任务，会被减 6 次，因此endSignal维护的值最终会变为 0，所以 main 线程在latch.await();阻塞结束，才能继续往下执行。

另外，须要注意的是，当调用 CountDownLatch 的 countDown 方法时，当前线程是不会被阻塞，会继续往下执行，好比在该例中会继续输出pool-1-thread-4 到达终点。

2. 循环栅栏：CyclicBarrier

CyclicBarrier 也是一种多线程并发控制的实用工具，和 CountDownLatch 同样具备等待计数的功能，可是相比于 CountDownLatch 功能更增强大。

为了理解 CyclicBarrier，这里举一个通俗的例子。开运动会时，会有跑步这一项运动，咱们来模拟下运动员入场时的状况，假设有 6 条跑道，在比赛开始时，就须要 6 个运动员在比赛开始的时候都站在起点了，裁判员吹哨后才能开始跑步。跑道起点就至关于“barrier”，是临界点，而这 6 个运动员就类比成线程的话，就是这 6 个线程都必须到达指定点了，意味着凑齐了一波，而后才能继续执行，不然每一个线程都得阻塞等待，直至凑齐一波便可。cyclic 是循环的意思，也就是说 CyclicBarrier 当多个线程凑齐了一波以后，仍然有效，能够继续凑齐下一波。CyclicBarrier 的执行示意图以下：

CyclicBarrier执行示意图.jpg

当多个线程都达到了指定点后，才能继续往下继续执行。这就有点像报数的感受，假设 6 个线程就至关于 6 个运动员，到赛道起点时会报数进行统计，若是恰好是 6 的话，这一波就凑齐了，才能往下执行。**CyclicBarrier 在使用一次后，下面依然有效，能够继续当作计数器使用，这是与 CountDownLatch 的区别之一。**这里的 6 个线程，也就是计数器的初始值 6，是经过 CyclicBarrier 的构造方法传入的。

下面来看下 CyclicBarrier 的主要方法：

//等到全部的线程都到达指定的临界点
await() throws InterruptedException, BrokenBarrierException
//与上面的await方法功能基本一致，只不过这里有超时限制，阻塞等待直至到达超时时间为止 await(long timeout, TimeUnit unit) throws InterruptedException, BrokenBarrierException, TimeoutException
//获取当前有多少个线程阻塞等待在临界点上 int getNumberWaiting()
//用于查询阻塞等待的线程是否被中断 boolean isBroken()
复制代码//将屏障重置为初始状态。若是当前有线程正在临界点等待的话，将抛出BrokenBarrierException。 void reset()

另外须要注意的是，CyclicBarrier 提供了这样的构造方法：

public CyclicBarrier(int parties, Runnable barrierAction)
复制代码

能够用来，当指定的线程都到达了指定的临界点的时，接下来执行的操做能够由 barrierAction 传入便可。

一个例子

下面用一个简单的例子，来看下 CyclicBarrier 的用法，咱们来模拟下上面的运动员的例子。

public class CyclicBarrierDemo {
    //指定必须有6个运动员到达才行
    private static CyclicBarrier barrier = new CyclicBarrier(6, () -> {
        System.out.println("全部运动员入场，裁判员一声令下！！！！！");
    });
    public static void main(String[] args) {
        System.out.println("运动员准备进场，全场欢呼............");
    ExecutorService service = Executors.newFixedThreadPool(6);
    for (int i = 0; i < 6; i++) {
        service.execute(() -> {
            try {
                System.out.println(Thread.currentThread().getName() + " 运动员，进场");
                barrier.await();
                System.out.println(Thread.currentThread().getName() + "  运动员出发");
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
        });
    }
}
复制代码
}
复制代码ExecutorService service = Executors.newFixedThreadPool(6);
    for (int i = 0; i < 6; i++) {
        service.execute(() -> {
            try {
                System.out.println(Thread.currentThread().getName() + " 运动员，进场");
                barrier.await();
                System.out.println(Thread.currentThread().getName() + "  运动员出发");
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (BrokenBarrierException e) {
                e.printStackTrace();
            }
        });
    }
}
复制代码输出结果： 运动员准备进场，全场欢呼............ pool-1-thread-2 运动员，进场 pool-1-thread-1 运动员，进场 pool-1-thread-3 运动员，进场 pool-1-thread-4 运动员，进场 pool-1-thread-5 运动员，进场 pool-1-thread-6 运动员，进场 全部运动员入场，裁判员一声令下！！！！！ pool-1-thread-6 运动员出发 pool-1-thread-1 运动员出发 pool-1-thread-5 运动员出发 pool-1-thread-4 运动员出发 pool-1-thread-3 运动员出发 pool-1-thread-2 运动员出发

从输出结果能够看出，当 6 个运动员（线程）都到达了指定的临界点（barrier）时候，才能继续往下执行，不然，则会阻塞等待在调用await()处

3. CountDownLatch 与 CyclicBarrier 的比较

CountDownLatch 与 CyclicBarrier 都是用于控制并发的工具类，均可以理解成维护的就是一个计数器，可是这二者仍是各有不一样侧重点的：

1. CountDownLatch 通常用于某个线程 A 等待若干个其余线程执行完任务以后，它才执行；而 CyclicBarrier 通常用于一组线程互相等待至某个状态，而后这一组线程再同时执行；CountDownLatch 强调一个线程等多个线程完成某件事情。CyclicBarrier 是多个线程互等，等你们都完成，再携手共进。
2. 调用 CountDownLatch 的 countDown 方法后，当前线程并不会阻塞，会继续往下执行；而调用 CyclicBarrier 的 await 方法，会阻塞当前线程，直到 CyclicBarrier 指定的线程所有都到达了指定点的时候，才能继续往下执行；
3. CountDownLatch 方法比较少，操做比较简单，而 CyclicBarrier 提供的方法更多，好比可以经过 getNumberWaiting()，isBroken()这些方法获取当前多个线程的状态，而且 CyclicBarrier 的构造方法能够传入 barrierAction，指定当全部线程都到达时执行的业务功能；
4. CountDownLatch 是不能复用的，而 CyclicBarrier 是能够复用的。