閱讀291 返回首頁    go 阿裏雲 go 技術社區[雲棲]

並發工具類（一）等待多線程完成的CountDownLatch

簡介

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

應用場景

假如有這樣一個需求，當我們需要解析一個Excel裏多個sheet的數據時，可以考慮使用多線程，每個線程解析一個sheet裏的數據，等到所有的sheet都解析完之後，程序需要提示解析完成。在這個需求中，要實現主線程等待所有線程完成sheet的解析操作，最簡單的做法是使用join。代碼如下：

01 public class JoinCountDownLatchTest {
02  
03     public static void main(String[] args) throws InterruptedException {
04         Thread parser1 = new Thread(new Runnable() {
05             @Override
06             public void run() {
07             }
08         });
09  
10         Thread parser2 = new Thread(new Runnable() {
11             @Override
12             public void run() {
13                 System.out.println("parser2 finish");
14             }
15         });
16  
17         parser1.start();
18         parser2.start();
19         parser1.join();
20         parser2.join();
21         System.out.println("all parser finish");
22     }
23  
24 }

join用於讓當前執行線程等待join線程執行結束。其實現原理是不停檢查join線程是否存活，如果join線程存活則讓當前線程永遠wait，代碼片段如下，wait(0)表示永遠等待下去。

1 while (isAlive()) {
2  wait(0);
3 }

直到join線程中止後，線程的this.notifyAll會被調用，調用notifyAll是在JVM裏實現的，所以JDK裏看不到，有興趣的同學可以看看JVM源碼。JDK不推薦在線程實例上使用wait，notify和notifyAll方法。

而在JDK1.5之後的並發包中提供的CountDownLatch也可以實現join的這個功能，並且比join的功能更多。

01 <pre>public class CountDownLatchTest {
02  
03     static CountDownLatch c = new CountDownLatch(2);
04  
05     public static void main(String[] args) throws InterruptedException {
06         new Thread(new Runnable() {
07             @Override
08             public void run() {
09                 System.out.println(1);
10                 c.countDown();
11                 System.out.println(2);
12                 c.countDown();
13             }
14         }).start();
15  
16         c.await();
17         System.out.println("3");
18     }
19  
20 }

CountDownLatch的構造函數接收一個int類型的參數作為計數器，如果你想等待N個點完成，這裏就傳入N。

當我們調用一次CountDownLatch的countDown方法時，N就會減1，CountDownLatch的await會阻塞當前線程，直到N變成零。由於countDown方法可以用在任何地方，所以這裏說的N個點，可以是N個線程，也可以是1個線程裏的N個執行步驟。用在多個線程時，你隻需要把這個CountDownLatch的引用傳遞到線程裏。

其他方法

如果有某個解析sheet的線程處理的比較慢，我們不可能讓主線程一直等待，所以我們可以使用另外一個帶指定時間的await方法，await(long time, TimeUnit unit): 這個方法等待特定時間後，就會不再阻塞當前線程。join也有類似的方法。

注意：計數器必須大於等於0，隻是等於0時候，計數器就是零，調用await方法時不會阻塞當前線程。CountDownLatch不可能重新初始化或者修改CountDownLatch對象的內部計數器的值。一個線程調用countDown方法 happen-before 另外一個線程調用await方法。

最後更新：2017-05-23 17:32:01

  上一篇：go  分享一個HP EVA4400/6400/8400/P6000通用的數據恢複方法
  下一篇：go  處理JSON的Java API ：JSON的簡介