삭제 스케줄링에 대한 성능 개선

@Modifying(clearAutomatically = true, flushAutomatically = true)
@Query(value = "delete from UserHistory where actionDateTime < :localDateTime")
void deleteAllByActionDateTimeBefore(@Param("localDateTime") LocalDateTime localDateTime);

최적화 과정

기존 코드에서 사용하는 deleteAllById는 Iterable<?extends ID> ids를 순회하면서 delete 쿼리를 발생시킵니다.

deleteAllById 내부 코드

@Override
@Transactional
public void deleteAllById(Iterable<? extends ID> ids) {

    Assert.notNull(ids, "Ids must not be null!");

    for (ID id : ids) {
        deleteById(id);
    }
}

즉, ids의 크기가 100이면 100개의 쿼리가 실행되는 것 입니다.

deleteAllById 호출 결과

이처럼 여러번의 삭제 쿼리를 발생시키는 문제를 해결하기 위해서 deleteAllByIdInBatch를 사용해서 문제를 해결해보았습니다.

deleteAllByIdInBatch 내부 코드

@Override
@Transactional
public void deleteAllByIdInBatch(Iterable<ID> ids) {

    Assert.notNull(ids, "Ids must not be null!");

    if (!ids.iterator().hasNext()) {
        return;
    }

    if (entityInformation.hasCompositeId()) {

        List<T> entities = new ArrayList<>();
        // generate entity (proxies) without accessing the database.
        ids.forEach(id -> entities.add(getReferenceById(id)));
        deleteAllInBatch(entities);
    } else {

        String queryString = String.format(DELETE_ALL_QUERY_BY_ID_STRING, entityInformation.getEntityName(),
                entityInformation.getIdAttribute().getName());

        Query query = em.createQuery(queryString);
        /**
         * Some JPA providers require {@code ids} to be a {@link Collection} so we must convert if it's not already.
         */
        if (Collection.class.isInstance(ids)) {
            query.setParameter("ids", ids);
        } else {
            Collection<ID> idsCollection = StreamSupport.stream(ids.spliterator(), false)
                    .collect(Collectors.toCollection(ArrayList::new));
            query.setParameter("ids", idsCollection);
        }

        query.executeUpdate();
    }
}

내부 코드를 살펴보면 복합키가 아닌 경우에는 em.createQuery 메서드를 통해 DELETE_ALL_QUERY_BY_ID_STRING 쿼리가 실행되는 것을 확인할 수 있다.

그리고 QueryUtils클래스의 DELETE_ALL_QUERY_BY_ID_STRING 의 상수 값을 확인해보면 delete from %s where %s in :ids 값으로, in절로 여러개의 데이터를 삭제하는 것을 알 수 있다.

deleteAllByIdInBatch 호출 결과

@Query

하지만, deleteAllByIdInBatch 내부 코드를 살펴보았을 때, createQuery 메서드를 통해 in절을 호출하는 방법이라면, @Query 어노테이션을 사용해서 직접 사용자 지정 쿼리를 작성하는 방법과 동일하다. 아래에서 테스트를 통해 성능적인 측면에서 얼마나 차이나는지 확인해보자.

@Modifying(clearAutomatically = true, flushAutomatically = true)
@Query(value = "delete from UserHistory where actionDateTime < :localDateTime")
void deleteAllByActionDateTimeBefore(@Param("localDateTime") LocalDateTime localDateTime);