Spring Batch作业重启时遭遇NoSuchJobException问题咨询

我来帮你搞定这个Spring Batch的JobInstance和重启逻辑问题！下面是具体的实现思路和代码示例，完全贴合你的需求：

1. 核心逻辑梳理 #

• 不带--restart参数：生成唯一的JobParameters（比如加一个时间戳参数），确保每次运行都创建新的JobInstance。
• 带--restart参数：查询目标Job的最近一次处于STARTED/FAILED/STOPPING状态的JobExecution，如果存在就复用该执行的参数重启；如果没有未完成的执行，还是创建新的JobInstance。

2. CommandLineRunner的具体实现 #

我们需要注入JobLauncher（启动作业）、JobRegistry（获取所有注册的作业）、JobExplorer（查询历史执行记录），然后解析命令行参数处理两种场景：

import org.springframework.batch.core.*;
import org.springframework.batch.core.explore.JobExplorer;
import org.springframework.batch.core.launch.JobLauncher;
import org.springframework.batch.core.launch.NoSuchJobException;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;

@Component
public class BatchJobRunner implements CommandLineRunner {

    private final JobLauncher jobLauncher;
    private final JobRegistry jobRegistry;
    private final JobExplorer jobExplorer;

    // 构造注入依赖
    public BatchJobRunner(JobLauncher jobLauncher, JobRegistry jobRegistry, JobExplorer jobExplorer) {
        this.jobLauncher = jobLauncher;
        this.jobRegistry = jobRegistry;
        this.jobExplorer = jobExplorer;
    }

    @Override
    public void run(String... args) throws Exception {
        // 判断是否传入了--restart参数
        boolean isRestart = false;
        for (String arg : args) {
            if ("--restart".equals(arg)) {
                isRestart = true;
                break;
            }
        }

        // 遍历JobConfig中注册的所有作业
        for (String jobName : jobRegistry.getJobNames()) {
            Job job = jobRegistry.getJob(jobName);
            JobParameters jobParameters;

            if (isRestart) {
                // 查询最近一次未完成的JobExecution
                Optional<JobExecution> lastUnfinishedExecution = findLastUnfinishedJobExecution(job);
                if (lastUnfinishedExecution.isPresent()) {
                    JobExecution execution = lastUnfinishedExecution.get();
                    System.out.printf("重启未完成的作业执行：ID=%d，作业名=%s%n", execution.getId(), jobName);
                    jobParameters = execution.getJobParameters();
                } else {
                    // 没有未完成的执行，创建新的JobInstance
                    jobParameters = generateUniqueJobParameters();
                    System.out.printf("未找到未完成的作业执行，为%s创建新实例%n", jobName);
                }
            } else {
                // 不带restart参数，生成唯一参数创建新JobInstance
                jobParameters = generateUniqueJobParameters();
            }

            // 启动作业
            jobLauncher.run(job, jobParameters);
        }
    }

    // 查询指定作业的最近一次未完成执行记录
    private Optional<JobExecution> findLastUnfinishedJobExecution(Job job) {
        return jobExplorer.getJobExecutions(job).stream()
                .filter(execution -> execution.getStatus().isUnfinished())
                .sorted((e1, e2) -> e2.getStartTime().compareTo(e1.getStartTime()))
                .findFirst();
    }

    // 生成唯一的JobParameters（用当前时间戳保证唯一性）
    private JobParameters generateUniqueJobParameters() {
        Map<String, JobParameter<?>> parameters = new HashMap<>();
        String timestamp = LocalDateTime.now().format(DateTimeFormatter.ISO_LOCAL_DATE_TIME);
        parameters.put("runTimestamp", new JobParameter<>(timestamp, String.class));
        return new JobParameters(parameters);
    }
}

3. 关键注意事项 #

• JobParameters唯一性：Spring Batch是通过Job+JobParameters的组合来标识JobInstance的，所以每次不带重启参数时，必须加入唯一标识（比如时间戳），确保每次参数不同，从而生成新的实例。
• JobExplorer的作用：它是Spring Batch提供的用来查询历史执行记录的组件，我们用它来判断是否存在未完成的作业执行，这是实现重启逻辑的核心。
• 作业状态判断：execution.getStatus().isUnfinished()会自动匹配所有未完成的状态（包括运行中、失败、正在停止），完全符合我们需要重启的场景。
• JobRegistry适配：如果你的JobConfig是通过其他方式收集作业的（比如手动注入List），只需要把遍历jobRegistry.getJobNames()的逻辑换成遍历你的作业列表即可。

4. 测试验证 #

• 首次无参数运行：查看Spring Batch的数据库表（比如BATCH_JOB_INSTANCE），会生成新的实例记录，作业执行状态为COMPLETED（成功）或FAILED（失败）。
• 作业运行中/失败后带--restart运行：会复用之前的JobInstance，找到对应的未完成JobExecution继续执行，不会生成新的实例。

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