本文旨在解决在使用ExecutorService时，如何在超时或其他特定情况下停止正在运行的线程，同时避免完全关闭ExecutorService，以便后续任务仍然可以提交。我们将探讨使用shutdownNow()方法结合Thread.interrupted()标志的正确方式，以及如何在Runnable任务中优雅地处理中断请求。
在使用ExecutorService处理并发任务时，有时需要在特定条件下（例如超时）停止正在执行的任务。直接粗暴地终止线程是不推荐的，因为它可能导致资源泄漏或数据不一致。更优雅的方式是请求线程停止，并让线程自身处理停止逻辑。
使用 shutdownNow() 请求停止线程
ExecutorService 提供了 shutdownNow() 方法，该方法尝试停止所有正在执行的任务，暂停处理正在等待的任务，并返回等待执行的任务列表。重要的是，shutdownNow() 并不会立即杀死线程，而是设置每个线程的 interrupted 标志。
ExecutorService executor = Executors.newFixedThreadPool(10);
List<List<Object>> objectGroup = Lists.partition(objectList, 5);
for (List<Object> eachGroup : objectGroup ) {
CountDownLatch latch = new CountDownLatch(eachGroup.size());
for (Object obj: eachGroup) {
executor.submit(() -> {
try {
doTask(obj);
} catch (InterruptedException e) {
// 处理中断异常，例如清理资源
System.out.println("Task interrupted!");
} finally {
latch.countDown();
}
});
}
try {
if (!latch.await(15, TimeUnit.MINUTES)) {
// 超时，请求停止所有线程
List<Runnable> droppedTasks = executor.shutdownNow();
System.out.println("Timeout! Tasks interrupted: " + droppedTasks.size());
}
}
catch (InterruptedException e) {
System.out.println("Interrupted while waiting for latch!");
executor.shutdownNow();
}
}
// 不要忘记在所有任务完成后关闭 ExecutorService
executor.shutdown();
try {
if (!executor.awaitTermination(60, TimeUnit.SECONDS)) {
System.err.println("Executor did not terminate in 60 seconds");
executor.shutdownNow();
}
} catch (InterruptedException ex) {
executor.shutdownNow();
Thread.currentThread().interrupt();
}在 Runnable 任务中检查中断状态
为了使线程能够响应中断请求，需要在 Runnable 任务中定期检查 Thread.interrupted() 标志。如果该标志为 true，则应该执行清理操作并退出任务。
public void doTask(Object obj) throws InterruptedException {
while (!Thread.interrupted()) {
// 执行任务的一部分
// ...
// 模拟耗时操作
try {
Thread.sleep(100);
} catch (InterruptedException e) {
// 重新设置中断标志，以便在方法栈的更高层也能感知到中断
Thread.currentThread().interrupt();
throw e; // 抛出 InterruptedException
}
}
// 执行清理操作，例如释放资源
System.out.println("Task is stopping gracefully.");
}注意事项:
- Thread.interrupted() 方法会清除当前线程的中断状态。如果需要在方法栈的更高层继续响应中断，需要在捕获 InterruptedException 后重新设置中断标志 Thread.currentThread().interrupt()。
- executor.shutdownNow() 会返回一个List，其中包含尚未开始执行的任务。你可以根据需要处理这些被丢弃的任务。
- 在所有任务完成后，务必调用 executor.shutdown() 来关闭 ExecutorService，并使用 awaitTermination() 方法等待所有线程完成。如果超时，可以再次调用 executor.shutdownNow()。
- shutdownNow() 之后不能再提交新的任务，如果需要继续提交任务，则不应该调用此方法。
总结
通过结合使用 shutdownNow() 方法和在 Runnable 任务中检查 Thread.interrupted() 标志，可以实现优雅地停止 ExecutorService 中线程的目的。 这种方法避免了强制终止线程可能带来的问题，并允许线程自身负责清理资源和退出。 记住，在处理并发编程时，良好的错误处理和资源管理至关重要。
