Spring Batch远程分区：如何动态用Worker步骤名而非Bean名称

可以实现需求，核心是通过**自定义StepLocator**替换Spring Batch默认的BeanFactoryStepLocator，让分区处理器能够通过步骤的逻辑名称（即stepBuilder.get("stepName")定义的名称）而非Bean名称查找对应的Worker Step。

问题根源 #

Spring Batch默认的TaskExecutorPartitionHandler使用BeanFactoryStepLocator，它直接从Spring容器中按Bean名称查找Step实例。如果你的Worker Step的Bean名称（@Bean(name="xxx")）和步骤逻辑名称（stepBuilder.get("yyy")）不一致，就会触发"Bean未找到"错误。

解决方案：自定义Step名称定位器 #

实现一个基于步骤名称映射的StepLocator，将容器中所有Step的逻辑名称与Bean实例关联，这样就能通过配置文件中的步骤名称找到对应的Worker Step。

1. 自定义StepLocator实现类

import org.springframework.batch.core.Step;
import org.springframework.batch.core.configuration.StepLocator;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import java.util.HashMap;
import java.util.Map;

public class StepNameBasedLocator implements StepLocator, ApplicationContextAware {
    private final Map<String, Step> stepNameToBeanMap = new HashMap<>();

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        // 扫描容器中所有Step类型的Bean，建立"步骤名称→Bean实例"的映射
        Map<String, Step> allStepBeans = applicationContext.getBeansOfType(Step.class);
        allStepBeans.forEach((beanName, step) -> stepNameToBeanMap.put(step.getName(), step));
    }

    @Override
    public Step getStep(String stepName) throws IllegalArgumentException {
        Step targetStep = stepNameToBeanMap.get(stepName);
        if (targetStep == null) {
            throw new IllegalArgumentException("不存在名称为[" + stepName + "]的Worker Step");
        }
        return targetStep;
    }

    @Override
    public Map<String, Step> getSteps() {
        return stepNameToBeanMap;
    }
}

2. 配置分区处理器与自定义Locator

在批处理配置类中，将自定义的StepNameBasedLocator注入到分区处理器，并从配置文件读取Worker步骤名称：

@Configuration
public class PartitionBatchConfig {

    // 从配置文件读取Worker步骤名称
    @Value("${batch.worker.step.name}")
    private String workerStepName;

    @Bean
    public StepNameBasedLocator stepNameBasedLocator() {
        return new StepNameBasedLocator();
    }

    @Bean
    public PartitionHandler taskExecutorPartitionHandler(TaskExecutor taskExecutor, StepNameBasedLocator stepLocator) {
        TaskExecutorPartitionHandler handler = new TaskExecutorPartitionHandler();
        handler.setTaskExecutor(taskExecutor);
        // 替换默认的StepLocator为自定义实现
        handler.setStepLocator(stepLocator);
        // 设置从配置文件读取的Worker步骤名称
        handler.setStepName(workerStepName);
        handler.setGridSize(5); // 分区数量，可配置化
        return handler;
    }

    // Worker Step：Bean名称与步骤名称不一致的示例
    @Bean(name = "userExportWorkerBean")
    public Step userExportWorkerStep(StepBuilderFactory stepBuilderFactory) {
        // 步骤逻辑名称为"userExportWorkerStep"，与Bean名称"userExportWorkerBean"不同
        return stepBuilderFactory.get("userExportWorkerStep")
                .tasklet((contribution, chunkContext) -> {
                    // 业务逻辑实现
                    return RepeatStatus.FINISHED;
                })
                .build();
    }

    @Bean
    public Step masterStep(StepBuilderFactory stepBuilderFactory, PartitionHandler partitionHandler) {
        return stepBuilderFactory.get("masterStep")
                .partitioner(workerStepName, new SimplePartitioner())
                .partitionHandler(partitionHandler)
                .build();
    }
}

注意事项 #

• 确保所有Worker Step的逻辑名称唯一，避免映射时出现覆盖问题。
• 若存在多个Worker Step，配置文件中的batch.worker.step.name需对应正确的步骤逻辑名称。
• 自定义Locator会在Spring容器初始化时扫描所有Step Bean，需保证Worker Step都被正确注册到容器中。

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