概述
CountDownLatch:计数器,允许一个或多个线程一直等待,直到其他线程执行完后再执行。计数器的初始值为线程的数量 ,每当一个线程完成了自己的任务后,计数器的值就会减1。当计数器值到达0时,它表示所有的线程已经完成了任务 。
业务场景
例如用户模拟多个用户并发访问某个API,测试API的性能;如有一个任务,需要等其他5个任务执行完成后,才能执行。
实现
CountDownLatch(int count):创建一个指定数量的计数器,count数就是线程数。
countDown():释放线程,每调用一次这个方法,在构造函数中初始化的count值就减1
await():阻塞线程, 当count等于0,主线程就能通过await()方法,恢复执行自己的任务
getCount():获取当前线程数。
public class CountDownLatchTest {
public static void main(String[] args) {
final CountDownLatch latch = new CountDownLatch(5); //计数器
// 模拟5个线程
for (int i = 0; i < 5; i++) {
final int user = i+1;
new Thread(new Runnable() {
@Override
public void run() {
try {
System.out.println("任务"+user+"---正在执行。。。");
Thread.sleep((long) (Math.random() * 10000));
System.out.println("任务"+user+"---执行完成");
latch.countDown(); // 计数器减1
System.out.println("当前任务数:"+latch.getCount());
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}).start();
}
try {
latch.await(); // 阻塞,等待其他线程执行完成
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("主线程开始执行。。");
}
}
输出结果:
任务1---正在执行。。。
任务3---正在执行。。。
任务2---正在执行。。。
任务5---正在执行。。。
任务4---正在执行。。。
任务5---执行完成
当前任务数:4
任务1---执行完成
当前任务数:3
任务4---执行完成
当前任务数:2
任务2---执行完成
当前任务数:1
任务3---执行完成
当前任务数:0
所有线程执行完成,主线程开始执行。。
总结
CountDownLatch和CyclicBarrier一样都可以实现线程同步
CountDownLatch是一个计数器,当减到0时,生命周期就结束了,而CyclicBarrier可以重复使用;
CountDownLatch是一个线程(或者多个), 等待另外多个线程完成某个事情之后才能执行,而 CyclicBarrier是多个线程相互等待,任何一个线程完成之前,所有的线程都必须等待。