多線程執行完某個業務時,需要等到所有線程結束後在執行後面的邏輯的實現

最近在項目中需要開啓多個線程去批量分析數據,但是必須等全部數據分析結束後執行後續操作。

下面記錄一種簡單的實現方案:

在全局定義一個計數器,初始化需要執行的次數,在每一次執行完一個線程的時候進行減一操作,直到所有的線程執行完成後,計數器會變爲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("所有定時任務結束!");
    }
}

 

 

 

 

 

 


 

 

 

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