Spring Batch中CompositeItemWriter下@BeforeStep方法未触发的解决方法

问题核心原因：Spring Batch的CompositeItemWriter不会自动将其内部委托的ItemWriter注册为步骤监听器，因此带有@BeforeStep注解的方法无法被容器识别并触发。

解决方案：手动将委托Writer注册为Step监听器 #

在构建Step时，从CompositeItemWriter中提取所有委托的ItemWriter，筛选出实现了StepListener接口的实例（@BeforeStep注解本质是Spring Batch对StepListener的注解式支持），将它们手动添加到Step的监听器列表中。

修改chunkStep方法如下：

import org.springframework.batch.core.listener.StepListener;
import java.util.stream.Collectors;

public Step chunkStep(Map<String, Object> jobParametersMap) {
    JobContext jobContext = mapper.convertValue(jobParametersMap.get(BATCH_JOB_CONTEXT), JobContext.class);
    int chunkSize = jobContext.getFileProcessingConfig().getChunkSize();
    
    // 获取构建好的CompositeItemWriter
    ItemWriter<?> compositeWriter = writer(jobParametersMap);
    // 提取Composite内的所有委托Writer
    List<ItemWriter<?>> delegateWriters = ((CompositeItemWriter<?>) compositeWriter).getDelegates();
    // 筛选出所有StepListener类型的实例
    List<StepListener> stepListeners = delegateWriters.stream()
            .filter(StepListener.class::isInstance)
            .map(StepListener.class::cast)
            .collect(Collectors.toList());
    
    // 构建Step并添加所有监听器
    StepBuilder stepBuilder = this.stepBuilderFactory.get("fileProcessingStep")
            .chunk(chunkSize)
            .reader(reader(jobParametersMap))
            .processor(processor(jobParametersMap))
            .writer(compositeWriter);
    
    // 批量添加监听器
    stepListeners.forEach(stepBuilder::listener);
    
    return stepBuilder.build();
}

补充说明 #

• 该方案可批量处理所有动态创建的委托Writer，无需单独针对某个自定义Writer做配置，适配你的动态加载场景。
• 完成配置后，@BeforeStep注解的方法会在Step启动前正常触发，可顺利获取ExecutionContext。

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