[JPA] BatchSize 적용 시 Entity가 1개인 경우 In 절에 ID가 중복되어 설정되는 문제 (Hibernate 6.x 버전 이상)

 

 

 

이 문제 해결하려고 거의 5시간동안 디버깅 해서 원인 찾아냈는데, 알고보니까 1월 26일인가(불과 한달 전) 해결되어 있더라구요..

다만 아직 스프링부트 jpa dependency 버전에 반영이 안되어서 오류가 발생했다는 것을 보고 그냥 슬퍼하는 중입니다.

 

아마 곧 해결될 것 같은데, 해당 오류는 Hibernate 6.2 이전 버전에서 발생합니다.

 

 

 

 

 

🧐 문제 상황 재현

아래와 같이 1대 다 연관관계를 가진 One과 Many를 정의하였습니다.

@Entity
public class One {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @OneToMany(fetch = FetchType.LAZY, orphanRemoval = true, cascade = CascadeType.ALL, mappedBy = "one")
    private List<Many> manys = new ArrayList<>();

    public Long id() {
        return id;
    }

    public List<Many> manys() {
        return manys;
    }

    public void saveMany() {
        manys.add(new Many(this));
    }
}

@Entity
public class Many {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "one_id")
    private One one;

    public Many() {
    }

    public Many(final One one) {
        this.one = one;
    }

    public Many(final One one, final String d) {
        this.one = one;
    }

    public Long id() {
        return id;
    }

    public One one() {
        return one;
    }
}

 

 

 

 

 

application.yml 파일은 다음과 같습니다.

logging:
  level:
    root: info
    org:
      hibernate:
        orm.jdbc.bind: trace

spring:
  jpa:
    show-sql: true
    properties:
      hibernate:
        format_sql: true
        default_batch_fetch_size: 30

    hibernate:
      ddl-auto: create

 

 

 

 

 

그리고 다음과 같이 코드를 작성해 주겠습니다.

@SpringBootTest
@Transactional
class ManyTest {

    @Autowired
    private EntityManager em;

    @Autowired
    private OneRepository oneRepository;

    @Test
    void ddd() {
        One one1 = new One();

        one1.saveMany();
        oneRepository.save(one1);
        em.flush();
        em.clear();

        System.out.println("========================");
        one1 = oneRepository.findById(one1.id()).get();
        System.out.println("Many 크기: " + one1.manys().size());
    }
}

 

위 코드의 결과는 다음과 같습니다.

 

In 절에 BatchSize 만큼의 중복된 ID가 바인딩되는 문제가 발생했습니다.

 

이와 관련된 문제를 찾기 위해 구글링과 chatGPT까지 사용해 보았지만 결국 해답을 찾을 수 없어서 디버깅을 시도했습니다.

(디버깅은 왜이리 스킵되는건지... 진짜 찾는데 너무 힘들었어요ㅠㅠ)

 

아직도 어떻게 원인을 찾아냈는지는 모르겠지만, Hibernate 관련해서 breakpoint를 이곳 저곳 다 걸다 보니 결국

CollectionLoaderBatchKey에 도달할 수 있었습니다.

 

 

 

 

 

 

 

 

🧐 CollectionLoaderBatchKey

다음과 같은 기능을 하는 클래스입니다.

A one-time use CollectionLoader for applying a batch fetch

 

 

해당 클래스의 load 메서드는 다음과 같이 구현되어 있습니다.

 

중요한 부분은 주황색 박스로 표시해 두었는데요,

조회할 ID가 1개인 경우 numberOfKeysToLoad가 batchSize만큼 설정되기 때문에 위와 같이 In절에 하나의 Id가 중복되어 발생했던 것입니다.

 

 

 

 

 

이는 해당 커밋에서 해결되었으나, 스프링부트 jpa dependency 버전으로는 아직 적용되지 않았습니다.

https://github.com/hibernate/hibernate-orm/pull/5991

HHH-16043 Hibernate 6.x breaks collection batch fetching by mbladel · Pull Request #5991 · hibernate/hibernate-orm

 

 

 

아마 곧 해결될 것 같으나, 혹시 지금 바로 해결하고 싶으시다면 다음과 같이 의존성을 추가해 주시면 해결 가능합니다.

dependencies {
   implementation 'org.springframework.boot:spring-boot-starter-data-jpa'

   // 추가
   implementation 'org.hibernate.orm:hibernate-core:6.2.0.CR2'
   
   // 이외 의존성은 생략
   ...
}

 

추가 이후에는 다음과 같이 정상 작동합니다.