Spring Batch并行块处理中如何避免嵌套懒加载集合的LazyInitializationException

问题背景 #

使用Spring Batch开发并行块处理作业时，通过JpaPagingItemReader读取数据、JpaTransactionManager管理事务、SimpleAsyncTaskExecutor实现并行处理，但在ItemProcessor中访问JPA实体的嵌套懒加载集合时，触发LazyInitializationException：

org.hibernate.LazyInitializationException: failed to lazily initialize a collection of role: com.myapp.Order.types: could not initialize proxy - no Session

环境配置：

• Spring Boot: 3.3.4
• Spring Batch: 5.1.2
• 事务管理器: JpaTransactionManager
• 任务执行器: SimpleAsyncTaskExecutor

已尝试JOIN FETCH抓取，但因Order实体包含多个嵌套集合，会导致查询数据量过大、效率低下，无法使用；原本预期JpaTransactionManager能管理事务范围，但问题依旧。

原因分析 #

并行块处理时，JpaPagingItemReader在主线程的事务中读取实体，读取完成后事务关闭，实体变为脱管状态。这些脱管实体被传递到异步线程的ItemProcessor中，此时原EntityManager的Session已关闭，无法初始化懒加载集合，从而抛出异常。

可行解决方案 #

方案一：读取阶段在事务内初始化懒加载集合 #

在主线程读取实体的事务范围内，提前初始化需要访问的懒加载集合，避免后续异步线程中访问时无Session可用。推荐使用JPA实体图（Entity Graph），相比JOIN FETCH，它能精准控制加载的关联集合，避免笛卡尔积问题。

步骤1：定义实体图

在Order实体上定义命名实体图，指定需要加载的集合：

@Entity
@NamedEntityGraph(name = "Order.withTypes", attributeNodes = @NamedAttributeNode("types"))
public class Order {
    @Id
    private Long id;

    @OneToMany(mappedBy = "order", cascade = CascadeType.ALL, orphanRemoval = true)
    private List<Type> types;

    // 其他字段...
}

步骤2：修改Reader使用实体图

在JpaPagingItemReader中配置实体图查询提示，确保读取时加载指定集合：

@Bean
public JpaPagingItemReader<Customer> customerItemReader(EntityManagerFactory entityManagerFactory) {
    // 获取定义的实体图
    EntityGraph<?> orderWithTypesGraph = entityManagerFactory.createEntityManager()
            .getEntityGraph("Order.withTypes");

    return new JpaPagingItemReaderBuilder<Customer>()
            .name("customerItemReader")
            .entityManagerFactory(entityManagerFactory)
            .queryString("SELECT c FROM Customer c JOIN c.order o")
            // 添加实体图查询提示
            .queryHint("javax.persistence.fetchgraph", orderWithTypesGraph)
            .pageSize(100)
            .build();
}

方案二：Processor中重新加载实体 #

在异步线程的ItemProcessor中，通过EntityManager重新加载实体，确保在当前线程的事务内操作托管实体，从而正常初始化懒加载集合。

修改ItemProcessor

@Component
public class CustomerItemProcessor implements ItemProcessor<Customer, Customer> {

    @PersistenceContext
    private EntityManager entityManager;

    @Override
    public Customer process(Customer customer) throws Exception {
        // 使用实体图重新加载Customer，确保关联集合被初始化
        EntityGraph<?> orderWithTypesGraph = entityManager.getEntityGraph("Order.withTypes");
        Map<String, Object> hints = Map.of("javax.persistence.fetchgraph", orderWithTypesGraph);
        Customer managedCustomer = entityManager.find(Customer.class, customer.getId(), hints);

        // 复制实体（避免修改原托管实体）
        Customer newCustomer = new Customer();
        Order order = new Order();
        order.setTypes(new ArrayList<>(managedCustomer.getOrder().getTypes()));
        newCustomer.setOrder(order);
        return newCustomer;
    }
}

方案三：手动初始化集合（备选） #

如果无法使用实体图，可在Reader读取实体后，手动调用Hibernate.initialize()初始化集合，确保在Session关闭前完成：

// 扩展JpaPagingItemReader，在读取后初始化集合
@Bean
public JpaPagingItemReader<Customer> customerItemReader(EntityManagerFactory entityManagerFactory) {
    JpaPagingItemReader<Customer> reader = new JpaPagingItemReaderBuilder<Customer>()
            .name("customerItemReader")
            .entityManagerFactory(entityManagerFactory)
            .queryString("SELECT c FROM Customer c")
            .pageSize(100)
            .build();

    // 读取后初始化集合
    reader.setPostPageCallback(items -> {
        items.forEach(item -> {
            Hibernate.initialize(item.getOrder().getTypes());
            // 若有其他集合，同理初始化
        });
        return items;
    });

    return reader;
}

注意事项 #

• 避免使用Open Session in View（OSIV）：批处理场景下OSIV会导致Session长时间打开，引发性能和资源占用问题。
• 并行处理时，确保每个Chunk的事务独立：Spring Batch默认会为每个异步Chunk创建独立事务，无需额外配置。

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