java使用CountDownLatch等待多线程全部执行完成

java使用CountDownLatch等待多线程全部执行完成

前言

CountDownLatch 允许一个或多个线程等待其他线程完成操作。

应用场景

假如有一个列表的大量数据等待处理，最后全部处理完毕后返回处理结果omsNon。普通做法就是从头遍历，一个个顺序执行，这样单线程处理效率不高，我们希望使用多线程的方式处理，同时在主线程等待所有子线程处理完成。

CountDownLatch的构造函数接收一个int类型的参数作为计数器，如果你想等待N个点完成，这里就传入N。

当我们调用一次CountDownLatch的countDown方法时，N就会减1，CountDownLatch的await会阻塞当前线程，直到N变成零。由于countDown方法可以用在任何地方，所以这里说的N个点，可以是N个线程，也可以是1个线程里的N个执行步骤。用在多个线程时，你只需要把这个CountDownLatch的引用传递到线程里。

示例代码

public static void main(String args[]){

//初始化计数

AtomicInteger count = new AtomicInteger(0);

//创建10个线程

ExecutorService executorService = Executors.newFixedThreadPool(10);

//设置CountDownLatch为10

CountDownLatch countDownLatch = new CountDownLatch(10);

for(int i=0;i<10;i++){

executorService.execute(() -> {

//每个线程累加10000次

for(int j=0;j<10000;j++){

count.addAndGet(1);

}

//线程最后执行countDown

countDownLatch.countDown();

});

}

try{

//调用await方法等待

countDownLatch.await();

}catch(Exception e){

e.printStackTrace();

}

System.out.println(count.toString());

}

最后的执行结果就是10*10000=100000，可见所有线程都处理完了。

扩展

CountDownLatch的await方法还支持超时时间的设置，当等待超时时间子线程还没执行完将不再等待继续执行主线程。

public boolean await(long timeout, TimeUnit unit)

熟练使用多线程处理，可以再处理大量重复性工作时发挥多线程处理的效率优势。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。