CountDownLatch介绍
CountDownLatch是一个并发工具类, 作用是允许一个或多个线程等待其他线程完成操作。
我们有时会称之为发令枪。有一个个形象的例子能说明其功能:公司召开一个全体股东会,需要所有股东到场了才能正式开始。会议室准备了一个花名册,记录了需要到会的总人数,每一个股东来到会议室就要签名确认,同时总人数减一,当需要到会的总人数为0时,会议就正式开始。所以,CountDownLatch就有充当计数器和发令枪的功能。
模拟并发请求
通过CountDownLatch的计数和发令功能,我们可以通过它模拟并发请求的功能,常常能用于测试。下面的代码时一个模拟并发请求的示例:
//concurrentCount 为并发量
final CountDownLatch countDownLatch = new CountDownLatch(concurrentCount);
for (int i = 0; i < concurrentCount; i++) {
new Thread(() -> {
try {
System.out.println("线程: " + Thread.currentThread().getName() + "等待");
//使线程阻塞,等待计数器 concurrentCount 为 0 时,阻塞的线程会同时开始执行
countDownLatch.await();
//测试并发需要执行的代码,或者调用方法
} catch (InterruptedException e) {
e.printStackTrace();
}
}).start();
//计数器减 1
countDownLatch.countDown();
}