使用CountDownLatch等待多執行緒
阿新 • • 發佈:2018-12-18
前言 CountDownLatch 允許一個或多個執行緒等待其他執行緒完成操作。 應用場景 假如有一個列表的大量資料等待處理,最後全部處理完畢後返回處理結果。普通做法就是從頭遍歷,一個個順序執行,這樣單執行緒處理效率不高,我們希望使用多執行緒的方式處理,同時在主執行緒等待所有子執行緒處理完成。
CountDownLatch的建構函式接收一個int型別的引數作為計數器,如果你想等待N個點完成,這裡就傳入N。 當我們呼叫一次CountDownLatch的countDown方法時,N就會減1,CountDownLatch的await會阻塞當前執行緒,直到N變成零。由於countDown方法可以用在任何地方,所以這裡說的N個點,可以是N個執行緒,也可以是1個執行緒裡的N個執行步驟。用在多個執行緒時,你只需要把這個CountDownLatch的引用傳遞到執行緒裡。
示例程式碼
public static void main(String args[]){ //初始化計數 AtomicInteger count = new AtomicInteger(0); //建立10個執行緒 ExecutorService executorService = Executors.newFixedThreadPool(10); //設定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); } //執行緒最後執行countDown countDownLatch.countDown(); }); } try{ //呼叫await方法等待 countDownLatch.await(); }catch(Exception e){ e.printStackTrace(); } System.out.println(count.toString()); }
最後的執行結果就是10*10000=100000,可見所有執行緒都處理完了。
擴充套件 CountDownLatch的await方法還支援超時時間的設定,當等待超時時間子執行緒還沒執行完將不再等待繼續執行主執行緒。
public boolean await(long timeout, TimeUnit unit) 1 熟練使用多執行緒處理,可以再處理大量重複性工作時發揮多執行緒處理的效率優勢。