Java:为ThreadPool中的线程设置超时

时间:2013-01-09 13:47:45

标签: java multithreading timeout

我想为线程池中执行的线程设置超时。目前我有以下代码:

ExecutorService executor = Executors.newFixedThreadPool(8);
for(List<String> l: partition) {            
    Runnable worker = new WorkerThread(l);
    executor.execute(worker);
}       

executor.shutdown();
while (!executor.isTerminated()) {

}

代码只是将一大堆对象拆分为子列表,并在单个线程中处理这些子列表。但这不是重点。

我想给线程池中的每个单线程一个超时。对于池中只有一个线程,我找到了以下解决方案:

Future<?> future = null;

for (List<String> l : partition) {
    Runnable worker = new WorkerThread(l);
    future = executor.submit(worker);
}

try {
    System.out.println("Started..");
    System.out.println(future.get(3, TimeUnit.SECONDS));
    System.out.println("Finished!");
} catch (TimeoutException e) {
    System.out.println("Terminated!");
}

但这不适用于多个线程。也许我必须将每个线程放在List<Future>列表中并迭代这个列表并为每个future对象设置超时?

有什么建议吗?

使用CountDownLatch后编辑:

CountDownLatch doneSignal = new CountDownLatch(partition.size());
List<Future<?>> tasks = new ArrayList<Future<?>>();
ExecutorService executor = Executors.newFixedThreadPool(8);
for (List<String> l : partition) {
    Runnable worker = new WorkerThread(l);
    tasks.add(executor.submit(doneSignal, worker));
}

doneSignal.await(1, TimeUnit.SECONDS);
if (doneSignal.getCount() > 0) {
    for (Future<?> fut : tasks) {
    if (!fut.isDone()) {
        System.out.println("Task " + fut + " has not finshed!");
        //fut.cancel(true) Maybe we can interrupt a thread this way?!
    }
    }
}

到目前为止工作得很好。

接下来的问题是如何中断超时的线程?我尝试fut.cancel(true)并在工作线程的一些关键循环中添加以下构造:

if(Thread.interrupted()) {
    System.out.println("!!Thread -> " + Thread.currentThread().getName() + " INTERRUPTED!!");
        return;
}

因此工作线程在超时后被“杀死”。这是一个很好的解决方案吗?

此外:是否可以获取在Future界面上超时的线程的名称?目前,我必须在Thread.interrupted()构造的if条件下打印出名称。

感谢您的帮助!

此致

1 个答案:

答案 0 :(得分:3)

你看到了这个吗? ExecutorService.invokeAll

它应该是您想要的:调用一组工作人员,如果花费太长时间,他们会超时。

评论后编辑 - (新想法): 您可以使用CountDownLatch等待任务完成并通过await(long timeout, TimeUnit unit)超时! 然后你甚至可以执行shutdownNow,看看哪些任务花了太长时间......

编辑2:

更清楚:

  1. 完成后,每个工作人员都要对CountDownLatch进行倒计时。
  2. 在主执行线程await中,所述锁存器超时。
  3. 当该呼叫返回时,您可以检查锁存器的计数以查看是否存在超时命中(如果它是> 0)。
  4. a)count = 0,所有任务及时完成。 b)如果没有,循环期货并检查他们的isDone。您不必在ExecutorService上调用shutdown。
  5. 如果您不再需要Executor,请致电关机。
  6. 注意:工作人员可以在超时和调用Future的isDone()之间完成。