Java并发编程：CountDownLatch的使用以及一个容易踩到的陷阱

2019独角兽企业重金招聘Python工程师标准>>>

使用简介

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

运行结果:

转载于:https://my.oschina.net/u/2338224/blog/1612549