Java的CountDownLatch是Java1.5新出现的类,用作在多个线程都执行完成之后再执行特定方法。latch译作门闩,CountDownLatch可以理解为程序里若干门闩,或是计数器,至于到底有几个门闩,可以在CountDownLatch的构造方法中指定,当门闩都被打开后,程序才会继续执行。调用CountDownLatch的countDown方法就表示打开了一个门闩,门闩都被打开后,CountDownLatch的await()方法会被调用。
若有的线程执行时间可能会很长,此时不想主线程一直等的话,可以在await()方法中设置最长的等待时间。若构造方法中设置的门闩数量大于实际线程的数量,只要门闩都被打开,且到了主线程的最长等待时间,主线程就会继续执行。
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
public class CountDownTest {
public static void main(String[] args) {
CountDownLatch countDownLatch = new CountDownLatch(4);
System.out.println(countDownLatch.getCount());
new Thread(() -> {
System.out.println(Thread.currentThread().getName() + " 线程正在执行");
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
countDownLatch.countDown();
System.out.println(Thread.currentThread().getName() + " 线程执行完毕");
}).start();
new Thread(() -> {
System.out.println(Thread.currentThread().getName() + " 线程正在执行");
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
countDownLatch.countDown();
System.out.println(Thread.currentThread().getName() + " 线程执行完毕");
}).start();
System.out.println("两个子线程正在执行中");
try {
//主线程最多等待三秒钟,调用await方法的线程会被挂起
countDownLatch.await(3, TimeUnit.SECONDS);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("两个子线程执行完毕,继续执行主线程");
}
}