Executor Service的shutdown/awaitTermination原理、自定义实现及终止机制

一、shutdown()的内部工作机制 #

shutdown()是ExecutorService的温和关闭方法，核心逻辑分为两步：

1. 拒绝新任务：将线程池状态从RUNNING切换到SHUTDOWN，之后再调用execute()提交新任务会直接抛出RejectedExecutionException。
2. 中断空闲线程：遍历所有工作线程，仅中断那些处于空闲状态的线程——也就是正在阻塞等待任务（比如调用BlockingQueue.take()）的线程。正在执行的任务不会被中断，会继续运行直到完成。

以ThreadPoolExecutor为例，内部会调用interruptIdleWorkers()方法，逐个检查工作线程是否在执行任务，仅对未执行任务的线程调用Thread.interrupt()。

二、awaitTermination()的内部工作机制 #

awaitTermination(long timeout, TimeUnit unit)是阻塞等待终止的方法，作用是等待线程池进入TERMINATED状态（所有任务完成、所有工作线程终止），直到超时或当前线程被中断：

1. 状态检查循环：内部会反复调用isTerminated()判断线程池是否已终止。
2. 条件等待：如果未终止，会通过Condition对象进入等待状态，直到以下情况发生：
   • 超时时间到
   • 线程池完成所有任务进入TERMINATED状态（此时会唤醒所有等待的线程）
   • 当前线程被中断（抛出InterruptedException）

注意：这个方法不会主动触发关闭操作，必须先调用shutdown()或shutdownNow()，否则线程池会一直处于RUNNING状态，awaitTermination()会直到超时才返回false。

三、自定义shutdown()和awaitTermination() #

通常通过继承ThreadPoolExecutor来扩展这两个方法，核心是在父类逻辑的基础上添加自定义操作，不要完全重写原有核心逻辑，避免破坏线程池的正常行为。

自定义shutdown()示例 #

public class CustomShutdownExecutor extends ThreadPoolExecutor {
    public CustomShutdownExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue<Runnable> workQueue) {
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue);
    }

    @Override
    public void shutdown() {
        // 自定义前置操作：记录关闭日志
        System.out.printf("Executor shutdown triggered at %d%n", System.currentTimeMillis());
        // 调用父类shutdown()完成核心关闭逻辑
        super.shutdown();
        // 自定义后置操作：通知监控系统
        sendShutdownAlert();
    }

    private void sendShutdownAlert() {
        // 实现监控告警逻辑，比如发送邮件、调用监控API
    }
}

自定义awaitTermination()示例 #

@Override
public boolean awaitTermination(long timeout, TimeUnit unit) throws InterruptedException {
    long startTime = System.nanoTime();
    // 调用父类方法完成等待逻辑
    boolean isTerminated = super.awaitTermination(timeout, unit);
    
    if (!isTerminated) {
        // 超时未终止的自定义处理：记录告警日志
        System.err.println("Executor failed to terminate within specified timeout");
        // 可选：强制触发shutdownNow()
        // this.shutdownNow();
    }
    
    long elapsedMs = TimeUnit.NANOSECONDS.toMillis(System.nanoTime() - startTime);
    System.out.printf("Await termination took %d ms%n", elapsedMs);
    return isTerminated;
}

四、Executor框架如何实现线程终止（替代Thread.stop()） #

Thread.stop()被废弃的核心原因是它会强制终止线程并释放所有持有的锁，导致共享数据处于不一致状态，引发难以排查的并发问题。Executor框架采用协作式中断的方式实现优雅终止，核心依赖Thread.interrupt()，但需要任务主动配合：

核心实现逻辑 #

1. 中断信号传递：
   • 调用shutdown()时，仅中断空闲线程（阻塞在任务队列的线程），这类线程会因BlockingQueue.take()抛出InterruptedException而退出循环，最终终止。
   • 调用shutdownNow()时，会中断所有工作线程（包括正在执行任务的线程），同时返回队列中未执行的任务。
2. 任务主动响应中断：
   任务代码需要定期检查中断状态，或者处理InterruptedException，才能优雅终止。示例：

   public class CooperativeTask implements Runnable {
       @Override
       public void run() {
           try {
               // 循环执行任务，每次迭代检查中断状态
               while (!Thread.currentThread().isInterrupted()) {
                   doWork();
               }
           } catch (InterruptedException e) {
               // 捕获中断异常后，重置中断状态（供上层逻辑处理）
               Thread.currentThread().interrupt();
               // 清理资源
               cleanUp();
           }
       }
   
       private void doWork() throws InterruptedException {
           // 调用阻塞方法时会响应中断抛出异常
           TimeUnit.SECONDS.sleep(1);
       }
   
       private void cleanUp() {
           // 释放资源、保存状态
       }
   }
   

3. 线程池状态管理：
   ThreadPoolExecutor通过状态机（RUNNING → SHUTDOWN → STOP → TIDYING → TERMINATED）跟踪关闭过程，只有当所有工作线程终止、任务队列清空后，才会进入TERMINATED状态，完成终止。

内容的提问来源于stack exchange，提问作者Nisha Verma