Java:多线程等待所有线程结束(CountDownLatch/CyclicBarrier)

本文主要是参考官方文档做一学习用途。

官方链接：

http://docs.oracle.com/javase/1.5.0/docs/api/java/util/concurrent/CountDownLatch.html

http://docs.oracle.com/javase/1.5.0/docs/api/java/util/concurrent/CyclicBarrier.html

多线程设计过程中，经常会遇到需要等待其它线程结束以后再做其他事情的情况，比如多线程下载文件，每个线程都会下载文件的一部分，在所有线程结束以后，需要将各部分再次拼接成一个完整的文件。

有几种方案：

1.在主线程中设置一自定义全局计数标志，在工作线程完成时，计数减一。主线程侦测该标志是否为0，一旦为0，表示所有工作线程已经完成。

2.使用Java标准的类CountDownLatch来完成这项工作，原理是一样的，计数。

CountDownLatch

CountDownLatch 初始化设置count，即等待(await)count个线程或一个线程count次计数，通过工作线程来countDown计数减一，直到计数为0，await阻塞结束。

设置的count不可更改，如需要动态设置计数的线程数，可以使用CyclicBarrier.

下面的例子，所有的工作线程中准备就绪以后，并不是直接运行，而是等待主线程的信号后再执行具体的操作。

package com.example.multithread;

import java.util.concurrent.CountDownLatch;

class Driver
{
	private static final int TOTAL_THREADS = 10;
	private final CountDownLatch mStartSignal = new CountDownLatch(1);
	private final CountDownLatch mDoneSignal = new CountDownLatch(TOTAL_THREADS);

	void main()
	{
		for (int i = 0; i < TOTAL_THREADS; i++)
		{
			new Thread(new Worker(mStartSignal, mDoneSignal, i)).start();
		}
		System.out.println("Main Thread Now:" + System.currentTimeMillis());
		doPrepareWork();// 准备工作
		mStartSignal.countDown();// 计数减一为0，工作线程真正启动具体操作
		doSomethingElse();//做点自己的事情
		try
		{
			mDoneSignal.await();// 等待所有工作线程结束
		}
		catch (InterruptedException e)
		{
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		System.out.println("All workers have finished now.");
		System.out.println("Main Thread Now:" + System.currentTimeMillis());
	}

	void doPrepareWork()
	{
		System.out.println("Ready,GO!");
	}

	void doSomethingElse()
	{
		for (int i = 0; i < 100000; i++)
		{
			;// delay
		}
		System