latch java_Java并发编程:CountDownLatch的使用以及一个容易踩到的陷阱

使用简介

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

}

}

}

}

运行结果:

8a991f23849e2e815c9d06f2298b9c67.png


版权声明:本文为weixin_29942159原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。