Spring Boot 并行环境下避免 EntityManager 引起的 IllegalStateException

2024-03-03 13:27:32

在并行环境中安全使用实体管理器：避免 IllegalStateException

在多线程环境下使用 Spring Boot 应用程序时，你可能会遇到 IllegalStateException: Illegal pop() with non-matching JdbcValuesSourceProcessingState 异常。这种异常的出现通常是由于以下两个问题：

EntityManager 的线程安全问题： EntityManager 通常不是线程安全的，在多个线程中同时使用同一个 EntityManager 实例可能会导致数据不一致和异常。
CriteriaQuery 的延迟加载问题： CriteriaQuery 中的关联实体默认情况下是延迟加载的，这意味着它们不会在查询执行时立即加载，而是在访问它们时才加载。在并行调用中，如果多个线程尝试同时访问相同的关联实体，可能会出现线程安全问题。

解决方法

有几种方法可以解决这个异常：

1. 使用 JpaRepository

JpaRepository 是 Spring Data 提供的线程安全的存储库实现。它使用 EntityManagerFactory 来创建 EntityManager 实例，该实例是线程安全的，可以安全地用于并行调用。

2. 禁用延迟加载

禁用延迟加载可以强制在查询执行时立即加载关联实体。这可以通过在查询中使用 fetch join 语句来实现。

3. 使用 @Transactional 注解

在 webservice 方法上使用 @Transactional 注解可以确保以事务方式执行该方法。这将确保所有对数据库的操作都是原子性的，并且可以在并行调用中安全地执行。

4. 同步对实体的访问

如果禁用延迟加载或使用 JpaRepository 无法解决问题，则可以考虑使用同步机制来同步对实体的访问。这可以通过使用 synchronized 或 ReentrantLock 等锁机制来实现。

示例代码

使用 JpaRepository：

@Repository
public interface ParentRepository extends JpaRepository<Parent, Long> {}

@Transactional(readOnly = true, propagation = Propagation.REQUIRES_NEW)
public ResponseEntity<List<Parent>> getAllParents() {
    return ResponseEntity.ok(parentRepository.findAll());
}