Spring Batch：后续步骤满足条件时如何无限回退至前序步骤

嘿，这个需求很实际！针对你的场景（表头步骤+百万级表体步骤，需要在表体步骤满足条件时回退到表头步骤），Spring Batch其实可以通过作业流条件跳转+步骤重执行允许的组合来实现。下面给你详细拆解可行的方案和注意事项：

核心思路 #

Spring Batch的作业流支持基于步骤执行状态（ExitStatus）的条件跳转，再配合允许步骤重复执行的配置，就能实现“满足条件就回退到前序步骤”的逻辑。关键是要：

1. 让步骤允许重复执行（默认已完成的步骤不会再启动）
2. 在后续步骤中根据业务条件设置自定义的ExitStatus
3. 配置作业流，让特定ExitStatus触发回退跳转

具体实现步骤 #

1. 配置步骤允许重复执行 #

首先给你的stepHeader和stepBody都加上allowStartIfComplete(true)，这样即便步骤已经标记为COMPLETED，也能被重新启动：

@Bean
public Step stepHeader(JobRepository jobRepository, PlatformTransactionManager txManager) {
    return new StepBuilder("stepHeader", jobRepository)
            .<HeaderInput, FormattedHeader>chunk(1, txManager)
            .reader(headerFlatFileReader())
            .processor(headerFormatter())
            .writer(headerTxtWriter())
            .allowStartIfComplete(true) // 允许已完成的步骤重新执行
            .build();
}

@Bean
public Step stepBody(JobRepository jobRepository, PlatformTransactionManager txManager) {
    return new StepBuilder("stepBody", jobRepository)
            .<BodyRecord, FormattedBodyRecord>chunk(1000, txManager) // 百万级数据建议用大chunk提升性能
            .reader(bodyFlatFileReader())
            .processor(bodyFormatter())
            .writer(bodyTxtWriter())
            .allowStartIfComplete(true)
            .build();
}

2. 在表体步骤中设置条件ExitStatus #

在stepBody的Processor或Writer里，根据你的业务判断条件，设置自定义的ExitStatus（比如NEED_RETRY_HEADER）。这里以Processor为例：

@Component
public class BodyFormatter implements ItemProcessor<BodyRecord, FormattedBodyRecord> {

    @Override
    public FormattedBodyRecord process(BodyRecord item) throws Exception {
        // 你的业务格式化逻辑
        FormattedBodyRecord formattedRecord = formatBodyItem(item);

        // 判断是否需要回退到表头步骤（比如表头参数需要重新生成/校验）
        if (shouldRetryHeader()) {
            // 获取当前StepExecution，设置自定义ExitStatus
            StepExecution stepExecution = StepSynchronizationManager.getContext().getStepExecution();
            stepExecution.setExitStatus(new ExitStatus("NEED_RETRY_HEADER"));
        }

        return formattedRecord;
    }

    // 这里替换成你的实际业务判断逻辑
    private boolean shouldRetryHeader() {
        // 比如：检查表头参数是否过期、是否需要重新拉取最新配置等
        return someBusinessCondition;
    }
}

3. 配置作业流的条件跳转 #

最后在构建Job的时候，通过on()和to()方法配置跳转规则：当stepBody返回NEED_RETRY_HEADER时，跳回stepHeader；其他状态则结束作业：

@Bean
public Job flatFileProcessingJob(JobRepository jobRepository, Step stepHeader, Step stepBody) {
    return new JobBuilder("flatFileProcessingJob", jobRepository)
            .start(stepHeader)
            .next(stepBody)
            .on("NEED_RETRY_HEADER").to(stepHeader) // 触发回退到表头步骤
            .from(stepBody).on("*").end() // 其他所有状态都结束作业
            .build();
}

关键注意事项 #

• 设置终止条件：一定要给“无限回退”加上终止逻辑！比如在JobExecution的ExecutionContext中记录重试次数，达到上限就不再设置NEED_RETRY_HEADER状态，避免作业无限循环：

  // 在shouldRetryHeader()中加入计数判断
  private boolean shouldRetryHeader() {
      JobExecution jobExecution = StepSynchronizationManager.getContext().getStepExecution().getJobExecution();
      ExecutionContext jobContext = jobExecution.getExecutionContext();
      int retryCount = jobContext.getInt("headerRetryCount", 0);
      if (retryCount >= 5) { // 最多重试5次
          return false;
      }
      jobContext.putInt("headerRetryCount", retryCount + 1);
      // 你的业务判断
      return someBusinessCondition;
  }
  

• Reader重置问题：如果stepBody的Reader是FlatFileItemReader，重新执行时默认会从头读取。如果不需要从头读，你需要自定义Reader的定位逻辑（比如记录上次读取的位置，下次从该位置继续），避免重复处理百万级数据导致性能问题。
• 事务边界：确保你的步骤事务配置合理，尤其是回退时的资源清理（比如已写入的文件内容是否需要回滚），避免数据不一致。

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