异步执行一组任务等待所有任务执行完成的2种方式

 private static ThreadPoolExecutor threadPool = new ThreadPoolExecutor(5, 10, 60, TimeUnit.SECONDS,
        new ArrayBlockingQueue<>(1000), new ThreadFactoryBuilder().setNameFormat("metrix-pool-%d").build(), new ThreadPoolExecutor.AbortPolicy());

    /** 方式一
     * 主线程统一收集结果,适用于希望单一任务设置超时
     */
    @Test
    public void test1() {
        long begin = System.currentTimeMillis();

        List<Future<Integer>> result = IntStream.rangeClosed(1, 4).mapToObj(i -> threadPool.submit(getAsyncTask(i)))
            .collect(Collectors.toList());

        List<Integer> list = result.stream().map(future -> {

            try {
                return future.get();
            } catch (Exception e) {
                e.printStackTrace();
                return -1;

            }

        }).collect(Collectors.toList());

        log.info("result {} took {} ms", JSON.toJSONString(list), System.currentTimeMillis() - begin);

    }

    private Callable<Integer> getAsyncTask(int i) {

        return () -> {

            TimeUnit.SECONDS.sleep(i);
            return i;
        };

    }

    /**  方式二
     * 子线程自己添加结果,适用于希望整体有一个等待超时
     */
    @Test
    public void test02() {
        long begin = System.currentTimeMillis();
        CountDownLatch countDownLatch = new CountDownLatch(4);
        List<Integer> result = new ArrayList<>();

        IntStream.rangeClosed(1, 4).forEach(i ->
            threadPool.execute(executeAsyncTask(i, countDownLatch, result))
        );

        try {

            countDownLatch.await(6,TimeUnit.SECONDS);
        } catch (InterruptedException e) {

        }

        log.info("result {} took {} ms", JSON.toJSONString(result), System.currentTimeMillis() - begin);

    }

    private Runnable executeAsyncTask(int i, CountDownLatch countDownLatch, List<Integer> result) {
        return () -> {

            try {
                TimeUnit.SECONDS.sleep(i);

            } catch (Exception e) {
                e.printStackTrace();
            }

            synchronized (result) {
                result.add(i);
            }

            countDownLatch.countDown();
        };
    }

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值