Java CountDownLatch應用

燮羽 2011-01-08

展開全文

Java的concurrent包里面的CountDownLatch其實可以把它看作一個計數(shù)器，只不過這個計數(shù)器的操作是原子操作，同時只能有一個線程去操作這個計數(shù)器，也就是同時只能有一個線程去減這個計數(shù)器里面的值。

你可以向CountDownLatch對象設置一個初始的數(shù)字作為計數(shù)值，任何調(diào)用這個對象上的await()方法都會阻塞，直到這個計數(shù)器的計數(shù)值被其他的線程減為0為止。

CountDownLatch的一個非常典型的應用場景是：有一個任務想要往下執(zhí)行，但必須要等到其他的任務執(zhí)行完畢后才可以繼續(xù)往下執(zhí)行。假如我們這個想要繼續(xù)往下執(zhí)行的任務調(diào)用一個CountDownLatch對象的await()方法，其他的任務執(zhí)行完自己的任務后調(diào)用同一個CountDownLatch對象上的countDown()方法，這個調(diào)用await()方法的任務將一直阻塞等待，直到這個CountDownLatch對象的計數(shù)值減到0為止。

舉個例子，有三個工人在為老板干活，這個老板有一個習慣，就是當三個工人把一天的活都干完了的時候，他就來檢查所有工人所干的活。記住這個條件：三個工人先全部干完活，老板才檢查。所以在這里用Java代碼設計兩個類，Worker代表工人，Boss代表老板，具體的代碼實現(xiàn)如下：

Java代碼 
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 + "正在干活!");  
    }  
      
}  

Java代碼 
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("工人活都干完了，老板開始檢查了！");  
    }  
  
}  

Java代碼 
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這個對象的時候，你會發(fā)現(xiàn)是等所有的工人都干完了活，老板才來檢查，下面是我本地機器上運行的一次結果，可以肯定的每次運行的結果可能與下面不一樣，但老板檢查永遠是在后面的。

Java代碼 
王二正在干活!  
李四正在干活!  
老板正在等所有的工人干完活......  
張三正在干活!  
張三活干完了！  
王二活干完了！  
李四活干完了！  
工人活都干完了，老板開始檢查了！  

好了，就寫到這里，睡覺去了！

http://zapldy./blog/746458

另一個比較簡單的例子

public static void main(String[] args) throws InterruptedException {
int threadNumber = 10;
final CountDownLatch countDownLatch = new CountDownLatch(threadNumber);
for (int i = 0; i < threadNumber; i++) {
final int threadID = i;
new Thread() {
public void run() {
try {
Thread.sleep((long) (Math.random() * 10000));
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("threadID:[%s] finished!!", threadID));
countDownLatch.countDown();
}
}.start();
}

countDownLatch.await();
System.out.println("main thread finished!!");
}