CountDownLatch可用于管理一组线程,适用于将一个任务分成若干线程执行,等到所有线程执行完,再进行汇总处理。
package test;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class TestCountDownLatch {
public static void main(String[] args) {
ExecutorService service = Executors.newCachedThreadPool();
final CountDownLatch cdOrder = new CountDownLatch(1);
final CountDownLatch cdAnswer = new CountDownLatch(3);
for (int i = 0; i < 3; i++) {
Runnable runnable = new Runnable() {
public void run() {
try {
System.out.println("线程" + Thread.currentThread().getName() + "等待主线程任务");
cdOrder.await();
System.out.println("线程" + Thread.currentThread().getName() + "已接收任务");
Thread.sleep((long) (Math.random() * 10000));
System.out.println("线程" + Thread.currentThread().getName() + "任务完成,结果返回主线程");
cdAnswer.countDown();
} catch (Exception e) {
e.printStackTrace();
}
}
};
service.execute(runnable);
}
try {
Thread.sleep((long) (Math.random() * 10000));
System.out.println("线程" + Thread.currentThread().getName() + "给子线程发布任务");
cdOrder.countDown();
System.out.println("线程" + Thread.currentThread().getName() + "已发布任务,正在等待结果");
cdAnswer.await();
System.out.println("线程" + Thread.currentThread().getName() + "已收到所有响应结果");
} catch (Exception e) {
e.printStackTrace();
}
service.shutdown();
}
}
运行结果如下:
线程pool-1-thread-1等待主线程任务
线程pool-1-thread-2等待主线程任务
线程pool-1-thread-3等待主线程任务
线程main给子线程发布任务
线程main已发布任务,正在等待结果
线程pool-1-thread-1已接收任务
线程pool-1-thread-3已接收任务
线程pool-1-thread-2已接收任务
线程pool-1-thread-1任务完成,结果返回主线程
线程pool-1-thread-2任务完成,结果返回主线程
线程pool-1-thread-3任务完成,结果返回主线程
线程main已收到所有响应结果