Spring Batch中如何从父Step向子Step动态传递参数？

针对你遇到的分区父Step（StepLevel1）向Flow子Step（StepLevel2）传递唯一Step1Id的问题，核心思路是利用分区Step的独立ExecutionContext隔离每个父Step实例的标识，避免全局JobExecutionContext的冲突问题。以下是具体实现步骤：

1. 自定义Partitioner生成唯一Step1Id #

在分区器中为每个父Step实例生成唯一Step1Id，并放入对应分区的ExecutionContext中，确保每个分区的子Step能拿到专属标识：

public class Step1Partitioner implements Partitioner {
    @Override
    public Map<String, ExecutionContext> partition(int gridSize) {
        Map<String, ExecutionContext> partitions = new HashMap<>(gridSize);
        for (int i = 0; i < gridSize; i++) {
            ExecutionContext context = new ExecutionContext();
            // 生成唯一Step1Id，可根据业务逻辑替换为自定义规则
            String uniqueStep1Id = "STEP1-" + UUID.randomUUID().toString().substring(0, 8);
            context.putString("step1Id", uniqueStep1Id);
            partitions.put("partition-" + i, context);
        }
        return partitions;
    }
}

2. 修改父Step配置使用自定义Partitioner #

将原有的partitioner替换为上面的自定义实现，确保每个分区实例携带专属Step1Id：

@Bean(name = "StepLevel1")
@Lazy
public Step createStep() {
    return new StepBuilder("step1", jobRepository)
            .partitioner("step1Partition", new Step1Partitioner()) // 替换为自定义分区器
            .step(partitionedFlowStep())
            .gridSize(getGridSize())
            .build();
}

3. 在Flow子Step中获取Step1Id #

无论是Tasklet类型还是Chunk类型的子Step，都可以通过StepExecution获取当前分区的Step1Id：

3.1 Tasklet类型子Step（如doSomething） #

public Step doSomething() {
    return new StepBuilder("doSomething", jobRepository).tasklet((contribution, chunkContext) -> {
        // 从当前StepExecution的上下文获取Step1Id
        StepExecution stepExecution = chunkContext.getStepContext().getStepExecution();
        String step1Id = stepExecution.getExecutionContext().getString("step1Id");
        
        // 业务逻辑：使用step1Id处理任务
        System.out.println("子Step执行，所属父StepId：" + step1Id);
        
        return RepeatStatus.FINISHED;
    }, transactionManager).build();
}

3.2 Chunk类型子Step（如StepLevel2） #

如果是ItemReader/ItemProcessor/ItemWriter结构的Step，可以通过@BeforeStep注解注入StepExecution，进而获取Step1Id：

@Component
public class StepLevel2Reader implements ItemReader<YourData> {
    private StepExecution stepExecution;

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

    @Override
    public YourData read() throws Exception {
        String step1Id = stepExecution.getExecutionContext().getString("step1Id");
        // 根据step1Id读取对应数据
        return fetchDataByStep1Id(step1Id);
    }

    private YourData fetchDataByStep1Id(String step1Id) {
        // 业务实现
        return new YourData(step1Id);
    }
}

问题原因分析 #

• StepExecutionContext无法获取：之前可能错误地在父Step的全局逻辑中存入上下文，而非在Partitioner中为每个分区单独注入；或者子Step获取上下文的方式错误（比如误取JobExecutionContext）。
• JobExecutionContext冲突：JobExecutionContext是全局共享的，多个分区实例同时写入会覆盖彼此的Step1Id，而每个分区的StepExecutionContext是独立隔离的，不会出现冲突。

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