之前开发遇到一个场景,需要开启多线程上传图片到阿里云,将所以图片上传完成后,将图片地址一起提交到后端。那么这里就需要监听这些多线程是否所有都执行完成。之前做的是用AtomicInteger来计数,每执行完一个线程,将这个变量+1,然后判断该值是否等于启动的线程数,如果相等了,则表明所有线程执行完毕。那么这种场景还可以直接用CountDownLatch来进行实现。
CountDownLatch是一个同步工具类,它通过一个计数器来实现的,初始值为线程的数量。每当一个线程完成了自己的任务,计数器的值就相应得减1。当计数器到达0时,表示所有的线程都已执行完毕,然后在等待的线程就可以恢复执行任务。
方法详解
CountDownLatch(int count):count为计数器的初始值(一般需要多少个线程执行,count就设为几)。
countDown(): 每调用一次计数器值-1,直到count被减为0,代表所有线程全部执行完毕。
getCount():获取当前计数器的值。
await(): 等待计数器变为0,即等待所有异步线程执行完毕。
CountDownLatch的使用场景
1.某一线程在开始运行前等待n个线程执行完毕。
2.实现多个线程开始执行任务的最大并行性。
实例:业务线程等待子线程初始化执行完成再执行
public class CountDownLatchUtil {
private static CountDownLatch countDownLatch = new CountDownLatch(6);
public static void main(String args[]) {
for (int i=0;i<5;i++) {
Thread thread = new Thread(new InitThread());
thread.start();
}
Thread thread = new Thread(new BusiThread());
thread.start();
}
private static class InitThread implements Runnable {
@Override
public void run() {
System.out.println(Thread.currentThread().getName() + ":执行初始化任务");
countDownLatch.countDown();
}
}
private static class BusiThread implements Runnable {
@Override
public void run() {
try {
countDownLatch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("BusiThread" + "执行业务任务");
}
}
}
创建5个子线程执行run,run方法中调用countDown方法进行计数-1。同时创建BusiThread任务,在run中调用await,等待计数为0,再执行自己的业务任务。