Java多线程文件遍历完成后，如何等待所有线程结束？

针对你遇到的问题，这里有几个可靠的解决方案，以及可能的CountDownLatch误用排查：

方案1：使用ExecutorService的shutdown() + awaitTermination() #

这是最常用的等待线程池所有任务完成的方式，步骤清晰且稳定：

1. 文件遍历线程结束后，调用线程池的shutdown()方法——该方法会拒绝新任务提交，但允许已提交的任务继续执行。
2. 调用awaitTermination(long timeout, TimeUnit unit)等待所有任务完成，设置一个符合业务场景的超时时间（比如足够大的数值，或根据文件规模动态调整）。

示例代码：

// 文件遍历线程结束后执行以下逻辑
executorService.shutdown(); // 停止接受新任务
try {
    // 等待所有任务完成，这里设置超时为1小时，可根据实际需求调整
    if (!executorService.awaitTermination(1, TimeUnit.HOURS)) {
        // 超时后可选择强制关闭线程池，或做自定义处理
        executorService.shutdownNow();
    }
} catch (InterruptedException e) {
    // 处理中断异常
    executorService.shutdownNow();
    Thread.currentThread().interrupt();
}
// 此时所有索引任务已完成，可安全进入查询循环

方案2：排查CountDownLatch的误用 #

如果之前用CountDownLatch未生效，大概率是用法存在问题，正确的使用逻辑应该是：

1. 若能提前确定要提交的IndexTask总数，则用该数值初始化CountDownLatch；若任务总数动态变化（遍历过程中随时提交），则CountDownLatch不适用，优先选方案1。
2. 每个IndexTask的run()方法末尾，必须调用latch.countDown()，确保任务完成后计数减1。
3. 文件遍历线程结束后，在主程序中调用latch.await()，等待计数归0后再进入查询循环。

方案3：使用CompletionService（可选） #

如果需要跟踪每个任务的完成状态或处理任务返回结果，可以用CompletionService，但单纯等待所有任务完成的场景下，方案1的简洁性更优。

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