java使用CountDownLatch等待多線程全部執(zhí)行完成

發(fā)布時間：2020-10-04 04:11:25 來源：腳本之家閱讀：213 作者：張小鳴欄目：編程語言

前言

CountDownLatch 允許一個或多個線程等待其他線程完成操作。

應(yīng)用場景

假如有一個列表的大量數(shù)據(jù)等待處理，最后全部處理完畢后返回處理結(jié)果。普通做法就是從頭遍歷，一個個順序執(zhí)行，這樣單線程處理效率不高，我們希望使用多線程的方式處理，同時在主線程等待所有子線程處理完成。

CountDownLatch的構(gòu)造函數(shù)接收一個int類型的參數(shù)作為計數(shù)器，如果你想等待N個點完成，這里就傳入N。
當(dāng)我們調(diào)用一次CountDownLatch的countDown方法時，N就會減1，CountDownLatch的await會阻塞當(dāng)前線程，直到N變成零。由于countDown方法可以用在任何地方，所以這里說的N個點，可以是N個線程，也可以是1個線程里的N個執(zhí)行步驟。用在多個線程時，你只需要把這個CountDownLatch的引用傳遞到線程里。

示例代碼

public static void main(String args[]){
  //初始化計數(shù)
  AtomicInteger count = new AtomicInteger(0);
  //創(chuàng)建10個線程
  ExecutorService executorService = Executors.newFixedThreadPool(10);
  //設(shè)置CountDownLatch為10
  CountDownLatch countDownLatch = new CountDownLatch(10);
  for(int i=0;i<10;i++){
   executorService.execute(() -> {
    //每個線程累加10000次
    for(int j=0;j<10000;j++){
     count.addAndGet(1);
    }
    //線程最后執(zhí)行countDown
    countDownLatch.countDown();
   });
  }
  try{
   //調(diào)用await方法等待
   countDownLatch.await();
  }catch(Exception e){
   e.printStackTrace();
  }
  System.out.println(count.toString());
 }

最后的執(zhí)行結(jié)果就是10*10000=100000，可見所有線程都處理完了。

擴展

CountDownLatch的await方法還支持超時時間的設(shè)置，當(dāng)?shù)却瑫r時間子線程還沒執(zhí)行完將不再等待繼續(xù)執(zhí)行主線程。

public boolean await(long timeout, TimeUnit unit)

熟練使用多線程處理，可以再處理大量重復(fù)性工作時發(fā)揮多線程處理的效率優(yōu)勢。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。

向AI問一下細節(jié)

java使用CountDownLatch等待多線程全部執(zhí)行完成

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽