CountDownLatch類是一個同步計數器,構造時傳入int參數,該參數就是計數器的初始值,每調用一次countDown()方法,計數器減1,計數器大於0 時,await()方法會阻塞程序繼續執行。CountDownLatch可以看作是一個倒計數的鎖存器,當計數減至0時觸發特定的事件。利用這種特性,可以讓主線程等待子線程的結束。下面以一個模擬運動員比賽的例子加以說明。
CountDownLatch的一個非常典型的應用場景是:有一個任務想要往下執行,但必須要等到其他的任務執行完畢後纔可以繼續往下執行。假如我們這個想要繼續往下執行的任務調用一個CountDownLatch對象的await()方法,其他的任務執行完自己的任務後調用同一個CountDownLatch對象上的countDown()方法,這個調用await()方法的任務將一直阻塞等待,直到這個CountDownLatch對象的計數值減到0爲止。
下面是張孝祥老師所舉例子的程序代碼和運行結果:
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class CountDownLatchTest {
public static void main(String [] args){
ExecutorService service = Executors.newCachedThreadPool();
final CountDownLatch cdOrder = new CountDownLatch(1);
final CountDownLatch cdAnswer = new CountDownLatch(3);
for(int i=0;i<3;i++){
Runnable runnable=new Runnable(){
public void run(){
try {
System.out.println("線程"+Thread.currentThread().getName()+"正準備接受命令");
cdOrder.await();
System.out.println("線程"+Thread.currentThread().getName()+"已接受命令");
Thread.sleep((long)(Math.random()*10000));
System.out.println("線程"+Thread.currentThread().getName()+"迴應命令處理結果");
cdAnswer.countDown();
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
};
service.execute(runnable);
}
try {
System.out.println("線程"+Thread.currentThread().getName()+"即將發佈命令");
cdOrder.countDown();
System.out.println("線程"+Thread.currentThread().getName()+"已接受命令,正在等待結果");
cdAnswer.await();
Thread.sleep((long)(Math.random()*10000));
System.out.println("線程"+Thread.currentThread().getName()+"已收到所有響應結果");
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
service.shutdown();
}
}
運行結果:
線程pool-1-thread-1正準備接受命令
線程pool-1-thread-2正準備接受命令
線程pool-1-thread-3正準備接受命令
線程main即將發佈命令
線程main已接受命令,正在等待結果
線程pool-1-thread-2已接受命令
線程pool-1-thread-1已接受命令
線程pool-1-thread-3已接受命令
線程pool-1-thread-2迴應命令處理結果
線程pool-1-thread-1迴應命令處理結果
線程pool-1-thread-3迴應命令處理結果
線程main已收到所有響應結果