JAVA多線程之CountDownLatch

標籤：end   停止   log   業務   countdown   處理   main   對象   啟用   

前序:

上周測試給開發的同事所開發的模組提出了一個bug，並且還是偶現。

經過仔細查看代碼，發現是在業務中啟用了多線程，2個線程同時跑，但是新啟動的2個線程必須保證一個完成之後另一個再繼續運行，才能消除bug。

什麼時候用？

多線程是在很多地方都會用到的，但是我們如果想要實現在某個特定的線程運行完之後，再啟動另外一個線程呢，這個時候CountDownLatch就可以派上用場了

怎麼用？

先看看普通的多線程代碼：

 1 package code; 2  3 public class MyThread extends Thread {  4     public static void main(String[] args) {       5         MyThread th = new MyThread();    6         Thread t1 = new Thread(th, "Mythread");         7         t1.start();   8         System.out.println(Thread.currentThread().getName());   9         }        10         public void run() 11             {     12                 Mythread1 th2 = new Mythread1();   13                 Thread t2 = new Thread(th2, "Mythread1");        14                 t2.start();  15                 System.out.println(this.currentThread().getName());  16             }  17         class Mythread1 extends Thread18         {19             public void run() {    20                 try {21                     Thread.sleep(1000);22                 } catch (InterruptedException e) {23                     // TODO Auto-generated catch block24                     e.printStackTrace();25                 }26                 System.out.println(this.currentThread().getName()); 27                 }  28             29         }30     }

代碼如上，先用MyThread繼承了Thread類，然後在MyThread類內部又寫了一個MyThread1類，同樣也是繼承了Thread類，並且在run方法裡面讓它睡1秒，這樣運行代碼，就會列印出：

從上面的輸出順序可以看出，先是啟動了main線程，然後再啟動了MyThread線程，在MyThread線程中，又啟動了MyThread1線程。但是由於讓MyThread1線程睡了1秒，類比處理後續業務，這樣他就比MyThread運行完畢的時間晚一些。

現在，在代碼中加上CountDownLatch ，要讓MyThread1先運行完畢，再讓MyThread繼續運行。

 1 package code; 2  3 import java.util.concurrent.CountDownLatch; 4  5 public class MyThread extends Thread {  6     CountDownLatch countDownLatch = new CountDownLatch(1); 7     public static void main(String[] args) {       8         MyThread th = new MyThread();    9         Thread t1 = new Thread(th, "Mythread");        10         t1.start();  11         System.out.println(Thread.currentThread().getName());  12         }        13         public void run() 14             {     15                 Mythread1 th2 = new Mythread1();   16                 Thread t2 = new Thread(th2, "Mythread1");        17                 t2.start();  18                 try {19                     countDownLatch.await();20                 } catch (InterruptedException e) {21                     e.printStackTrace();22                 }23                 System.out.println(this.currentThread().getName());  24             }  25         class Mythread1 extends Thread26         {27             public void run() {    28                 try {29                     Thread.sleep(1000);30                 } catch (InterruptedException e) {31                     e.printStackTrace();32                 }33                 System.out.println(this.currentThread().getName()); 34                 countDownLatch.countDown();35                 }  36             37         }38     }

代碼寫法如上所示，大致分三步

1，我們先new一個CountDownLatch對象入參設定為1（我個人理解的這個就像是new一個數組一樣，什麼時候數組清空了，那就可以讓被中斷的線程繼續運行了）

2，在MyThread類中調用countDownLatch.await();讓當前線程停止運行。

3，在Mythread1類中調用countDownLatch.countDown()方法。當Mythread1全部執行完畢，再最後調用該方法，作用就是把我說的“數組”清空。

看看輸出的列印結果

結果如，是符合預期的結果的。

最後再說下CountDownLatch countDownLatch = new CountDownLatch(1)的入參，這塊設定的是1，那就需要調用一次countDownLatch.countDown()減去1。

如果是其他數字，那就要調用相應的次數，否則調用countDownLatch.await()的線程都不會被繼續執行。

 

JAVA多線程之CountDownLatch