最近在项目中需要开启多个线程去批量分析数据,但是必须等全部数据分析结束后执行后续操作。
下面记录一种简单的实现方案:
在全局定义一个计数器,初始化需要执行的次数,在每一次执行完一个线程的时候进行减一操作,直到所有的线程执行完成后,计数器会变为0,这时就可以在计数器为0时执行线程结束后的逻辑了。
在java中使用CountDownLatch对象可以帮助我们在每一次线程结束的时候减一,具体实现如下:
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class ThreadTest {
public static void main(String[] args) throws InterruptedException {
// 创建一个固定大小的线程池
int TEST_THREAD_COUNT = 5;
ExecutorService service = Executors.newFixedThreadPool(TEST_THREAD_COUNT);
// 模拟10次定时任务
for (int i = 0; i < 10; i++) {
// 需要启动的线程数
final CountDownLatch latch = new CountDownLatch(10);
for (int j = 0; j < 10; j++) {
// 分析每个数据
int finalJ = j;
service.execute(()->{
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
latch.countDown();
e.printStackTrace();
}
System.out.println("分析xx同学"+ finalJ);
latch.countDown();
});
}
try{
// 等待所有线程结束
latch.await();
}catch (Exception e){
e.printStackTrace();
}
System.out.println("定时任务"+i+"完成");
Thread.sleep(10000);
}
System.out.println("所有定时任务结束!");
}
}