上一篇:Java中的阻塞队列 BlockingQueue 详解
本文目录:
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