java 等待所有线程执行完毕

#问题的由来
主线程等待所有线程都运行完后,再执行逻辑,这个需求很普遍。比如,在处理数据时,为了效率期间,起了10个线程,分别处理一块数据,这样能缩短处理时间,10个线程都执行完后,继续进行下边的逻辑(有点map/reduce的意思)。这里我们就需要用到java的线程同步类 CountDownLatch ,count down是计数的意思,latch是门闩的意思,合起来也就是计数的开关。我们看一下CountDownLatch的API。

#CountDownLatch API介绍

一个同步辅助类,在完成一组正在其他线程中执行的操作之前,它允许一个或多个线程一直等待。

用给定的计数 初始化 CountDownLatch。由于调用了 countDown() 方法,所以在当前计数到达零之前,await
方法会一直受阻塞。之后,会释放所有等待的线程,await
的所有后续调用都将立即返回。这种现象只出现一次——计数无法被重置。如果需要重置计数,请考虑使用 CyclicBarrier。

CountDownLatch 是一个通用同步工具,它有很多用途。将计数 1 初始化的 CountDownLatch
用作一个简单的开/关锁存器,或入口:在通过调用 countDown() 的线程打开入口前,所有调用 await 的线程都一直在入口处等待。用
N 初始化的 CountDownLatch 可以使一个线程在 N 个线程完成某项操作之前一直等待,或者使其在某项操作完成 N
次之前一直等待。

CountDownLatch 的一个有用特性是,它不要求调用 countDown
方法的线程等到计数到达零时才继续,而在所有线程都能通过之前,它只是阻止任何线程继续通过一个 await。

从中可以看出,最重要的几个步骤,

第一,创建门闩,

第二,开启门闩,阻塞所有线程,

第三,其它线程递减次数,减为0时,门闩打开,主线程继续执行放开门闩

#样例

public class TestCountDownLatch {

	
	public static void main(String[] args) {
		
		int count = 5;
		final CountDownLatch latch = new CountDownLatch(count);//创建门闩
		
		//execute thread
		for(int i=0; i<count; i++){
			new Thread(new Worker(latch)).start();
		}
		//
		System.out.println("进入等待");
		
		
		try {
			latch.await();//开启门闩,阻塞所有线程,
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		
		//dosomething
		System.out.println("都执行完了");
	}
	
	
	
}

class Worker implements Runnable{
	
	private final CountDownLatch latch;
	
	
	
	public Worker(CountDownLatch latch) {
		super();
		this.latch = latch;
	}
	
	
	
	@Override
	public void run() {
		
		
		//do something
		System.out.println(Thread.currentThread().getName());
		
		try {
			Thread.sleep(5000);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		
		/*
		其它线程递减次数,减为0时,门闩打开,主线程继续执行放开门闩。
		每调用一次countdown()计数就减一次
		*/
		latch.countDown();//递减
	}
	
	
}

输出结果

这里写图片描述

#后续
这里,我们没有往线程中传值、也没有获取线程的计算结果,所以比较简单。如果需要往线程中传值,使用构造函数方法传参或者公共方法传参;获取其它线程的计算结果,可以使用Future

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值