Java的concurrent包裏面的CountDownLatch其實可以把它看作一個計數器,只不過這個計數器的操作是原子操作,同時只能有一個線程去操作這個計數器,也就是同時只能有一個線程去減這個計數器裏面的值。
你可以向CountDownLatch對象設置一個初始的數字作爲計數值,任何調用這個對象上的await()方法都會阻塞,直到這個計數器的計數值被其他的線程減爲0爲止。
CountDownLatch的一個非常典型的應用場景是:有一個任務想要往下執行,但必須要等到其他的任務執行完畢後纔可以繼續往下執行。假如我們這個想要繼續往下執行的任務調用一個CountDownLatch對象的await()方法,其他的任務執行完自己的任務後調用同一個CountDownLatch對象上的countDown()方法,這個調用await()方法的任務將一直阻塞等待,直到這個CountDownLatch對象的計數值減到0爲止。
舉個例子:有三個工人在爲老闆幹活,這個老闆有一個習慣,就是當三個工人把一天的活都幹完了的時候,他就來檢查所有工人所幹的活。記住這個條件:三個工人先全部幹完活,老闆才檢查。所以在這裏用Java代碼設計兩個類,Worker代表工人,Boss代表老闆,具體的代碼實現如下:
package org.zapldy.concurrent;
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 + "正在幹活!");
}
}
package org.zapldy.concurrent;
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("工人活都幹完了,老闆開始檢查了!");
}
}
package org.zapldy.concurrent;
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();
}
}
當你運行CountDownLatchDemo這個對象的時候,你會發現是等所有的工人都幹完了活,老闆纔來檢查,下面是我本地機器上運行的一次結果,可以肯定的每次運行的結果可能與下面不一樣,但老闆檢查永遠是在後面的。
王二正在幹活!
李四正在幹活!
老闆正在等所有的工人幹完活......
張三正在幹活!
張三活幹完了!
王二活幹完了!
李四活幹完了!
工人活都幹完了,老闆開始檢查了!