最近在項目中需要開啓多個線程去批量分析數據,但是必須等全部數據分析結束後執行後續操作。
下面記錄一種簡單的實現方案:
在全局定義一個計數器,初始化需要執行的次數,在每一次執行完一個線程的時候進行減一操作,直到所有的線程執行完成後,計數器會變爲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("所有定時任務結束!");
}
}