Spring Data JPA - 새로운 Entity 판별

신규 Entity 판단 방식

Spring Data JPA 는 내부적으로 JpaEntityInformation 의 isNew(T entity) 메서드를 호출해서 판단한다.

@Override
public boolean isNew(T entity) {
    if (versionAttribute.isEmpty()
        || versionAttribute.map(Attribute::getJavaType).map(Class::isPrimitive).orElse(false)) {
        return super.isNew(entity);
    }
    BeanWrapper wrapper = new DirectFieldAccessFallbackBeanWrapper(entity);
    return versionAttribute.map(it -> wrapper.getPropertyValue(it.getName()) == null).orElse(true);
}

• @Version 필드가 있다면 → 해당 버전 필드 값이 null 인지 여부로 판단
• @Version 필드가 없다면 → @Id 필드가 null 이거나, primitive 타입일 경우, 0인지 여부로 판단

직접 ID 를 지정한 경우의 동작

ID를 직접 지정하고 save() 를 호출하면, JPA 는 해당 Entity 가 기존에 존재한다고 판단하여 merge()를 수행한다.
이때 실제 Database 에 해당 ID가 존재하지 않으면, SELECT 쿼리는 0건을 반환하고
JPA 는 병합용 엔티티를 새로 생성하여 INSERT 를 시도하거나, 제약 조건 위반으로 예외가 발생할 수 있다.
결국, 신규 Entity 임에도 JPA 는 기존 데이터처럼 인식하게 되어, 의도치 않은 UPDATE 또는 INSERT 실패로 이어질 수 있다

이를 해결 하기 위한 방법으로는 Persistable<T> 인터페이스를 구현하면 된다.

// User.class

@Entity
@Table(name = "users")
@EntityListeners(UserEntityListener.class)
public class User implements Persistable<String> {

	@Id
	private String id;

	private String name;

	private boolean isNew = true;

	protected User() {

	}

	public User(String id, String name) {
		this.id = id;
		this.name = name;
	}

	@Override
	public String getId() {
		return this.id;
	}

	public void setId(String id) {
		this.id = id;
	}

	public String getName() {
		return this.name;
	}

	@Override
	public boolean isNew() {
		return this.isNew;
	}

	public void setIsNew(boolean isNew) {
		this.isNew = isNew;
	}

}

// UserEntityListener.class

public class UserEntityListener {
	@PostPersist
	@PostLoad
	public void setNotNew(User user) {
		System.out.println("@PostPersist/@PostLoad called");
		user.setIsNew(false);
	}

}

persist() vs merge()

신규 Entity 판단이 중요한 이유는 무엇일까?

Spring Data JPA 의 SimpleJpaRepository 는 save() 에서 다음과 같이 동작한다.

@Transactional
public <S extends T> S save(S entity) {
    if (entityInformation.isNew(entity)) {
        entityManager.persist(entity); // INSERT
        return entity;
    } else {
        return entityManager.merge(entity); // SELECT → UPDATE, 존재하면 UPDATE 없으면 INSERT 가능성 있음
    }
}

ID 를 직접 설정했지만, isNew() 는 false 가 되어, merge() 를 호출하게 되고,
Database 에는 해당 ID 가 존재하지 않지만, 신규 Entity 임에도 불구하고, SELECT 후 UPDATE 를 하게 되어 (이 때, database 조회) 실패 또는 데이터 무결성 오류가 발생할 수 있고, 비효율적이다.

요약

📚 Reference

• Spring Docs - JpaEntityInformation
• Spring Docs - entity-persistence