Spring Data JPA如何为不同Repository配置不同的batch size？

可以实现，Hibernate支持在单个Session维度动态覆盖全局批处理大小配置，修改仅对当前Session关联的操作生效，不会影响全局默认配置，完全可以满足不同Repository使用不同批处理大小的需求，具体实现方案如下：

该方案可以自定义批处理逻辑，同时明确指定每个Repository的批处理大小：

1. 为需要单独配置批处理的Repository定义自定义扩展接口

// ItemRepository的扩展接口
public interface CustomItemRepository {
    <S extends Item> List<S> batchSave(Iterable<S> entities);
}

// CarRepository的扩展接口
public interface CustomCarRepository {
    <S extends Car> List<S> batchSave(Iterable<S> entities);
}

2. 原有Repository继承对应的扩展接口

@Repository
public interface ItemRepository extends JpaRepository<Item, String>, CustomItemRepository {
   Page<Item> findItemByCreationDateBefore(Timestamp creationDate, Pageable pageable);
}

@Repository
public interface CarRepository extends JpaRepository<Car, String>, CustomCarRepository {
   @Query("SELECT c FROM Car c WHERE c.journeyCode = :journeyCode " +
          "AND c.currentLocationId = :currentLocationId")
   Page<Delivery> findCars(Long journeyCode, Long currentLocationId, Pageable pageable);
}

3. 实现扩展接口，单独配置对应Repository的批处理大小

// ItemRepository的实现类，指定批处理大小为20
@Repository
public class CustomItemRepositoryImpl implements CustomItemRepository {
    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public <S extends Item> List<S> batchSave(Iterable<S> entities) {
        int batchSize = 20;
        Session session = entityManager.unwrap(Session.class);
        // 覆盖全局批处理配置，仅对当前Session生效
        session.setJdbcBatchSize(batchSize);
        List<S> result = new ArrayList<>();
        int count = 0;
        for (S entity : entities) {
            entityManager.persist(entity);
            result.add(entity);
            count++;
            if (count % batchSize == 0) {
                entityManager.flush();
                entityManager.clear();
            }
        }
        entityManager.flush();
        entityManager.clear();
        return result;
    }
}

// CarRepository的实现类，指定批处理大小为10
@Repository
public class CustomCarRepositoryImpl implements CustomCarRepository {
    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public <S extends Car> List<S> batchSave(Iterable<S> entities) {
        int batchSize = 10;
        Session session = entityManager.unwrap(Session.class);
        session.setJdbcBatchSize(batchSize);
        List<S> result = new ArrayList<>();
        int count = 0;
        for (S entity : entities) {
            entityManager.persist(entity);
            result.add(entity);
            count++;
            if (count % batchSize == 0) {
                entityManager.flush();
                entityManager.clear();
            }
        }
        entityManager.flush();
        entityManager.clear();
        return result;
    }
}

4. 业务代码直接调用对应batchSave方法即可使用对应批处理大小。

如果不想修改原有Repository结构，可以通过切面拦截对应Repository的方法，自动设置批处理大小：

@Aspect
@Component
public class RepositoryBatchConfigAspect {
    @PersistenceContext
    private EntityManager entityManager;

    // 拦截ItemRepository的所有方法，自动设置批处理大小为20
    @Before("execution(* 你的项目Repository包路径.ItemRepository.*(..))")
    public void configItemBatchSize() {
        Session session = entityManager.unwrap(Session.class);
        session.setJdbcBatchSize(20);
    }

    // 拦截CarRepository的所有方法，自动设置批处理大小为10
    @Before("execution(* 你的项目Repository包路径.CarRepository.*(..))")
    public void configCarBatchSize() {
        Session session = entityManager.unwrap(Session.class);
        session.setJdbcBatchSize(10);
    }
}

• 动态设置的批处理大小仅对当前Session生效，不同Repository的配置互不干扰
• 原有全局的order_inserts、order_updates配置需要保持开启，才能保证批处理正常生效
• 如果需要自定义更新、删除操作的批处理大小，也可以参照上述逻辑实现

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