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

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

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)

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值