@Data
@AllArgsConstructor
public class FutureTaskWorker<T, R> {
/**
* @description 需要异步执行的任务
*/
private List<T> taskList;
/**
* @description 需要执行的方法
*/
private Function<T, CompletableFuture<R>> workFunction;
/**
* @description 搜集执行结果
* @return: java.util.List<R>
*/
public List<R> getAllResult() {
List<CompletableFuture<R>> futureList = taskList.stream().map(workFunction).collect(Collectors.toList());
CompletableFuture<Void> allCompletableFuture = CompletableFuture.allOf(futureList.toArray(new CompletableFuture[futureList.size()]));
return allCompletableFuture.thenApply(e -> futureList.stream().map(CompletableFuture::join).collect(Collectors.toList())).join();
}
}
调用
List<DTO> reqDTOS = new ArrayList<>();
reqDTOS.add(DTO);
...
-- getResultAndUpload 自己的操作
FutureTaskWorker<DTO, DTO> futureTaskWorker = new FutureTaskWorker<>(reqDTOS, (DTO e) -> getResultAndUpload(e));
long beginTime = System.currentTimeMillis();
List<DTO> allResult = futureTaskWorker.getAllResult();
long endTime = System.currentTimeMillis();
log.info("结束耗时:{}", (endTime - beginTime));