Java CountDownLatch示例代碼
Java的concurrent包里面的CountDownLatch其實可以把它看作一個計數器,只不過這個計數器的操作是原子操作,同時只能有一個線程去操作這個計數器,也就是同時只能有一個線程去減這個計數器里面的值。
你可以向CountDownLatch對象設置一個初始的數字作為計數值,任何調用這個對象上的await()方法都會阻塞,直到這個計數器的計數值被其他的線程減為0為止。
CountDownLatch的一個非常典型的應用場景是:有一個任務想要往下執行,但必須要等到其他的任務執行完畢后才可以繼續往下執行。假如我 們這個想要繼續往下執行的任務調用一個CountDownLatch對象的await()方法,其他的任務執行完自己的任務后調用同一個 CountDownLatch對象上的countDown()方法,這個調用await()方法的任務將一直阻塞等待,直到這個 CountDownLatch對象的計數值減到0為止。
舉個例子,有三個工人在為老板干活,這個老板有一個習慣,就是當三個工人把一天的活都干完了的時候,他就來檢查所有工人所干的活。記住這個條件:三 個工人先全部干完活,老板才檢查。所以在這里用Java代碼設計兩個類,Worker代表工人,Boss代表老板,具體的代碼實現如下:
import java.util.Random;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;public class Worker implements Runnable{
private CountDownLatch downLatch; private String name; public Worker(CountDownLatch downLatch, String name){ this.downLatch = downLatch; this.name = name; } public void run() { this.doWork(); try{ TimeUnit.SECONDS.sleep(new Random().nextInt(10)); }catch(InterruptedException ie){ } System.out.println(this.name + "活干完了!"); this.downLatch.countDown(); } private void doWork(){ System.out.println(this.name + "正在干活!"); }
} </pre>
import java.util.concurrent.CountDownLatch;public class Boss implements Runnable {
private CountDownLatch downLatch; public Boss(CountDownLatch downLatch){ this.downLatch = downLatch; } public void run() { System.out.println("老板正在等所有的工人干完活......"); try { this.downLatch.await(); } catch (InterruptedException e) { } System.out.println("工人活都干完了,老板開始檢查了!"); }
} </pre>
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;public class CountDownLatchDemo {
public static void main(String[] args) { ExecutorService executor = Executors.newCachedThreadPool(); CountDownLatch latch = new CountDownLatch(3); Worker w1 = new Worker(latch,"張三"); Worker w2 = new Worker(latch,"李四"); Worker w3 = new Worker(latch,"王二"); Boss boss = new Boss(latch); executor.execute(w3); executor.execute(w2); executor.execute(w1); executor.execute(boss); executor.shutdown(); }
} </pre>
當你運行CountDownLatchDemo這個對象的時候,你會發現是等所有的工人都干完了活,老板才來檢查,下面是我本地機器上運行的一次結果,可以肯定的每次運行的結果可能與下面不一樣,但老板檢查永遠是在后面的。
王二正在干活! 李四正在干活! 老板正在等所有的工人干完活...... 張三正在干活! 張三活干完了! 王二活干完了! 李四活干完了! 工人活都干完了,老板開始檢查了!