Java并发编程之——CountDownLatch的使用

最近在开发Android项目的时候有一个需求：在Service中开启多个线程并发的执行任务，当这三个线程执行完毕之后，主线程才能继续往下执行。刚开始使用的是AsyncTask+AtomicInteger的方式，然后在onpostexecute回调中对AtomicInteger进行自增且判断其值是否达到了要求的值，如果达到了要求的值就继续执行相应的操作，否则跳过。但是发现这种方式并不能很好的实现想要的效果！于是研究了下Java并发编程中的CountDownLatch使用。

CountDownLatch位于包java.util.concurrent下面，它的应用场景是：有一个任务，需要等待其他几个任务完成之后才能继续执行，这个时候就可以使用CountDownLatch了。

CountDownLatch有一个构造器，count代表计数值，也就是任务的个数。

 public CountDownLatch(int count) {
        if (count < 0) throw new IllegalArgumentException("count < 0");
        this.sync = new Sync(count);
    }

还有三个方法：

public void await() throws InterruptedException {} //调用await线程会被挂起，直到count计数值为0才继续执行
public boolean await(long timeout, TimeUnit unit)throws InterruptedException {}//和await类似，只不过当等待时间后如果count还没变为0就继续执行。
public void countDown() {    }//count计数值减一

下面是演示代码：

package com.easyliu.java.demo;

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

public class CountDownLatchTest {
	private static CountDownLatch sCountDownLatch = null;
	private static final int THREAD_NUMBER = 3;

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		sCountDownLatch = new CountDownLatch(THREAD_NUMBER);
		//线程池
		ExecutorService fixedThreadPool = Executors
				.newFixedThreadPool(THREAD_NUMBER);
		//执行线程
		fixedThreadPool.execute(new ConsumeRunnable("one"));
		fixedThreadPool.execute(new ConsumeRunnable("two"));
		fixedThreadPool.execute(new ConsumeRunnable("three"));
		System.out.println("等待3个子线程执行完毕...");
		try {
			sCountDownLatch.await();
			System.out.println("3个子线程已经执行完毕");
			System.out.println("继续执行主线程");
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
	}

	private static class ConsumeRunnable implements Runnable {
		private String mName;

		public ConsumeRunnable(String name) {
			this.mName = name;
		}
		public void run() {
			System.out.println("子线程" + mName + "正在执行");
			try {
				Thread.sleep(3000);// 模拟耗时操作
				System.out.println("子线程" + mName + "执行完毕");
				sCountDownLatch.countDown();
			} catch (InterruptedException e) {
				e.printStackTrace();
			}
		}
	}
}

输入如下所示：

等待3个子线程执行完毕...
子线程two正在执行
子线程one正在执行
子线程three正在执行
子线程one执行完毕
子线程three执行完毕
子线程two执行完毕
3个子线程已经执行完毕
继续执行主线程