SpringBoot线程池,@Async 注解。实现主线程等待所有子线程执行完毕再结束。

        log.info("---主线程开始");
        List<Future> futureList = new ArrayList<>();

        List<VideoInfoDTO> videos = specsConfig.getVideos();
        if (CollectionUtils.isNotEmpty(videos)) {
            videos.forEach(video ->{
                Future future = videoCheckService.videoCheck(video,jsonVO.get());
                futureList.add(future);
            });
        }
        Set<String> msgResult = new HashSet<>();
        while (true) {
            if (CollectionUtils.isNotEmpty(futureList)) {
                boolean isAllDone = true;
                for (Future future : futureList) {
                    if (null == future || !future.isDone()) {
                        isAllDone = false;
                    }else {
                        try {
                            String msg =  (String) future.get();
                            if (StringUtils.isNotEmpty(msg)) {
                                msgResult.add(msg);
                            }
                        } catch (Exception e) {
                            log.info("---视频校验线程池处理单个视频校验出错!error:{}",ErrorUtils.errInfo(e));
                        }
                    }
                }
                if (isAllDone) {
                    break;
                }
            }
        }
        log.info("---主线程结束");
    @Override
    @Async("checkVideoExecutor")
    public Future videoCheck(VideoInfoDTO videoInfo, SpecsConfigJsonVO vo){

        String msg = "视频:" + videoInfo.getVideoName();
        boolean pass = true;
        //校验规格
        VideoInfoVo infoVo = mediaService.getVideoInfo(videoInfo.getVideoId());
        if (!Optional.ofNullable(infoVo).isPresent()) {
            msg = msg + ",查询不到上传视频信息!";
            return new AsyncResult(msg);
        }
        if (vo.getFileSizeUpperLimit() != null && !BigDecimal.ZERO.equals(vo.getFileSizeUpperLimit())) {
            int fileSizeUpper =  getFireSize(vo,2);
            if (infoVo.getFileSize().intValue() > fileSizeUpper) {
                msg = msg + "文件大小大于最大文件大小!";
                pass = false;
            }
        }
        if (pass){
            return new AsyncResult("");
        }else {
            return new AsyncResult(msg);
        }
    }

videoCheckService 的 videoCheck方法相当于是执行子线程逻辑的一个方法,使用了线程池注解@Async。

大体逻辑就是每个子线程方法都会有一个返回对象Future,然后主线程代码逻辑里有一个while循环再一直判断所有的子线程是否已经全部完成了,完成了那么主线程再开始走下一步

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值