CountDownLatch是在java1.5被引入的,存在于java.util.concurrent包下。CountDownLatch这个类能够使一个线程等待一定数量(计数器)线程完成各自的工作后再执行后续代码。例如,应用程序的主线程希望在负责启动框架服务的线程已经启动所有的框架服务之后再执行。
CountDownLatch是通过一个计数器来实现的,计数器的初始值为线程的数量。每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务,然后在闭锁上等待的线程就可以恢复执行任务。
实例
源码:
/**
* 线程实现类
*/
public class Worker implements Runnable {
/**
* 线程计数器
*/
private CountDownLatch countDownLatch ;
/**
* 线程睡眠时长
*/
private long timeout;
public Worker(CountDownLatch countDownLatch,long timeout){
this.countDownLatch = countDownLatch;
this.timeout = timeout;
}
@Override
public void run() {
try {//模拟业务操作,睡眠1秒
TimeUnit.SECONDS.sleep(timeout);
System.out.println("线程:" + Thread.currentThread().getName() + "执行完毕!");
} catch (InterruptedException e) {
e.printStackTrace();
}
countDownLatch.countDown();
}
}
/**
* Created by wanghao30 on 2017/9/6.
*/
public class CountDownLatchMain {
//执行线程数
private static int theadNumbers = 10;
public static void main(String[] args) throws InterruptedException {
//创建计数器
CountDownLatch countDownLatch = new CountDownLatch(theadNumbers);
for(int i = 0 ; i < theadNumbers ; i++){
new Thread(new Worker(countDownLatch,i)).start();
}
StringBuilder sb = new StringBuilder();
while (countDownLatch.getCount() > 0){
sb.append("=");
double percent = (double) ((theadNumbers - countDownLatch.getCount()) * 100) / theadNumbers;
System.out.println(sb.toString() + percent + "%");
TimeUnit.MILLISECONDS.sleep(1000);
}
double percent = (double) ((theadNumbers - countDownLatch.getCount()) * 100) / theadNumbers;
System.out.println(sb.toString() + percent + "%");
countDownLatch.await();
System.out.println("-------------------------------------------------------------------------------------");
System.out.println("所有线程执行完毕,主线程继续执行!");
}
}
输出
线程:Thread-0执行完毕!
=10.0%
线程:Thread-1执行完毕!
==20.0%
线程:Thread-2执行完毕!
===30.0%
线程:Thread-3执行完毕!
====40.0%
线程:Thread-4执行完毕!
=====50.0%
线程:Thread-5执行完毕!
======60.0%
线程:Thread-6执行完毕!
=======70.0%
线程:Thread-7执行完毕!
========80.0%
线程:Thread-8执行完毕!
=========90.0%
线程:Thread-9执行完毕!
=========100.0%
-------------------------------------------------------------------------------------
所有线程执行完毕,主线程继续执行!
注意事项
- 构造器中的计数值(count)实际上就是闭锁需要等待的线程数量。这个值只能被设置一次,而且CountDownLatch没有提供任何机制去重新设置这个计数值。
- 每个线程必须拥有闭锁对象
- 在所有线程启动后,主线程必须使用await()进入等待。
- 线程执行业务结束,调用countDown(),进行计数器操作。
- 使用getCount()可以获取当前还未执行完成的线程数。
使用场景
- 开始执行前等待n个线程完成各自任务:例如应用程序启动类要确保在处理用户请求前,所有N个外部系统已经启动和运行了。