多线程执行完某个业务时,需要等到所有线程结束后在执行后面的逻辑的实现

最近在项目中需要开启多个线程去批量分析数据,但是必须等全部数据分析结束后执行后续操作。

下面记录一种简单的实现方案:

在全局定义一个计数器,初始化需要执行的次数,在每一次执行完一个线程的时候进行减一操作,直到所有的线程执行完成后,计数器会变为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("所有定时任务结束!");
    }
}

 

 

 

 

 

 


 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章