代码优化:CompletableFuture + countDownLatch + Vector
需求:根据查询出来的一批合同编号,查询每个合同对应流程节点的最后审批人,最后导出合同编号及其对应的最后审批人。
本次代码优化总共分为以下四步骤:
- 遍历查询数据库
- 使用CompletableFuture将串行改为并行
- 简化代码,并引出线程安全问题
- 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; }
- 在并发过程中,CountDownLatch减少至 0 的前提下,执行相关操作。
- CompletableFuture 使得我们的任务单独运行在与主线程分离的其他线程中,并且通过回调可以在主线程中得到异步任务的执行状态,是否完成,和是否异常等信息。
- 当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; }