Spring Batch 5中如何提升节流限制？替代已弃用的throttleLimit

在Spring Batch 5之前，我们常用.throttleLimit(10)将默认节流限制从4提升至10，配合自定义线程池充分利用多线程能力，旧版实现如下：

旧版（Spring Batch 5之前）实现 #

自定义线程池配置 #

@Bean
public ThreadPoolTaskExecutor jobTaskExecutor() {
  ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
  taskExecutor.setCorePoolSize(10);
  taskExecutor.setMaxPoolSize(10);
  taskExecutor.initialize(); // Spring Boot 2.x+ 建议显式调用初始化，避免潜在问题
  return taskExecutor;
}

步骤配置 #

@Bean
public Step sampleStep(JobRepository jobRepository, PlatformTransactionManager transactionManager) {
  return new StepBuilder("sampleStep", jobRepository)
            .<String, String>chunk(100, transactionManager)
            .reader(itemReader())
            .writer(itemWriter())
            .taskExecutor(jobTaskExecutor())
            .throttleLimit(10)
            .build();
 }

Spring Batch 5 替代方案（使用 RepeatOperations） #

在Spring Batch 5中，throttleLimit已被弃用，官方推荐通过RepeatOperations接口的实现类RepeatTemplate控制并发执行限制，具体实现如下：

配置 RepeatTemplate #

@Bean
public RepeatOperations repeatOperations(ThreadPoolTaskExecutor jobTaskExecutor) {
  RepeatTemplate repeatTemplate = new RepeatTemplate();
  // 设置并发执行任务数限制，对应原 throttleLimit 的值
  repeatTemplate.setConcurrencyLimit(10);
  // 指定用于并发执行的线程池
  repeatTemplate.setTaskExecutor(jobTaskExecutor);
  return repeatTemplate;
}

步骤中使用 RepeatOperations #

@Bean
public Step sampleStep(JobRepository jobRepository, PlatformTransactionManager transactionManager, RepeatOperations repeatOperations) {
  return new StepBuilder("sampleStep", jobRepository)
            .<String, String>chunk(100, transactionManager)
            .reader(itemReader())
            .writer(itemWriter())
            .repeatOperations(repeatOperations)
            .build();
 }

说明 #

• RepeatTemplate的concurrencyLimit参数对应原throttleLimit的作用，用于控制同时执行的任务数量
• 线程池配置逻辑和旧版保持一致，确保线程资源合理分配
• 若无需自定义线程池，可直接使用默认任务执行器，但建议根据业务需求调整线程池参数

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