如何使用CountDownLatch

這篇文章主要介紹“如何使用CountDownLatch”，在日常操作中，相信很多人在如何使用CountDownLatch問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”如何使用CountDownLatch”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

“CountDownLatch”

想象一下這樣一個場景，當我們需要等待某些線程執(zhí)行完之后，再執(zhí)行主線程的代碼，要怎么實現(xiàn)?

可能有人會說，簡單，用 join() 方法等待線程執(zhí)行完成之后再執(zhí)行主線程就行了，實現(xiàn)代碼是這樣的：

// 創(chuàng)建線程1 Thread t1 = new Thread(new Runnable() {     @Override     public void run() {         // do something     } }); t1.start();  // 創(chuàng)建線程2 Thread t2 = new Thread(new Runnable() {     @Override     public void run() {         // do something     } }); t2.start();  // 等待線程 1和線程 2 執(zhí)行完 t1.join(); t2.join();

當然，如果使用的是 Thread 來執(zhí)行任務，那這種寫法也是可行的。然而真實的(編碼)環(huán)境中我們是不會使用 Thread  來執(zhí)行多任務的，而是會使用線程池來執(zhí)行多任務，這樣可以避免線程重復啟動和銷毀所帶來的性能開銷，實現(xiàn)代碼如下：

// 創(chuàng)建固定線程數的線程池 ExecutorService executorService = Executors.newFixedThreadPool(2); // 任務一 executorService.submit(new Runnable() {     @Override     public void run() {         // do something     } }); // 任務二 executorService.submit(new Runnable() {     @Override     public void run() {         // do something     } });

那么這時候問題來了，線程池是沒有 join() 方法的，那要怎么實現(xiàn)等待呢?

這個時候就要派出我方大將“CountDownLatch”啦。

吾有上將潘鳳，可斬華雄... 出場數秒，潘鳳...“卒”。

等等導演，我覺得劇情應該是這樣的...

CountDownLatch使用

為了實現(xiàn)等待所有線程池執(zhí)行完之后再執(zhí)行主線程的邏輯，我決定使用  AQS(AbstractQueuedSynchronizer，抽象同步框架)下的著名類 CountDownLatch 來實現(xiàn)此功能，具體的實現(xiàn)代碼如下：

public static void main(String[] args) throws InterruptedException {     // 創(chuàng)建 CountDownLatch     CountDownLatch countDownLatch = new CountDownLatch(2);      // 創(chuàng)建固定線程數的線程池     ExecutorService executorService = Executors.newFixedThreadPool(2);     // 任務一     executorService.submit(new Runnable() {         @Override         public void run() {             // do something             try {                 // 讓此任務執(zhí)行 1.2s                 Thread.sleep(1200);             } catch (InterruptedException e) {                 e.printStackTrace();             }             System.out.println("我是任務一");             countDownLatch.countDown();         }     });     // 任務二     executorService.submit(new Runnable() {         @Override         public void run() {             // do something             try {                 // 讓此任務執(zhí)行 1.2s                 Thread.sleep(1000);             } catch (InterruptedException e) {                 e.printStackTrace();             }             System.out.println("我是任務二");             countDownLatch.countDown();         }     });          // 等待任務執(zhí)行完成     countDownLatch.await();     System.out.println("程序執(zhí)行完成~"); }

以上程序執(zhí)行結果如下：

從上述結果可以看出，主線程的執(zhí)行是等待任務一和任務二都執(zhí)行完成之后才執(zhí)行的。

CountDownLatch實現(xiàn)原理

CountDownLatch 中 count down 是倒數的意思，latch  則是門閂的含義。整體含義可以理解為倒數的門栓，似乎有點“321，芝麻開門”的感覺，CountDownLatch 的作用也正是如此。

CountDownLatch 在創(chuàng)建的時候需要傳入一個整數，在這個整數“倒數”到 0  之前，主線程需要一直掛起等待，直到其他的線程都執(zhí)行之后，主線才能繼續(xù)執(zhí)行。

CountDownLatch執(zhí)行流程

CountDownLatch 的實現(xiàn)是在其內部創(chuàng)建并維護了一個 volatile 類型的整數計數器，當調用 countDown()  方法時，會嘗試將整數計數器 -1，當調用 wait() 方法時，當前線程就會判斷整數計數器是否為 0，如果為 0，則繼續(xù)往下執(zhí)行，如果不為  0，則使當前線程進入等待狀態(tài)，直到某個線程將計數器設置為 0，才會喚醒在 await() 方法中等待的線程繼續(xù)執(zhí)行。

CountDownLatch常用方法

// 線程被掛起直到 count 值為 0 才繼續(xù)執(zhí)行 public void await() throws InterruptedException { };     // 和 await() 類似，只不過等待一定的時間后 count 值還沒變?yōu)?nbsp;0 的話就會繼續(xù)執(zhí)行 public boolean await(long timeout, TimeUnit unit) throws InterruptedException { };    // 將 count 值減 1 public void countDown() { };

到此，關于“如何使用CountDownLatch”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續(xù)學習更多相關知識，請繼續(xù)關注億速云網站，小編會繼續(xù)努力為大家?guī)砀鄬嵱玫奈恼拢?/p>