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

时间:2022-09-23 09:48:29

前言

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

应用场景

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

countdownlatch的构造函数接收一个int类型的参数作为计数器,如果你想等待n个点完成,这里就传入n。
当我们调用一次countdownlatch的countdown方法时,n就会减1,countdownlatch的await会阻塞当前线程,直到n变成零。由于countdown方法可以用在任何地方,所以这里说的n个点,可以是n个线程,也可以是1个线程里的n个执行步骤。用在多个线程时,你只需要把这个countdownlatch的引用传递到线程里。

示例代码

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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方法还支持超时时间的设置,当等待超时时间子线程还没执行完将不再等待继续执行主线程。

?
1
public boolean await(long timeout, timeunit unit)

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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/Roy_70/article/details/80396273