springboot 判断@Async注解下所有线程是否结束
因为在项目中想知道程序运行的时间,百度都只有判断单线程结束的方法,所以记录下自己的解决方案
//这是一个带有回调的多线程函数
@Async
public Future<String> getWaterValveInfo(String tableName) {
//方法略
return new AsyncResult<>( "任务完成");
}
public void getAllInformatin() throws ExecutionException, InterruptedException {
System.out.println("开始运行" + new Date());
List<HashMap<String, Object>> mapList = labMapper.selectAll();
//定义一个存所有线程回调函数的list
List<Future<String>> futureList=new ArrayList<>();
List<String> tableNameList=new ArrayList<>();
long start=System.currentTimeMillis();
if (!mapList.isEmpty()) {
for (int i = 0; i < mapList.size(); i++) {
String tableName = String.valueOf(mapList.get(i).get("machinename")).trim();
//将回调的信息存入list中
futureList.add(insertMongo.getWaterValveInfo(tableName));
}
}
**//判断进程是否全部结束**
while (true){
**//回调信息空了就结束**
if (futureList.isEmpty()){
break;
}
for (int i = 0; i < futureList.size(); i++) {
if (futureList.get(i).isDone()){
//判断线程结束,输出回调信息,并将该回调清除
System.out.println(futureList.get(i).get());
futureList.remove(i);
}else {
continue;
}
}
}
System.out.println("任务完成,耗时:"+ (System.currentTimeMillis()-start)+"ms");
}
}