使用简介
CountDownLatch是通过一个计数器来实现的,当我们在new 一个CountDownLatch对象的时候需要带入该计数器值,该值就表示了线程的数量。每当一个线程完成自己的任务后,计数器的值就会减1。当计数器的值变为0时,就表示所有的线程均已经完成了任务,然后就可以恢复等待的线程继续执行了。
CountDownLatch所描述的是”在完成一组正在其他线程中执行的操作之前,它允许一个或多个线程一直等待“。在API中是这样描述的:
用给定的计数 初始化 CountDownLatch。由于调用了 countDown() 方法,所以在当前计数到达零之前,await 方法会一直受阻塞。之后,会释放所有等待的线程,await 的所有后续调用都将立即返回。这种现象只出现一次——计数无法被重置。如果需要重置计数,请考虑使用 CyclicBarrier。
应用示例
示例使用开会案例。老板进入会议室等待5个人全部到达会议室才会开会。所以这里有两个线程老板等待开会线程、员工到达会议室,下面这段程序有惊天陷阱,请勿copy!:
陷阱程序:
package com.chenjun.testxxx; import java.util.concurrent.CountDownLatch; public class CountDownLatchTest { private static CountDownLatch countDownLatch = new CountDownLatch(5); static class BossThread extends Thread { @Override public void run() { System.out.println("Boss在会议室等待,总共有" + countDownLatch.getCount() + "个人开会..."); try { // Boss等待 countDownLatch.await(); } catch (InterruptedException e) { e.printStackTrace(); } System.out.println("所有人都已经到齐了,开会吧..."); } // 员工到达会议室 static class EmpleoyeeThread extends Thread { @Override public void run() { try { System.out.println(Thread.currentThread().getName() + ",到达会议室...."); }finally { // 员工到达会议室 count - 1 countDownLatch.countDown(); } } } public static void main(String[] args) { // Boss线程启动 new BossThread().start(); for (long i = 0; i < countDownLatch.getCount(); i++) { new EmpleoyeeThread().start(); } } } }
运行这段程序发现经常程序无法结束,死等在原地, 原因就出现在main里面的for循环那一行,countDownLatch.getCount()是个会变的东西, 用来放在for循环里面,会影响循环计数,这是个很隐秘的陷阱!
改造后的程序:
package com.chenjun.testxxx; import java.util.concurrent.CountDownLatch; public class CountDownLatchTest { private static CountDownLatch countDownLatch = new CountDownLatch(5); static class BossThread extends Thread { @Override public void run() { System.out.println("Boss在会议室等待,总共有" + countDownLatch.getCount() + "个人开会..."); try { // Boss等待 countDownLatch.await(); } catch (InterruptedException e) { e.printStackTrace(); } System.out.println("所有人都已经到齐了,开会吧..."); } // 员工到达会议室 static class EmpleoyeeThread extends Thread { @Override public void run() { try { System.out.println(Thread.currentThread().getName() + ",到达会议室...."); }finally { // 员工到达会议室 count - 1 countDownLatch.countDown(); } } } public static void main(String[] args) { // Boss线程启动 new BossThread().start(); long cnt = countDownLatch.getCount(); for (long i = 0; i < cnt; i++) { new EmpleoyeeThread().start(); } } } }
运行结果: