Spring Batch配置TaskExecutor后Step重复执行5次的问题

环境 #

Spring Boot 3.2.1、Spring Batch 5.1.2、Java 17

问题描述 #

为Tasklet类型的Step配置自定义ThreadPoolTaskExecutor后，该Step被执行5次，但预期仅执行一次。移除TaskExecutor后执行正常，但需要保留自定义线程命名。

相关配置 #

Step配置 #

@Bean("dynamicStep0")
public Step dynamicStep0() {
    return new StepBuilder("step1Tasklet", jobRepository)
            .tasklet(new Step0Tasklet(), transactionManager)
            .taskExecutor(dynamicBatchStepExecutor())
            .build();
}

TaskExecutor配置 #

@Bean("dynamicBatchStepTaskExecutor")
public ThreadPoolTaskExecutor dynamicBatchStepExecutor() {
    ThreadPoolTaskExecutor tpe = new ThreadPoolTaskExecutor();
    tpe.setCorePoolSize(1);
    tpe.setMaxPoolSize(1);
    tpe.setThreadNamePrefix("DynamicSteps-JOB-EXEC-");
    tpe.initialize();
    return tpe;
}

Tasklet实现 #

@RequiredArgsConstructor
@Slf4j
@StepScope
public class Step0Tasklet implements Tasklet {
    @Override
    public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws Exception {
        log.info("Step0Tasklet executed, this is a dummy step");
        return RepeatStatus.FINISHED;
    }
}

已尝试的无效方案 #

• 为Step添加已废弃的throttleLimit(1)；
• 设置TaskExecutor的queueCapacity为1；
• 移除TaskExecutor后执行正常，但需保留自定义线程命名。

核心问题 #

1. 为何corePoolSize和maxPoolSize设为1时，Step仍重复执行5次？
2. 如何保留自定义线程命名同时确保Step仅执行一次？

问题解答 #

1. 重复执行的原因 #

Spring Batch中，当为Tasklet类型的Step配置TaskExecutor时，默认会触发分区执行模式——此时Step会被当作分区Step处理，默认分区数为5（框架内置默认值）。即使线程池大小设为1，也会依次执行5个分区任务，导致Tasklet被调用5次。

这里的关键误区是：Tasklet Step配置TaskExecutor并非单纯用线程池执行单次任务，而是框架默认启用了分区机制，每个分区都会调用一次Tasklet。

2. 解决方案（保留自定义线程命名+单次执行） #

方案一：用自定义线程池包装Tasklet执行（推荐）

直接在Tasklet逻辑内部使用自定义线程池执行任务，避免触发Spring Batch的分区机制：

@Bean("dynamicStep0")
public Step dynamicStep0() {
    return new StepBuilder("step1Tasklet", jobRepository)
            .tasklet((contribution, chunkContext) -> {
                // 用自定义线程池执行Tasklet逻辑
                return dynamicBatchStepExecutor().submit(() -> {
                    try {
                        return new Step0Tasklet().execute(contribution, chunkContext);
                    } catch (Exception e) {
                        throw new RuntimeException(e);
                    }
                }).get();
            }, transactionManager)
            .build();
}

方案二：替换为SimpleAsyncTaskExecutor

SimpleAsyncTaskExecutor支持自定义线程名前缀，且配置到Step时不会触发分区机制（框架对该执行器的处理逻辑不同）：

@Bean("dynamicBatchStepTaskExecutor")
public SimpleAsyncTaskExecutor dynamicBatchStepExecutor() {
    SimpleAsyncTaskExecutor executor = new SimpleAsyncTaskExecutor();
    executor.setThreadNamePrefix("DynamicSteps-JOB-EXEC-");
    return executor;
}

Step配置保持原有代码即可，此时Step仅执行一次，同时使用自定义命名的线程。

方案三：显式配置单分区

若必须使用ThreadPoolTaskExecutor，可显式设置Step为单分区：

@Bean("dynamicStep0")
public Step dynamicStep0() {
    return new StepBuilder("step1Tasklet", jobRepository)
            .partitioner("step1Tasklet", (gridSize) -> {
                Map<String, ExecutionContext> partitions = new HashMap<>();
                // 仅创建一个分区
                partitions.put("partition1", new ExecutionContext());
                return partitions;
            })
            .step(singleTaskletStep())
            .taskExecutor(dynamicBatchStepExecutor())
            .build();
}

// 定义仅执行一次的Tasklet Step
@Bean
public Step singleTaskletStep() {
    return new StepBuilder("singleTaskletStep", jobRepository)
            .tasklet(new Step0Tasklet(), transactionManager)
            .build();
}

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