代码优化:CompletableFuture + countDownLatch + Vector

需求:根据查询出来的一批合同编号,查询每个合同对应流程节点的最后审批人,最后导出合同编号及其对应的最后审批人。

本次代码优化总共分为以下四步骤:

  1. 遍历查询数据库
  2. 使用CompletableFuture将串行改为并行
  3. 简化代码,并引出线程安全问题
  4. CompletableFuture + countDownLatch + Vector

最终版代码如下:

public List<ContractMan> selectContractMan() throws InterruptedException{
          
   
    CountDownLatch countDownLatch = new CountDownLatch(5);
    List<String> contractNumers= zhbExcelMapper.selectContractNumer();
    Vector<ContractMan> man=new Vector<>();

    List<List<String>> lists = averageAssign(contractNumers, 5);

    for(int i=0;i<lists.size();i++){
          
   
        int finalI = i;
        CompletableFuture future1 = CompletableFuture.supplyAsync(() -> {
          
   
            List<String> stringList = lists.get(finalI);
            for(String s:stringList){
          
   
                ContractMan contractMan =new ContractMan();
                contractMan.setContractNumber(s);
                String name= zhbExcelMapper.selectMan(s);
                contractMan.setName(name);
                man.add(contractMan);
            }
            return Boolean.TRUE;
        }).whenComplete((v,e)->{
          
   
            countDownLatch.countDown();
        });
    }

    countDownLatch.await();
    return man;
}
  1. 在并发过程中,CountDownLatch减少至 0 的前提下,执行相关操作。
  2. CompletableFuture 使得我们的任务单独运行在与主线程分离的其他线程中,并且通过回调可以在主线程中得到异步任务的执行状态,是否完成,和是否异常等信息。
  3. 当CompletableFuture 完成时 CountDownLatch 执行 countDown() 方法,全部完成后,countDownLatch.await(); 放行

评论区:单纯使用CompletableFuture实现

public List<ContractMan> selectContractMan() throws InterruptedException, ExecutionException {
          
   
        List<String> contractNumers= zhbExcelMapper.selectContractNumer();

        List<ContractMan> man=new ArrayList<>();

        List<List<String>> lists = averageAssign(contractNumers, 5);

        CompletableFuture[] futureArray = lists.stream()
                .map(idList -> CompletableFuture.supplyAsync(()->{
          
   
                            List<ContractMan> subMan=new ArrayList<>();
                            idList.forEach(i->{
          
   
                                ContractMan contractMan =new ContractMan();
                                contractMan.setContractNumber(i);
                                String name= zhbExcelMapper.selectMan(i);
                                contractMan.setName(name);
                                subMan.add(contractMan);
                            });
                            return subMan;
                        }
                )).toArray(CompletableFuture[]::new);
        CompletableFuture.allOf(futureArray).join();
        for (CompletableFuture t : futureArray) {
          
   
            man.addAll((Collection<? extends ContractMan>) t.get());
        }
        return man;
    }
经验分享 程序员 微信小程序 职场和发展