使用 CountDownLatch 控制多個線程執行順序

已同步更新至：http://dxjia.cn/2015/08/countdownlatch-use/    

有時候會有這樣的需求，多個線程同時工作，然后其中幾個可以隨意並發執行，但有一個線程需要等其他線程工作結束后，才能開始。舉個例子，開啟多個線程分塊下載一個大文件，每個線程只下載固定的一截，最后由另外一個線程來拼接所有的分段，那么這時候我們可以考慮使用CountDownLatch來控制並發。

    CountDownLatch是JAVA提供在java.util.concurrent包下的一個輔助類，可以把它看成是一個計數器，其內部維護着一個count計數，只不過對這個計數器的操作都是原子操作，同時只能有一個線程去操作這個計數器，CountDownLatch通過構造函數傳入一個初始計數值，調用者可以通過調用CounDownLatch對象的cutDown()方法，來使計數減1；如果調用對象上的await()方法，那么調用者就會一直阻塞在這里，直到別人通過cutDown方法，將計數減到0，才可以繼續執行。

示例

import java.util.concurrent.CountDownLatch;

public class Sample {
    /**
     * 計數器，用來控制線程
     * 傳入參數2，表示計數器計數為2
     */
    private final static CountDownLatch mCountDownLatch = new CountDownLatch(2);

    /**
     * 示例工作線程類
     */
    private static class WorkingThread extends Thread {
        private final String mThreadName;
        private final int mSleepTime;
        public WorkingThread(String name, int sleepTime) {
            mThreadName = name;
            mSleepTime = sleepTime;
        }
        
        @Override
        public void run() {
            System.out.println("[" + mThreadName + "] started!");
            try {  
                    Thread.sleep(mSleepTime);  
            } catch (InterruptedException e) {  
                    e.printStackTrace();  
            }
            mCountDownLatch.countDown();
            System.out.println("[" + mThreadName + "] end!"); 
        }
    }
    
    /**
     * 示例線程類
     */
    private static class SampleThread extends Thread {
        
        @Override
        public void run() {
            System.out.println("[SampleThread] started!");
            try {
                // 會阻塞在這里等待 mCountDownLatch 里的count變為0；
                // 也就是等待另外的WorkingThread調用countDown()
                mCountDownLatch.await();
            } catch (InterruptedException e) {
                
            }
            System.out.println("[SampleThread] end!");
        }
    }
    
    public static void main(String[] args) throws Exception {
        // 最先run SampleThread
        new SampleThread().start();
        // 運行兩個工作線程
        // 工作線程1運行5秒
        new WorkingThread("WorkingThread1", 5000).start();
        // 工作線程2運行2秒
        new WorkingThread("WorkingThread2", 2000).start();
    }
}

 

運行結果：

[SampleThread] started!
[WorkingThread1] started!
[WorkingThread2] started!
[WorkingThread2] end!
[WorkingThread1] end!
[SampleThread] end!

達到了目的。當然還有其他方式可以做到這樣的效果，本文僅僅是介紹了一種使用CountDownLatch的方式。