Spring Batch+Spring Integration SFTP上传完成后进程无法自动停止

我来帮你分析下这个问题——文件成功上传但进程停不下来，核心原因大概率是线程池里的非守护线程没有被正确终止，或者SFTP相关资源没有释放，导致JVM无法退出。下面给你几个针对性的解决方案：

1. 修正ThreadPoolTaskExecutor的配置 #

默认情况下，ThreadPoolTaskExecutor的核心线程不会自动超时关闭，Job结束后核心线程会一直处于空闲状态，阻止JVM退出。你需要添加几个关键配置：

@Bean
public ThreadPoolTaskExecutor sftpUploadTaskExecutor() {
    ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
    executor.setCorePoolSize(5); // 根据业务需求调整核心线程数
    executor.setMaxPoolSize(10); // 最大线程数
    executor.setQueueCapacity(20); // 任务队列容量
    executor.setThreadNamePrefix("sftp-upload-worker-"); // 线程名称前缀，方便排查问题
    
    // 关键配置：允许核心线程空闲超时关闭
    executor.setAllowCoreThreadTimeOut(true);
    // 关闭时等待所有任务完成再终止线程池
    executor.setWaitForTasksToCompleteOnShutdown(true);
    // 设置等待终止的超时时间（单位：秒）
    executor.setAwaitTerminationSeconds(30);
    
    executor.initialize();
    return executor;
}

2. 完善JobExecutionListener的线程池关闭逻辑 #

你已经尝试了JobExecutionListener，但可能关闭线程池的逻辑不够完整。在afterJob方法里，需要先优雅关闭线程池，超时后再强制终止：

@Bean
public JobExecutionListener sftpJobExecutionListener(ThreadPoolTaskExecutor sftpUploadTaskExecutor) {
    return new JobExecutionListener() {
        @Override
        public void beforeJob(JobExecution jobExecution) {
            // 可选：Job启动前的初始化操作
        }

        @Override
        public void afterJob(JobExecution jobExecution) {
            // 先尝试优雅关闭线程池
            sftpUploadTaskExecutor.shutdown();
            
            try {
                // 等待指定时间让线程池终止
                if (!sftpUploadTaskExecutor.awaitTermination(60, TimeUnit.SECONDS)) {
                    // 超时后强制关闭
                    sftpUploadTaskExecutor.shutdownNow();
                    // 再次等待确认终止
                    if (!sftpUploadTaskExecutor.awaitTermination(60, TimeUnit.SECONDS)) {
                        System.err.println("SFTP上传线程池无法正常终止，可能存在未释放的资源");
                    }
                }
            } catch (InterruptedException e) {
                // 被中断时强制关闭线程池
                sftpUploadTaskExecutor.shutdownNow();
                Thread.currentThread().interrupt();
            }
        }
    };
}

别忘了把这个Listener注册到你的Job上：

@Bean
public Job sftpUploadJob(JobBuilderFactory jobBuilderFactory, Step sftpUploadStep, JobExecutionListener listener) {
    return jobBuilderFactory.get("sftpUploadJob")
            .listener(listener)
            .flow(sftpUploadStep)
            .end()
            .build();
}

3. 确保SFTP会话资源被正确释放 #

如果你的上传逻辑是手动获取SFTP会话，一定要确保会话被关闭，否则会持有底层连接和线程。推荐使用try-with-resources语法自动释放资源：

@Autowired
private SftpSessionFactory sftpSessionFactory;

public void uploadFile(File file, String remotePath) throws Exception {
    // try-with-resources会自动关闭session，无需手动调用close()
    try (Session<ChannelSftp.LsEntry> session = sftpSessionFactory.getSession()) {
        // 执行文件上传操作
        session.write(new FileInputStream(file), remotePath + "/" + file.getName());
    } catch (Exception e) {
        // 异常处理逻辑
        throw new RuntimeException("文件上传失败: " + file.getName(), e);
    }
}

4. 检查Spring Batch Job的结束状态 #

确保你的Job确实进入了COMPLETED或FAILED状态，没有处于挂起的STARTED状态。可以在afterJob方法里打印状态，确认Job是否正常结束：

@Override
public void afterJob(JobExecution jobExecution) {
    System.out.println("Job结束状态: " + jobExecution.getStatus());
    // 后续的线程池关闭逻辑...
}

如果Job状态异常，需要排查Step的执行逻辑，看是否有未处理的异常导致Job无法正常结束。

5. Spring Boot环境的额外配置 #

如果是Spring Boot应用，还可以通过配置文件直接设置线程池的关闭行为，避免硬编码：

在application.properties中添加：

# 线程池关闭时等待任务完成
spring.task.execution.shutdown.await-termination=true
# 等待终止的超时时间
spring.task.execution.shutdown.await-termination-period=30s

按照上面的步骤调整后，应该能解决进程无法终止的问题。核心思路就是确保线程池被正确关闭，且所有SFTP相关资源都被释放，让JVM没有活跃的非守护线程，从而正常退出。

内容的提问来源于stack exchange，提问作者Karthick A.S