Java 并发编程（十二）：并发工具 CountDownlatch

1、CountDownlatch 计数器

Java多线程编程中经常会有这样的场景，某个线程需要等待一个或多个线程操作结束（或达到某种状态）才开始执行，此时可以通过CountDownLatch轻松实现。

2、类图

CountDownLatch工作原理相对简单，可以简单看成一个倒计数器，通过指定的count值进行初始化，调用await方法的线程将被阻塞，直到count值通过countDown()方法减小到0，所有等待的线程才会被释放继续执行。另外CountDownLatch不可能重新初始化或者修改CountDownLatch对象的内部计数器的值

countDown() ： 如果当前计数器的值大于1，则将其减1；若当前值为1，则将其置为0并唤醒所有通过await等待的线程；若当前值为0，则什么也不做直接返回。

await()： 等待计数器的值为0，若计数器的值为0则该方法返回；若等待期间该线程被中断，则抛出InterruptedException并清除该线程的中断状态。

await(long timeout, TimeUnit unit)： 在指定的时间内等待计数器的值为0，若在指定时间内计数器的值变为0，则该方法返回true；若指定时间内计数器的值仍未变为0，则返回false；若指定时间内计数器的值变为0之前当前线程被中断，则抛出InterruptedException并清除该线程的中断状态。

getCount() ： 读取当前计数器的值

3、代码示例

package com.lkf.concurrent;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;

import static java.util.concurrent.Executors.newFixedThreadPool;


public class CountDownlatchTest {
    /**
     * 计数器，用来控制线程数量，传入参数2，表示计数器计数为2
     */
    private final static CountDownLatch M_COUNT_DOWN_LATCH = new CountDownLatch(2);

    /**
     * 示例工作线程类
     */
    private static class WorkerThreadA implements Runnable {
        private final String mThreadName;
        private final int mSleepTime;

        public WorkerThreadA(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();
            }
            M_COUNT_DOWN_LATCH.countDown();
            System.out.println("[" + mThreadName + "] end!");
        }
    }

    /**
     * 工作线程类
     */
    private static class WorkerThreadB implements Runnable {

        @Override
        public void run() {
            System.out.println("[WorkerThread] started!");
            try {
                // 阻塞在这里等待 mCountDownLatch 里的count变为0；                
                M_COUNT_DOWN_LATCH.await();
            } catch (InterruptedException e) {

            }
            System.out.println("[WorkerThread] end!");
        }
    }

    public static void main(String[] args) throws Exception {
        ExecutorService executorService = newFixedThreadPool(3);
        // 最先run WorkerThread
        executorService.submit(new WorkerThreadB());
        // 运行两个工作线程
        // 工作线程1运行3秒        
        executorService.submit(new WorkerThreadA("WorkingThread1", 3000));
        // 工作线程2运行2秒
        executorService.submit(new WorkerThreadA("WorkingThread2", 2000));
    }
}

注意：计数器必须大于等于0，当它等于0时，调用await 方法时是不会阻塞当前线程的。另外 CountDownLatch 对象的内部计数器的值不能重新初始化或赋值。