等待 Java 线程池中的线程子集

Question

假设我有一个包含 X 项的线程池，并且给定任务使用这些项中的 Y（其中 Y 比 Y 小得多） >X）。

我想等待给定任务（Y 项）的所有线程完成，而不是整个线程池。

如果线程池的 execute() 方法返回对所使用线程的引用，我可以简单地 join() 到这些 Y 线程中的每一个，但是事实并非如此。

有谁知道一种优雅的方法来实现这一点？谢谢。

Answer 1

创建以 Y 作为计数的 CountDownLatch，并在每个 Y 任务中执行 latch.countDown()。最后，latch.await() 将确保所有 Y 任务都完成。

Answer 2

您可以使用 CyclicBarrier 并让每个线程仅在其类型为 Y 时等待。例如。

ExecutorService executor = Executors.newFixedThreadPool(X.size);

public void executeAllAndAwaitCompletion(List<? extends Y> myRunnableY){
   final CyclicBarrier barrier = new CyclicBarrier(myRunnable.size()+1);
   for(final Y y : myRunnableY){
       executor.submit(new Runnable(){
           public void run(){
                y.run();//for the sake of this example y has a run method
                barrier.await();
           }
       }); 
    }
   barrier.await();
}

因此，运行类型 Y 的每个线程都会等待，直到所有这些 Y 都完成。请注意，您必须将屏障大小加 1，以考虑到最初执行的线程也将等待。

另请注意：如果迈克尔·博格沃特的示例适合您，那就最好了。但是，如果您需要线程池，对于运行 Y 的每个线程不运行任何其他非 Y 线程，那么我的解决方案将是您可以做到这一点的唯一方法。 Future.get() 只会阻塞调用线程，当线程池完成 Y 的执行时，它将开始执行一些其他（可能是非 Y）任务。

Answer 3

而不是 execute()ing Runnable，只需 invokeAll() 一些 Callable - 然后你为每个对象获取一个 Future，您可以在其上调用 get()，它会阻塞直到任务完成。

Answer 4

您应该使用 CompletionService 正是用于此目的。

• 创建 Executor
• 使用 Executor 创建 ExecutorCompletionService 通过
• CompletionService 提交任务
• 使用 take 或 poll 在 CompletionService 上等待任务完成
• 重复直到您提交的所有任务都完成
• 完成

您可以与其他东西共享 Executor，只需在顶部创建 CompletionService 并将其用于您的特定任务任务。