JAVA多線程之CountDownLatch
前序:
上周測試給開發的同事所開發的模塊提出了一個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 = newMythread1(); 13 Thread t2 = new Thread(th2, "Mythread1"); 14 t2.start(); 15 System.out.println(this.currentThread().getName()); 16 } 17 class Mythread1 extends Thread 18 { 19 public void run() {20 try { 21 Thread.sleep(1000); 22 } catch (InterruptedException e) { 23 // TODO Auto-generated catch block 24 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 Thread 26 { 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