CountDownLatch
作用和原理:
- Java1.5后引入,作用:使得一个线程等待其他线程执行完之后再执行该线程
- 实现原理:通过一个计数器实现,该计数器的初始值就是需要等待执行完的线程的数量,每当一个线程执行完毕后,计数器的值就-1,当计数器的值为0的时候,表示需要等待执行完的线程都执行完了,此时在闭锁上等待的线程就可以恢复工作了
源码:
- countDownLatch类中只提供了一个构造器:
//参数count为计数值
public CountDownLatch(int count) { };
- 类中有三个方法是最重要的:
//调用await()方法的线程会被挂起,它会等待直到count值为0才继续执行
public void await() throws InterruptedException { };
//和await()类似,只不过等待一定的时间后count值还没变为0的话就会继续执行
public boolean await(long timeout, TimeUnit unit) throws InterruptedException { };
//将count值减1
public void countDown() { };
代码示例和注解:
- 使用CountDownLatch的情况下:
import java.util.concurrent.CountDownLatch;
/**
* CountDownLatch:倒计时计数类 Java1.5后引入,作用:使得一个线程等待其他线程执行完之后再执行该线程
* 实现原理:通过一个计数器实现,该计数器的初始值就是需要等待执行完的线程的数量
* ,每当一个线程执行完毕后,计数器的值就-1,当计数器的值为0的时候,表示需要等待执行完的线程都执行完了,此时在闭锁上等待的线程就可以恢复工作了
*
*/
public class CountDownLatchDemo {
public static void main(String[] args) {
System.out.println("主线程开始执行");
// 模拟需要等待6个线程被执行完
CountDownLatch countDownLatch = new CountDownLatch(6);
for (int i = 0; i < 6; i++) {
new Thread(() -> {
try {
Thread.sleep(1);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(Thread.currentThread().getName()+ "\t 线程开始执行");
countDownLatch.countDown();// 执行计数
}, String.valueOf(i)).start();
}
// 主线程等待
try {
countDownLatch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("主线程继续执行...");
}
}
执行结果:
- 如果将下面这句注释掉:
countDownLatch.countDown();// 执行计数
那么执行结果:
那么程序会卡在这里,因为无法通过下面代码的校验,就无法进行下一步
countDownLatch.await();
- 未使用使用CountDownLatch的情况下:
打印结果: