Spring Batch中失败后如何重试关联的两个步骤？

方案一：使用Job Decider 实现步骤组重试 #

首先明确：StepListener 无法实现这个需求，它仅用于监听步骤的执行前后、异常等生命周期事件，没有权限修改作业的执行流程。而Job Decider 是Spring Batch 中专门控制作业流程分支的组件，完全可以实现你要的逻辑。

具体实现思路：

1. 在主流程中通过Decider 判断step4的执行状态
2. 自定义Decider，当step4执行失败且未达到最大重试次数时，返回跳转至step3的结果；成功则继续后续流程；超过重试次数则标记失败
3. 必须设置重试次数上限，避免无限循环

示例代码大致如下：

// 自定义Decider
public class Step4RetryDecider implements JobExecutionDecider {
    private static final int MAX_RETRY = 3; // 最大重试次数

    @Override
    public FlowExecutionStatus decide(JobExecution jobExecution, StepExecution stepExecution) {
        // 从Job上下文获取已重试次数，避免作业重启后计数重置
        int retryCount = jobExecution.getExecutionContext().getInt("step4RetryCount", 0);
        
        if (stepExecution.getStatus() == BatchStatus.FAILED && retryCount < MAX_RETRY) {
            retryCount++;
            jobExecution.getExecutionContext().putInt("step4RetryCount", retryCount);
            return new FlowExecutionStatus("RETRY");
        } else if (stepExecution.getStatus() == BatchStatus.COMPLETED) {
            return new FlowExecutionStatus("CONTINUE");
        } else {
            return FlowExecutionStatus.FAILED;
        }
    }
}

// 作业定义调整
jobBuilders.get("WorkFlow")
        .flow(stepsA.step1())
        .next(stepsA.step2())
        .next(stepsB.step3())
        .next(stepsB.step4())
        .next(new Step4RetryDecider())
        .from(new Step4RetryDecider())
            .on("RETRY").to(stepsB.step3())
            .on("CONTINUE").end()
            .on("FAILED").end()
        .end()
        .build();

注意：如果step3本身有独立的重试/跳过配置，需要确保和Decider的重试逻辑不冲突；若需要持久化重试次数，可借助JobExecution的上下文存储，避免作业重启后计数丢失。

方案二：合并步骤并使用RetryTemplate #

把step3和step4的业务逻辑整合到一个Tasklet中，在Tasklet内部用RetryTemplate实现整体重试，这种方案更简单直接。

示例代码大致如下：

// 整合后的Tasklet
public class CombinedStep3And4Tasklet implements Tasklet {
    private final RetryTemplate retryTemplate;
    private final Step3Service step3Service; // step3的业务逻辑实现
    private final Step4Service step4Service; // step4的业务逻辑实现

    // 构造函数注入依赖
    public CombinedStep3And4Tasklet(RetryTemplate retryTemplate, Step3Service step3Service, Step4Service step4Service) {
        this.retryTemplate = retryTemplate;
        this.step3Service = step3Service;
        this.step4Service = step4Service;
    }

    @Override
    public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
        retryTemplate.execute(context -> {
            step3Service.execute(); // 执行原step3逻辑
            step4Service.execute(); // 执行原step4逻辑
            return null;
        });
        return RepeatStatus.FINISHED;
    }
}

// 配置RetryTemplate
@Bean
public RetryTemplate retryTemplate() {
    RetryTemplate retryTemplate = new RetryTemplate();
    // 设置重试策略
    SimpleRetryPolicy retryPolicy = new SimpleRetryPolicy();
    retryPolicy.setMaxAttempts(3);
    retryTemplate.setRetryPolicy(retryPolicy);
    // 设置退避策略（可选）
    FixedBackOffPolicy backOffPolicy = new FixedBackOffPolicy();
    backOffPolicy.setBackOffPeriod(1000); // 重试间隔1秒
    retryTemplate.setBackOffPolicy(backOffPolicy);
    return retryTemplate;
}

// 作业定义调整
jobBuilders.get("WorkFlow")
        .flow(stepsA.step1())
        .next(stepsA.step2())
        .next(stepBuilderFactory.get("combinedStep3And4")
                .tasklet(combinedStep3And4Tasklet())
                .build())
        .end()
        .build();

这种方案优点是实现简单，无需调整作业流程结构；缺点是失去了step3和step4的独立监控能力，无法在Spring Batch监控界面查看两个步骤各自的执行状态与统计数据。

选型建议 #

• 如果需要保留step3和step4的独立步骤监控、或两个步骤间存在其他流程依赖，优先选方案一（Job Decider）
• 如果step3和step4是强关联的原子操作，不需要独立监控，优先选方案二（合并步骤+RetryTemplate），实现成本更低

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