优雅关闭ExecutorService的正确方式

文章小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《优雅关闭ExecutorService的正确方法》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

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

理论要掌握，实操不能落！以上关于《优雅关闭ExecutorService的正确方式》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！