如何在Spring Batch中异步执行SystemCommandTasklet

问题原因 #

• 原生SystemCommandTasklet为同步阻塞实现：内部执行命令后会调用process.waitFor()等待脚本执行完成，无论JobLauncher是否配置异步，执行Tasklet的线程都会被阻塞。
• 为SimpleJobLauncher配置SimpleAsyncTaskExecutor后任务失败无日志，核心原因是异步线程未正确传递Spring Batch作业上下文，导致@StepScope修饰的Bean注入jobParameters失败，异常未被日志捕获所以无报错输出。

解决方案 #

分两种业务场景适配：

场景1：仅需启动脚本，无需追踪脚本最终执行结果 #

直接自定义非阻塞Tasklet替换原生SystemCommandTasklet，启动进程后直接返回，不需要等待脚本执行完成：

@Component
@StepScope
public class NonBlockSystemCommandTasklet implements Tasklet, StepExecutionListener {
    @Value("#{jobParameters['dir']}")
    private String dir;
    @Value("#{jobParameters['command']}")
    private String command;
    private Process process;

    @Override
    public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
        // 启动shell进程
        process = Runtime.getRuntime().exec(command, null, new File(dir));
        // 可选：将进程PID存入作业上下文方便后续追溯
        chunkContext.getStepContext().getStepExecution()
                .getJobExecution().getExecutionContext()
                .put("shellPid", process.pid());
        // 直接返回完成，不阻塞线程
        return RepeatStatus.FINISHED;
    }

    @Override
    public void beforeStep(StepExecution stepExecution) {
        // 可自定义前置执行逻辑
    }

    @Override
    public ExitStatus afterStep(StepExecution stepExecution) {
        // 进程启动成功则标记Step执行成功
        return process.isAlive() ? ExitStatus.COMPLETED : ExitStatus.FAILED;
    }
}

修改Step的Tasklet注入为上述自定义实现即可，无需修改JobLauncher配置。

场景2：需要Spring Batch追踪脚本最终执行状态，同时避免线程阻塞 #

需要配置支持上下文传递的异步任务执行器，绑定到Step级别使用：

1. 配置带上下文传递的异步任务执行器

@Bean
public TaskExecutor asyncStepTaskExecutor() {
    SimpleAsyncTaskExecutor executor = new SimpleAsyncTaskExecutor("async-step-");
    // 配置任务装饰器传递作业上下文
    executor.setTaskDecorator(runnable -> {
        JobExecutionContext context = JobSynchronizationManager.getContext();
        return () -> {
            try {
                JobSynchronizationManager.register(context);
                runnable.run();
            } finally {
                JobSynchronizationManager.release();
            }
        };
    });
    return executor;
}

2. 修改Step配置绑定异步执行器

@Bean
public Step myStep(Tasklet tasklet, TaskExecutor asyncStepTaskExecutor) {
    return this.stepBuilderFactory.get("myStep")
            .listener(tasklet)
            .tasklet(tasklet)
            .taskExecutor(asyncStepTaskExecutor)
            // 可根据业务需求调整并发上限
            .throttleLimit(10)
            .build();
}

日志排查建议 #

如果需要定位异步执行的报错信息，可将日志框架中org.springframework.batch包的日志级别调整为DEBUG，即可输出完整的异常栈信息。

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