Spring Batch多作业实例并行运行的线程安全解决方案咨询

你的判断是正确的，当前实现存在线程安全风险。

问题原因 #

默认情况下，Spring Batch的组件（如你的ItemWriter实现类）是单例Bean。你在ItemWriter中用成员变量stepExecution保存当前Step的执行上下文，当多个Job实例并行运行时，不同线程会修改这个共享的成员变量，导致数据串用，出现不可预期的错误。

而你配置的ExecutionContextPromotionListener本身是线程安全的——每个Job实例拥有独立的JobExecution和StepExecution，它们的ExecutionContext是互相隔离的，Listener只是将指定Key从Step上下文提升到Job上下文，不会跨Job实例共享数据。

解决方案 #

1. 将ItemWriter改为Step作用域 #

给ItemWriter添加@StepScope注解，让Spring为每个Step执行创建一个独立的ItemWriter实例，这样每个实例的stepExecution成员变量只会对应当前Step的执行，不会被其他Job实例干扰：

@StepScope
@Component
public class InvoiceItemWriter implements ItemWriter<Invoice> {
    private StepExecution stepExecution;

    @BeforeStep
    public void saveStepExecution(StepExecution stepExecution) {
        this.stepExecution = stepExecution;
    }

    @Override
    public void write(List<? extends Invoice> items) throws Exception {
        // 发票写入逻辑
    }
}

2. 避免持有StepExecution成员变量 #

如果不需要在ItemWriter的整个生命周期保留StepExecution，可以在需要的方法中直接获取上下文，比如通过ChunkContext：

@Override
public void write(List<? extends Invoice> items) throws Exception {
    // 从ChunkContext中获取StepExecution
    StepExecution stepExecution = ChunkContextHolder.getContext().getStepContext().getStepExecution();
    // 后续逻辑
}

3. 依赖ExecutionContext传递数据 #

继续使用ExecutionContextPromotionListener将需要共享的数据提升到JobExecutionContext，后续Step可以通过JobExecution获取这些数据，无需持有StepExecution引用：

// 在Step中获取JobExecutionContext的数据
@BeforeStep
public void setup(StepExecution stepExecution) {
    JobExecution jobExecution = stepExecution.getJobExecution();
    Invoice invoice = (Invoice) jobExecution.getExecutionContext().get("invoice");
    // 使用invoice数据
}

关键说明 #

Spring Batch中每个Job实例的执行是完全隔离的，各自拥有独立的JobExecution和StepExecution上下文。只要确保组件不持有跨Job实例共享的状态，并行运行就不会有线程安全问题。

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