CountDownLatch是Java语言提供的一种同步工具类,它可以使线程一直等待,直到计数器减到0,才能执行后面的任务。它的作用类似于等待线程池中的所有任务完成。在多线程编程中,我们可能需要在某个线程执行完后再执行另一个线程,或者等待多个线程执行完后再执行某个任务,这时就可以使用CountDownLatch来实现线程间的同步。CountDownLatch经常被用来控制初始化线程等待应用程序中的其他线程完成预备操作。
我们可以规定线程数量为10,验证CountDownLatch的特性
int threadCount = 10;
java.util.concurrent.CountDownLatch countDownLatch = new java.util.concurrent.CountDownLatch(threadCount);
使用线程池的工具类创建一个固定数量的线程池
ExecutorService executorService = Executors.newFixedThreadPool(10);
循环10个任务开启10个线程
for (int i = 0; i < threadCount; i++) {
executorService.execute(
() -> {
try {
Thread.sleep(1000);
} catch (Exception e) {
e.printStackTrace();
}
System.out.println("线程" + Thread.currentThread().getName() + "开启运行");
countDownLatch.countDown();
}
);
}
在每个子线程中执行完各自任务后我们调用其方法countDown()进行计数减一。
public static void main(String[] args) {
int threadCount = 10;
java.util.concurrent.CountDownLatch countDownLatch = new java.util.concurrent.CountDownLatch(threadCount);
ExecutorService executorService = Executors.newFixedThreadPool(10);
//启动十个线程任务
for (int i = 0; i < threadCount; i++) {
executorService.execute(
() -> {
try {
Thread.sleep(1000);
} catch (Exception e) {
e.printStackTrace();
}
System.out.println("线程" + Thread.currentThread().getName() + "开启运行");
countDownLatch.countDown();
}
);
}
try {
countDownLatch.await();
} catch (Exception e) {
e.printStackTrace();
}
executorService.shutdown();
System.out.println("全部执行完毕");
}
这是整个main的运行,mian方法为主线程,在主线程里面调用await()方法将任务进行阻塞,直到CountDownLatch的计数为0之后,代表所有子线程全部执行完毕,接下来主线程继续执行就会打印执行完毕。