Java ExecutorService：所有递归创建的任务的awaitTermination

Question

我使用ExecutorService来执行任务。该任务可以递归创建提交给同一任务的其他任务，ExecutorService那些子任务也可以做到这一点。

我现在遇到的问题是，我要等到所有任务都完成（即所有任务都已完成并且它们没有提交新任务）后再继续。

我无法ExecutorService.shutdown()在主线程中调用，因为这会阻止接受新任务ExecutorService。

ExecutorService.awaitTermination()如果shutdown没有被呼叫，呼叫似乎无能为力。

所以我有点卡在这里。ExecutorService看到所有工人都闲着不难，不是吗？我能想到的唯一优雅的解决方案是直接使用a
ThreadPoolExecutor并getPoolSize()偶尔查询一次。真的没有更好的方法吗？

Answer 1

如果最初不清楚递归任务树中的任务数量，则最简单的方法可能是实现您自己的同步原语（某种“反信号量”），并在您的任务之间共享它。在提交每个任务之前，您需要增加一个值，当任务完成时，它会减少该值，然后等待直到该值为0。

将其实现为从任务中显式调用的单独原语，可以将此逻辑与线程池实现分离开来，并允许您将多个独立的递归任务树提交到同一池中。

像这样：

public class InverseSemaphore {
    private int value = 0;
    private Object lock = new Object();

    public void beforeSubmit() {
        synchronized(lock) {
            value++;
        }
    }

    public void taskCompleted() {
        synchronized(lock) {
            value--;
            if (value == 0) lock.notifyAll();
        }
    }

    public void awaitCompletion() throws InterruptedException {
        synchronized(lock) {
            while (value > 0) lock.wait();
        }
    }
}

请注意，taskCompleted()应在finally块内调用它，以使其不受可能的异常影响。

还要注意，beforeSubmit()应在任务提交之前由提交线程调用，而不是由任务本身调用，以避免在完成旧任务而尚未启动新任务时可能出现的“错误完成”。

编辑： 使用模式已修复的重要问题。