java 重用执行器

Question

我研究一个模拟系统，在每个时间步，我必须模拟许多模型。我使用了固定线程池来加速计算：

ExecutorService executor = Executors.newFixedThreadPool(nThread);
for (Model m : models) {
  executor.execute( m.simulationTask() );
}
executor.shutdown();
while ( ! executor.awaitTermination(10, TimeUnit.MINUTES) ) { 
  System.out.println("wait"); 
}

现在，调用 shutdown() 后，执行器不能用于 execute() 新任务。有没有办法重置执行器，以便我可以在下一个模拟步骤中重用现有的执行器（及其线程）？

Answer 1

如果您稍微重构一下代码，您可以重用执行程序服务。

Collection<Callable<Integer>> tasks = new ArrayList<Callable<Integer>>(16);
for (Model m : models) {
  tasks.add(m.simulationTask());
}

ExecutorService executor = Executors.newFixedThreadPool(nThread);
try {
  executor.invokeAll(tasks);
} catch(InterruptedException ie) {
  // Handle this
}

基本上，您收集所有任务，执行它们，并在继续之前等待执行。当然，您也可以为每个时间步骤使用新的执行器服务，但至少您有选择。

注意事项：我没有编译代码，因此可能会出现错误。为了方便起见，我还假设了 Integer 参数类型。

Answer 2

您可以编写自己的 Executor 接口实现。除此之外，我知道的大多数默认实现都会在 shutdown() 之后进行内存清理，因此没有（据我所知）预先制定的解决方案。

考虑到 shutdown() 可能会进行大量清理和垃圾收集，目前尚不清楚为什么重新启动会比获取新的 Executor 更好，也许您应该看看进入教程 关于使用暂停/恢复方法集扩展 ThreadPoolExecutor，而不是添加取消关闭的功能。

Answer 3

将您的 ExecutorService 声明为成员您的课程并根据需要重用它。不要对其调用 shutdown()，因为它将不再接受任何任务。当然，你的任务应该很好地结束，并且它们也应该在某个时刻终止。

Answer 4

只需获取另一个 ExecutorService 即可。无论如何，开销是最小的。

如果你坚持重用同一个执行器，你可以实现自己的屏障机制。提交新任务后，自动递增计数器。当任务完成时，自动递减计数器。在主线程中等待，直到计数器为零。类似于：

// globally visible objects
AtomicInteger counter = new AtomicInteger(0);
Object signal = new Object();

ExecutorService executor = Executors.newFixedThreadPool(nThread);
for (Model m : models) {
  counter.getAndIncrement();
  executor.execute( m.simulationTask() );
}

synchronized(signal) {
   while(count.get() > 0) {
       signal.wait();
   }
}

然后在任务的 run 中：

public void run() {
    // original code
    // at the end:
    synchronized(signal) {
       counter.getAndDecrement();
       signal.notify();
    }        
}