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(); }