1. 程式人生 > >使用CountDownLatch等待多執行緒

使用CountDownLatch等待多執行緒

前言  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 熟練使用多執行緒處理,可以再處理大量重複性工作時發揮多執行緒處理的效率優勢。