当 ThreadPoolExecutor 没有活动工作线程时执行操作

我有一个缓存的线程池，其中以相当不可预测的方式生成新任务。这些任务不会生成任何结果（它们是Runnables而不是Callables）。

我希望在池没有活动辅助角色时执行一个操作。

但是我不想关闭池（显然使用 awaitTermination），因为当新任务到达时，我必须再次重新初始化它（因为它可能会不可预测地到达，即使在关闭期间也是如此）。

我想出了以下可能的方法：

有一个额外的线程（池外），每当生成新任务并且ThreadPoolExecutor没有活动工作线程时，就会生成该线程。然后，它应该不断检查getActiveWorkers()，直到它返回 0，如果是，则执行所需的操作。
有一些线程安全队列（哪一个？），其中添加了每个新生成任务的Future。每当队列中至少有一个条目时，生成一个额外的线程（池外部），该线程会等到队列为空并执行所需的操作。
实现要与池一起使用的PriorityBlockingQueue，并为工作线程分配比执行所需操作的线程（现在从池内部）更高的优先级。

我的问题：我想知道是否有一些更干净的解决方案，它使用一些不错的同步对象（如 CountDownLatch ，但不能在这里使用，因为我事先不知道任务的数量）？

如果我是你，我会为你的线程池实现一个装饰器，它跟踪计划的任务，并修改运行的任务。这样，无论何时安排Runnable，您都可以安排另一个能够跟踪其自身进程的去装饰Runnable。

这个装饰器看起来像这样：

class RunnableDecorator implements Runnable {
  private final Runnable delegate;
  // this task counter must be increased on any 
  // scheduling of a task by the thread pool
  private final AtomicInteger taskCounter;
  // Constructor omitted
  @Override
  public void run() {
    try {
      delegate.run();
    } finally {
      if (taskCounter.decrementAndGet() == 0) {
        // spawn idle action
      }
    }
  }
}

当然，线程池必须在每次计划任务时递增计数器。因此，不得将此操作的逻辑添加到Runnable中，而应添加到ThreadPool中。最后，由您决定是否要在同一线程中运行空闲操作，或者是否要提供对正在执行的线程池的引用以运行新线程。但是，如果决定使用后者，请注意，空闲操作的完成将触发另一个空闲操作。但是，您也可以为某种原始计划提供一种方法。您还可以将修饰添加到线程队列中，但是这使得提供这种原始调度变得更加困难。

这种方法是非阻塞的，不会过多地扰乱您的代码库。请注意，胎面池在创建时不会启动操作，因此根据定义是空的。

如果你看一下Executors.newCachedThreadPool()背后的源代码，你可以看到它是如何用ThreadPoolExecutor创建的。使用它，重写execute和afterExecute方法来添加计数器。这样，递增和递减逻辑就隔离在一个位置。前任：

ExecutorService executor = new ThreadPoolExecutor(0, Integer.MAX_VALUE, 60L, TimeUnit.SECONDS,
        new SynchronousQueue<Runnable>()) {
    private AtomicInteger counter = new AtomicInteger(0);
    @Override
    public void execute(Runnable r) {
        counter.incrementAndGet();
        super.execute(r);
    }
    @Override
    public void afterExecute(Runnable r, Throwable t) {
        if (counter.decrementAndGet() == 0) {
            // thread pool is idle - do something
        }
        super.afterExecute(r, t);
    }
};

相关内容

最新更新

热门标签：