Spring Batch：如何在Java配置中延迟设置MultiResourcePartitioner的资源

我之前在将XML配置的Spring Batch多文件上传转为Java配置时，也遇到过完全一样的问题。核心原因是XML配置中默认的bean作用域和初始化时机，与Java配置的默认行为不同：XML里的MultiResourcePartitioner通常绑定到Step作用域，在Step运行时才初始化，这时候Job参数已经可用；而Java配置中如果不指定作用域，bean会在Spring上下文启动时就被创建，这时候Job参数还没传入，自然会导致资源为Null。

下面是几个经过验证的解决方案：

方案一：使用@StepScope延迟初始化Partitioner #

这是最直接有效的方法，通过@StepScope注解让MultiResourcePartitioner在Step运行时才初始化，此时Job参数已经可以正常获取：

import org.springframework.batch.core.partition.support.MultiResourcePartitioner;
import org.springframework.batch.core.scope.StepScope;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

@Bean
@StepScope
public MultiResourcePartitioner partitioner(@Value("#{jobParameters['inputFilePath']}") String inputFilePath) throws IOException {
    MultiResourcePartitioner partitioner = new MultiResourcePartitioner();
    // 根据传入的文件路径模式解析资源
    Resource[] resources = new PathMatchingResourcePatternResolver().getResources(inputFilePath);
    partitioner.setResources(resources);
    // 设置分区数量（可选，默认会根据资源数量自动处理）
    partitioner.setPartitionsNumber(resources.length);
    return partitioner;
}

这里的inputFilePath是你在启动Job时传入的参数（比如file:/path/to/files/*.csv），通过SpEL表达式#{jobParameters['inputFilePath']}在Step运行时动态注入，完美实现了延迟加载资源的效果，和XML配置的行为一致。

方案二：自定义Partitioner实现延迟资源注入 #

如果需要更灵活的控制，可以自定义一个Partitioner，实现InitializingBean来延迟获取资源：

import org.springframework.batch.core.partition.support.MultiResourcePartitioner;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;

public class LazyMultiResourcePartitioner extends MultiResourcePartitioner implements InitializingBean {

    @Value("#{jobParameters['inputFilePath']}")
    private String inputFilePath;

    @Override
    public void afterPropertiesSet() throws Exception {
        // 在Bean初始化完成后（此时Job参数已可用）解析资源
        Resource[] resources = new PathMatchingResourcePatternResolver().getResources(inputFilePath);
        super.setResources(resources);
    }
}

然后在Java配置中注册这个自定义Partitioner，并指定@StepScope：

@Bean
@StepScope
public LazyMultiResourcePartitioner partitioner() {
    return new LazyMultiResourcePartitioner();
}

方案三：通过Step监听动态设置资源 #

如果你的资源路径需要从JobExecution中动态获取，也可以在Step的配置中通过监听方法设置资源：

@Bean
public Step partitionedStep(StepBuilderFactory stepBuilderFactory, ItemReader<YourItem> reader, 
                            ItemWriter<YourItem> writer, MultiResourcePartitioner partitioner) {
    return stepBuilderFactory.get("partitionedStep")
            .partitioner("slaveStep", partitioner)
            .step(slaveStep())
            .taskExecutor(taskExecutor())
            .listener(new StepExecutionListener() {
                @Override
                public void beforeStep(StepExecution stepExecution) {
                    JobExecution jobExecution = stepExecution.getJobExecution();
                    String inputFilePath = jobExecution.getJobParameters().getString("inputFilePath");
                    try {
                        Resource[] resources = new PathMatchingResourcePatternResolver().getResources(inputFilePath);
                        partitioner.setResources(resources);
                    } catch (IOException e) {
                        throw new RuntimeException("Failed to resolve resources", e);
                    }
                }

                @Override
                public ExitStatus afterStep(StepExecution stepExecution) {
                    return null;
                }
            })
            .build();
}

这种方式在Step执行前的监听方法中设置资源，确保资源是在运行时才被解析和注入的。

总结一下，最推荐的是方案一，它最简洁且贴合Spring Batch的设计理念，通过@StepScope完美模拟了XML配置中的延迟初始化行为，解决资源为Null的问题。

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