需求:根据查询出来的一批合同编号,查询每个合同对应流程节点的最后审批人,最后导出合同编号及其对应的最后审批人。
本次代码优化总共分为以下四步骤:
- 遍历查询数据库
- 使用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;
}