等待SchduledExecutorService中的所有作业完成，同时允许添加新作业

Question

我使用ScheduledExecutorService跨多个线程调度和处理作业。在我的应用程序中，作业可以安排新作业（在相同的ScheduledExecutorService上），作为某种后续操作。

在主线程中，我想等到所有作业都完成，作为同步点。有关闭（）和waitTeroration（）方法，但这不允许任何正在运行或挂起的作业安排新作业。在我的情况下，我实际上希望允许这样做，接受我们永远不会完成（或遇到一些超时）的风险。

我如何等待所有工作以及后续工作完成？

Answer 1

可以跟踪活动作业的数量。实际上，您提交的每个作业必须包装如下（伪代码）：

increase active jobs by one
try {
    run actual job
} finally {
    decrease active jobs by one
}

当然，ScheduledExecutorService需要完全封装才能工作。

下一步是找到合适的并发机制，它不会引入任何繁忙的等待。

在发出不再有作业信号的条件下，快速而肮脏地尝试使用锁：

private final Lock lock = new ReentrantLock();
private final Condition noJobs = lock.newCondition();

private long jobCount = 0;

private void increaseJobCount() {
    lock.lock();
    try {
        jobCount++;
    } finally {
        lock.unlock();
    }
}

private void decreaseJobCount() {
    lock.lock();
    try {
        jobCount--;
        if (jobCount == 0) {
            noJobs.signalAll();
        }
    } finally {
        lock.unlock();
    }
}

public void awaitNoJobs() throws InterruptedException {
    lock.lock();
    try {
        while (jobCount > 0) {
            noJobs.await();
        }
    } finally {
        lock.unlock();
    }
}

我检查了一些其他已知的并发类，但除了BlockingQueue之外，我找不到任何类。但这些会占用更多内存，因为您必须为提交的每个作业添加和删除一些内容。CountDownLatch接近，但不允许增加计数。