Spring Batch中afterStep异常传递给配置类异常处理器的方案咨询

方案1（推荐）：模板方法模式改造抽象Tasklet，适配原生异常处理流程 #

这个方案完全绕开afterStep()不能抛出异常的限制，符合Spring Batch原生异常处理逻辑，原有配置无需修改，无冗余代码：

• 定义抽象Tasklet基类，不直接开放execute()方法给子类重写，新增doExecute()抽象方法供子类实现业务逻辑
• 基类的execute()方法固定执行逻辑：先调用子类的doExecute()执行业务，再执行统一校验逻辑，校验抛出的异常直接从execute()往外抛，自动触发Step上配置的logExceptionHandler()

示例代码：

// 抽象基类，统一封装执行流程和校验逻辑
public abstract class AbstractTaskletTemplate implements Tasklet {
    // 子类实现业务逻辑
    protected abstract RepeatStatus doExecute(StepContribution contribution, ChunkContext chunkContext) throws Exception;

    // 统一校验逻辑，所有子类复用，可直接抛出异常
    protected void commonValidate() throws Exception {
        // 你的公共校验逻辑
        throwable_function();
    }

    @Override
    public final RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
        RepeatStatus result = doExecute(contribution, chunkContext);
        // 业务执行完成后执行校验
        commonValidate();
        return result;
    }
}

// 子类业务实现，仅需关注自身业务逻辑
public class ExampleTasklet extends AbstractTaskletTemplate {
    @Override
    protected RepeatStatus doExecute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
        // 原有业务逻辑
        return RepeatStatus.FINISHED;
    }
}

原有Step配置无需做任何修改，logExceptionHandler()会自动处理校验逻辑抛出的所有异常。

方案2：保留afterStep()回调的实现方案 #

如果你必须使用StepExecutionListener的afterStep()回调，可以按以下方式实现：

• 将logExceptionHandler()声明为公共Bean，在抽象Tasklet基类中统一注入，无需子类重复配置
• 在afterStep()中捕获校验逻辑抛出的异常，手动调用logExceptionHandler()处理，同时标记步骤执行失败

示例代码：

// 配置类中声明异常处理器为公共Bean
@Bean
public ExceptionHandler logExceptionHandler() {
    // 原有异常处理逻辑
}

// 抽象Tasklet基类
public abstract class AbstractListenerTasklet implements Tasklet, StepExecutionListener {
    @Autowired
    private ExceptionHandler logExceptionHandler;

    // 子类实现业务逻辑
    @Override
    public abstract RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception;

    // 统一校验逻辑
    protected abstract void commonValidate() throws Exception;

    @Override
    public ExitStatus afterStep(StepExecution stepExecution) {
        try {
            commonValidate();
        } catch (Exception e) {
            // 手动调用公共异常处理器
            try {
                logExceptionHandler.handleException(e);
            } catch (Exception ex) {
                // 无需额外处理，异常处理器已完成日志等预设逻辑
            }
            // 标记步骤执行失败，同步异常信息到执行上下文
            stepExecution.addFailureException(e);
            return ExitStatus.FAILED;
        }
        return stepExecution.getExitStatus();
    }
}

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