feat: 비관적 락 적용

kakao-tech-campus-2nd-step3 · Nov 14, 2024 · f2ccd6e · f2ccd6e
1 parent 731a17e
commit f2ccd6e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/main/java/com/potatocake/everymoment/repository/LikeRepository.java b/src/main/java/com/potatocake/everymoment/repository/LikeRepository.java
@@ -2,12 +2,18 @@
 
 import com.potatocake.everymoment.entity.Diary;
 import com.potatocake.everymoment.entity.Like;
+import jakarta.persistence.LockModeType;
 import java.util.Optional;
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Lock;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
 
 public interface LikeRepository extends JpaRepository<Like, Long> {
 
-    Optional<Like> findByMemberIdAndDiaryId(Long memberId, Long diaryId);
+    @Lock(LockModeType.PESSIMISTIC_WRITE)
+    @Query("SELECT l FROM Like l WHERE l.member.id = :memberId AND l.diary.id = :diaryId")
+    Optional<Like> findByMemberIdAndDiaryIdWithLock(@Param("memberId") Long memberId, @Param("diaryId") Long diaryId);
 
     Long countByDiary(Diary diary);
 

diff --git a/src/main/java/com/potatocake/everymoment/service/LikeService.java b/src/main/java/com/potatocake/everymoment/service/LikeService.java
@@ -48,7 +48,7 @@ public void toggleLike(Long memberId, Long diaryId) {
         Member member = memberRepository.findById(memberId)
                 .orElseThrow(() -> new GlobalException(ErrorCode.MEMBER_NOT_FOUND));
 
-        Optional<Like> existingLike = likeRepository.findByMemberIdAndDiaryId(memberId, diaryId);
+        Optional<Like> existingLike = likeRepository.findByMemberIdAndDiaryIdWithLock(memberId, diaryId);
 
         if (existingLike.isPresent()) {
             // 이미 좋아요가 존재하면 삭제 (좋아요 취소)