本文旨在解决在使用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 中线程的目的。 这种方法避免了强制终止线程可能带来的问题，并允许线程自身负责清理资源和退出。 记住，在处理并发编程时，良好的错误处理和资源管理至关重要。
评论（已关闭）
评论已关闭