Spring Batch：如何自动重启失败的多线程定时任务？

一、多线程Step对重启的影响 #

Spring Batch的重启依赖StepExecution状态持久化，如果你的多线程Step的Reader未实现ItemStream接口（未保存读取状态），重启时会重复读取数据：

• 若业务支持幂等处理（重复执行不影响结果），可直接重启；
• 若不允许重复处理，必须给Reader添加状态保存逻辑：实现ItemStream的open/update/close方法，用ExecutionContext存储当前读取位置，重启时从上下文恢复。示例代码：

public class StatefulItemReader implements ItemReader<String>, ItemStream {
    private List<String> data = Arrays.asList("data1", "data2", "data3");
    private int currentIndex = 0;
    private static final String INDEX_KEY = "currentIndex";

    @Override
    public String read() throws Exception {
        return currentIndex < data.size() ? data.get(currentIndex++) : null;
    }

    @Override
    public void open(ExecutionContext executionContext) {
        currentIndex = executionContext.containsKey(INDEX_KEY) ? executionContext.getInt(INDEX_KEY) : 0;
    }

    @Override
    public void update(ExecutionContext executionContext) {
        executionContext.putInt(INDEX_KEY, currentIndex);
    }

    @Override
    public void close() {}
}

二、JobOperator.restart调用位置问题 #

为什么Scheduler中调用无效+@Retryable失效 #

• 定时任务线程与Spring Batch执行上下文隔离，直接在@Scheduled方法中调用jobOperator.restart，无法正确绑定Job执行环境，导致重启逻辑不触发；
• @Retryable基于Spring AOP实现，若在Scheduler线程直接调用Bean方法，未经过AOP代理，注解失效。

正确调用方式 #

方案1：Spring管理的JobExecutionListener触发重启

这是最直接的自动重启方式，但要解决Listener中Autowired Bean为null的问题（见下文）。示例：

@Component
public class FailedJobRestartListener extends JobExecutionListenerSupport {
    @Autowired
    private JobOperator jobOperator;

    @Override
    public void afterJob(JobExecution jobExecution) {
        if (ExitStatus.FAILED.getExitCode().equals(jobExecution.getExitStatus().getExitCode())) {
            int restartCount = jobExecution.getExecutionContext().getInt("restartCount", 0);
            // 限制重启次数，避免死循环
            if (restartCount < 3) {
                jobExecution.getExecutionContext().putInt("restartCount", restartCount + 1);
                try {
                    jobOperator.restart(jobExecution.getId());
                } catch (Exception e) {
                    // 日志记录或异常处理
                }
            }
        }
    }
}

// 在Job配置中注入Listener
@Configuration
public class BatchConfig {
    @Autowired
    private FailedJobRestartListener restartListener;

    @Bean
    public Job myJob(JobBuilderFactory jobFactory, Step step1, Step step2) {
        return jobFactory.get("myJob")
                .listener(restartListener)
                .start(step1)
                .next(step2)
                .on("FAILED").continue() // 你的条件流配置
                .end()
                .build();
    }
}

方案2：独立失败任务监控组件

写一个独立的Spring Bean，定时扫描Batch元数据表，触发符合条件的失败任务重启，与原任务解耦：

@Component
public class FailedJobMonitor {
    @Autowired
    private JobOperator jobOperator;
    @Autowired
    private JobExplorer jobExplorer;

    @Scheduled(fixedRate = 60000) // 每分钟扫描一次
    public void restartFailedJobs() {
        List<JobInstance> instances = jobExplorer.getJobInstances("myJob", 0, 100);
        for (JobInstance instance : instances) {
            for (JobExecution execution : jobExplorer.getJobExecutions(instance)) {
                if (ExitStatus.FAILED.getExitCode().equals(execution.getExitStatus().getExitCode()) && !execution.isRunning()) {
                    int restartCount = execution.getExecutionContext().getInt("restartCount", 0);
                    if (restartCount < 3) {
                        try {
                            jobOperator.restart(execution.getId());
                        } catch (Exception e) {
                            // 异常处理
                        }
                    }
                }
            }
        }
    }
}

三、自定义JobExecutionListener中Autowired Bean为null的解决 #

出现此问题的核心原因是Listener未被Spring容器管理：

• 必须给Listener添加@Component注解，或在Batch配置类中用@Bean声明为Spring Bean；
• 禁止手动new Listener实例，必须通过Spring注入到Job配置中。

四、完整重启实现步骤 #

1. 确保Job可重启：Job的restartable属性默认是true，若手动设置为false需改回；
2. 配置JobOperator：Spring Boot会自动配置，若手动配置需绑定JobExplorer、JobRepository、JobLauncher和JobRegistry；
3. 处理条件流ExitStatus：若最后一步失败导致Job标记为FAILED是业务需要，保持现有配置即可触发重启；若想让Job结果为COMPLETED，可修改条件流为.on("FAILED").end(ExitStatus.COMPLETED)，但此配置不会触发重启；
4. 验证元数据表：确保Spring Batch的元数据表（BATCH_JOB_EXECUTION、BATCH_STEP_EXECUTION等）已正确创建且能正常写入状态。

五、常见错误排查 #

• 只有状态为FAILED或STOPPED的JobExecution才能被重启，COMPLETED状态无法重启；
• 使用@Retryable时，确保注解加在public方法上，且引入了spring-retry依赖；
• 检查数据库连接权限，确保Batch能正常读写元数据。

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