좋아요 기능 최적화 (feat. redis)

Sihwan Kim·2024년 5월 30일

DB

목록 보기

4/4

앞선 블로그에서 좋아요 기능을 좋아요 테이블과 Join과 Count 쿼리를 통해 좋아요 수를 조회하였다. 하지만 100만개 정도의 좋아요를 가진 테이블을 조회할 때 아무리 쿼리를 최적화 하여도 2초 이상의 시간이 걸렸다. 하나를 조회하는 것에서도 이정도이니 좋아요 수와 함께 리스트를 조회하려면 더 많은 시간을 소요해야한다. 따라서 이를 최적화하는 과정을 기록하고자 한다.

Service

redis의 increment를 사용해서 좋아요 수를 증가 시킨다.

    @Transactional
    public void likeTrip(Long tripId, Long userId) throws InterruptedException {
        String key = LIKE_KEY_PREFIX + tripId;
        tripLikeRepository.save(TripLike.builder().tripId(tripId).registeredAt(LocalDateTime.now()).userId(userId).build());
        likeRedisTemplate.opsForValue().increment(key);
        Thread.sleep(10000L);
    }

스케쥴링

1분마다 redis에 있는 좋아요수를 모두 불러와서 db에 batch update를 통해 한번에 업데이트 했다.

@Service
@Slf4j
@RequiredArgsConstructor
public class TripLikeBatchService {

    private final RedisTemplate<String, Object> likeRedisTemplate;
    private final JdbcTemplate jdbcTemplate;

    private final String sql = "UPDATE trip SET like_count = ? where id = ?";
    @Scheduled(fixedRate = 60000) // 1분마다 실행
    public void syncLikes() {
        log.info("scheduled");
        try{
            long startTime = System.currentTimeMillis();
            Set<String> keys = likeRedisTemplate.keys(LIKE_KEY_PREFIX + "*");
            List<Object[]> batchArgs = new ArrayList<>();
            if (keys != null) {
                for (String key : keys) {
                    Long tripId = Long.valueOf(key.replace(LIKE_KEY_PREFIX, ""));
                    Integer likeCount = (Integer) likeRedisTemplate.opsForValue().get(key);
                    if (likeCount != null && likeCount > 0) {
                        batchArgs.add(new Object[] {likeCount,tripId});
                    }
                }
                jdbcTemplate.batchUpdate(sql,batchArgs);
                long endTime = System.currentTimeMillis();
                log.info("실행 시간 : " + (endTime - startTime));
            }
            else{
                log.info("keys 가 null");
            }
        }
        catch (EntityNotFoundException e){
            log.error(e.getMessage());
        }
        catch (Exception e){
            log.error(e.getMessage());
        }

    }

}