三、Java多线程并发同步之CountDownLatch

版权声明：本文为博主原创文章，欢迎转载，转载请注明出处 https://blog.csdn.net/hnkd16/article/details/84258188

概述

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是多个线程相互等待，任何一个线程完成之前，所有的线程都必须等待。

阅读更多