【搞定Java并发编程】第24篇：Java中的并发工具类之CountDownLatch

上一篇：Java中的阻塞队列 BlockingQueue 详解

本文目录：

1、CountDownLatch的基本概述

2、CountDownLatch的使用案例

3、CountDownLatch的源码分析

---

1、CountDownLatch的基本概述

CountDownLatch允许一个或多个线程等待其他线程完成操作。

CountDownLatch又称为“闭锁”，它是一个同步辅助类，在完成一组正在其他线程中执行的操作之前，它允许一个或多个线程一直等待，直到等待的线程操作完成或者等待超时，自己再开始执行。

CountDownLatch可以延迟线程的进度直到其他线程到达终止状态，它可以用来确保某些任务在其他任务都完成后再继续进行：

• 确保某个计算在其需要的所有资源都被初始化之后再继续执行；
• 确保某个服务在其依赖的所有其他服务都已经启动之后才启动；
• 等待直到某个操作所有参与者都准备就绪后再继续执行。

另外它还提供了一个countDown方法来操作计数器的值，每调用一次countDown方法计数器都会减1，直到计数器的值减为0时就代表条件已成熟，所有因调用await方法而阻塞的线程都会被唤醒。

这就是CountDownLatch的内部机制，看起来很简单，无非就是阻塞一部分线程让其在达到某个条件之后再执行。但是CountDownLatch的应用场景却比较广泛，只要你脑洞够大利用它就可以玩出各种花样。最常见的一个应用场景是开启多个线程同时执行某个任务，等到所有任务都执行完再统计汇总结果。下图动态演示了闭锁阻塞线程的整个过程。

上图演示了有5个线程因调用await方法而被阻塞，它们需要等待计数器的值减为0才能继续执行。计数器的初始值在构造闭锁时被指定，后面随着每次countDown方法的调用而减1。

---

2、CountDownLatch的使用案例

【案例1】

我们看下 Doug Lea 在 java doc 中给出的例子，这个例子非常实用，我们经常会写这个代码。

假设我们有 N ( N > 0 ) 个任务，那么我们会用 N 来初始化一个 CountDownLatch，然后将这个 latch 的引用传递到各个线程中，在每个线程完成了任务后，调用 latch.countDown() 代表完成了一个任务。调用 latch.await() 的方法的线程会阻塞，直到所有的任务完成。

package com.zju.CountDownLatch;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;

public class Driver {

	public static void main(String[] args) throws InterruptedException {
		
		CountDownLatch doneSignal = new CountDownLatch(10);
		Executor e = Executors.newFixedThreadPool(4);
		
		// 创建10个任务，提交给线程池来执行
		for (int i = 0; i < 10; i++) {
			e.execute(new WorkerRunnable(doneSignal, i));
		}
		
		// 等待所有的任务都完成了，这个方法才会返回
		doneSignal.await();
	}
}

class WorkerRunnable i