如何在ForkJoinPool提交的异步任务全部执行完成后执行后续代码

你调用的isDone()、isCompletedNormally()属于非阻塞的状态查询方法，本身设计就不会等待任务执行完成，自然会立即返回。下面是不同场景下的可行实现方案：

场景1：允许阻塞当前线程，等待任务完成后再返回响应 #

直接调用submit()返回的ForkJoinTask对象的get()方法，该方法会阻塞当前线程直到任务执行结束：

ForkJoinPool customThreadPool = new ForkJoinPool(4);
ForkJoinTask<?> task = customThreadPool.submit(() -> {
    list.parallelStream().forEach(this::process);
});
// 阻塞等待，可传入时间参数设置超时，避免无限等待
task.get(30, TimeUnit.MINUTES);
// 任务执行完成后，运行后续逻辑
afterProcess();

注意需要处理InterruptedException、ExecutionException、TimeoutException三类异常。

场景2：控制器需立即返回响应，后续逻辑异步触发 #

不需要阻塞请求线程，通过CompletableFuture的回调机制实现任务完成后自动触发后续逻辑：

ForkJoinPool customThreadPool = new ForkJoinPool(4);
CompletableFuture.runAsync(() -> {
    list.parallelStream().forEach(this::process);
}, customThreadPool).whenComplete((res, err) -> {
    if (err == null) {
        // 任务正常结束，执行后续逻辑
        afterProcess();
    } else {
        // 任务执行异常，自行补充异常处理逻辑
        log.error("并行处理任务失败", err);
    }
});
// 控制器直接返回响应
return Result.success("任务已后台启动");

补充：池用完即销毁的场景 #

如果当前ForkJoinPool是一次性使用、不需要复用，可以调用awaitTermination方法等待所有任务完成：

customThreadPool.shutdown();
// 等待所有已提交任务执行完成，最多等待1小时
customThreadPool.awaitTermination(1, TimeUnit.HOURS);
// 后续处理逻辑
afterProcess();

注意调用shutdown()后线程池将不再接收新的任务，不适用线程池单例复用的场景。

内容的提问来源于stack exchange，提问作者user3145373 ツ