[Spring] 비관적 락을 이용한 songCnt 업데이트 문제

[DoohyunHwang97]

@PFCJeong

@Transactional
    override fun addSong(userId: Long, customPlaylistId: Long, songId: Long): CustomPlaylistBrief {
        val customPlaylist = customPlaylistRepository.findById(customPlaylistId).orElseThrow { CustomPlaylistNotFoundException() }

        if (customPlaylist.userId != userId) throw CustomPlaylistNotFoundException()

        val songToAdd = songRepository.findById(songId).orElseThrow { SongNotFoundException() }
        val songsInCustomPlaylist = customPlaylist.songs.map { it.song }

        if (!songsInCustomPlaylist.contains(songToAdd)) {
            customPlaylist.songs.add(CustomPlaylistSongEntity(customPlaylist = customPlaylist, song = songToAdd))
            // songCnt 업데이트
            customPlaylistRepository.increaseSongCnt(customPlaylistId)
        }

        return CustomPlaylistBrief(id = customPlaylistId, title = customPlaylist.title, songCnt = customPlaylist.songCnt)
    }

interface CustomPlaylistRepository : JpaRepository<CustomPlaylistEntity, Long> {
    fun findAllByUserId(userId: Long): List<CustomPlaylistEntity>

    @Lock(value = LockModeType.PESSIMISTIC_WRITE)
    @Query("UPDATE custom_playlists c SET c.songCnt = c.songCnt + 1 WHERE c.id = :playlistID")
    fun increaseSongCnt(playlistID: Long)
}

테스트 코드는 아래와 같이 트랜잭션 템플릿을 이용했습니다.

@Test
    fun `커스텀 플레이리스트에 곡 추가시, songCnt가 올라가고 연결 테이블의 row가 생성된다`() {
        TransactionTemplate(txManager).executeWithoutResult {
            val created = customPlaylistService.create(userId = 1L)

            customPlaylistService.addSong(
                    userId = 1L,
                    customPlaylistId = created.id,
                    songId = 1L
            )

            // songCnt 컬럼 체크
            val customPlaylistEntity = customPlaylistRepository.findById(created.id).get()

            assertThat(customPlaylistEntity.songCnt).isEqualTo(1)

            // customPlaylistSong 연결 테이블 체크
            val customPlaylist = customPlaylistService.get(userId = 1L, customPlaylistId = created.id)

            assertThat(customPlaylist.songs.size).isEqualTo(1)
            assertThat(customPlaylist.songs.first().id).isEqualTo(1L)
        }
    }

업데이트시에 아래와 같은 문제가 발생했는데 무엇이 잘못된지 모르겠어서 질문드립니다 ㅠ

org.springframework.dao.InvalidDataAccessApiUsageException: Expecting a SELECT query : UPDATE custom_playlists c SET c.songCnt = c.songCnt + 1 WHERE c.id = :playlistID

감사합니다

[jj1kim]

답변은 아니지만, 4번 구현할 때 @transactional을 쓰지 말라고 하셔서 고민중이었는데, 혹시 두현님처럼 구현하는 방식도 상관이 없는건가요..?

[PFCJeong]

@Transactional을 테스트 코드에서 쓰지 말라는 것이어서, 실제 코드에서는 써도 상관 없습니다.

[jj1kim]

아하ㅠㅜㅠㅜ 한참 고민하고 있었습니다... 감사합니다!

[PFCJeong]

안녕하세요 @Lock(value = LockModeType.PESSIMISTIC_WRITE) 제거하고 @Modifying로 시도해보시면 좋을 것 같습니다.

[DoohyunHwang97]

@PFCJeong
해결했습니다! 감사합니다:)
근데 동시성 이슈는 아직 해결이 안되었는데

@Test
    fun `커스텀 플레이리스트에 곡 추가 시, 동시성 이슈는 발생하지 않아야 한다`() {
        TransactionTemplate(txManager).executeWithoutResult {
            val created = customPlaylistService.create(userId = 1L)

            val threads = Executors.newFixedThreadPool(10)
            val parallelJobs = (1L..10L).map { songId ->
               // 멀티쓰레드로 인한 영속성 이슈 때문에 created 엔티티를 인식하지 못하는 것이 아닌지?
                threads.submit {
                    customPlaylistService.addSong(
                            userId = 1L,
                            customPlaylistId = created.id,
                            songId = songId
                    )
                }
            }
            parallelJobs.forEach { it.get() }

            // songCnt 컬럼 체크
            val customPlaylistEntity = customPlaylistRepository.findById(created.id).get()

            assertThat(customPlaylistEntity.songCnt).isEqualTo(10)

            // customPlaylistSong 연결 테이블 체크
            val customPlaylist = customPlaylistService.get(userId = 1L, customPlaylistId = created.id)

            assertThat(customPlaylist.songs.size).isEqualTo(10)
            assertThat(customPlaylist.songs.map { it.id }.toSet()).isEqualTo((1L..10L).toSet())
        }
    }

위 테스트를 진행할 때

@Transactional
    override fun addSong(userId: Long, customPlaylistId: Long, songId: Long): CustomPlaylistBrief {
       // 에러 발생 부분 findById시 null을 반환하는 것으로 보임
        val customPlaylist = customPlaylistRepository.findById(customPlaylistId).orElseThrow { CustomPlaylistNotFoundException() }
        //
        if (customPlaylist.userId != userId) throw CustomPlaylistNotFoundException()

        val songToAdd = songRepository.findById(songId).orElseThrow { SongNotFoundException() }
        val songsInCustomPlaylist = customPlaylist.songs.map { it.song }

        if (!songsInCustomPlaylist.contains(songToAdd)) {
            customPlaylist.songs.add(CustomPlaylistSongEntity(customPlaylist = customPlaylist, song = songToAdd))
            // songCnt 업데이트
            customPlaylistRepository.increaseSongCnt(customPlaylistId)
        }

        return CustomPlaylistBrief(id = customPlaylistId, title = customPlaylist.title, songCnt = customPlaylist.songCnt)
    }

위에서 예외 발생부분 보시면 테스트에서 CustomPlaylistNotFoundException 이 발생하는 것으로 보아 만들어져 있어야하는 엔티티 객체를 인식하지 못하는 것 같습니다. 이유를 생각해봤는데 테스트 코드 내에서 멀티쓰레드로 인한 영속성 이슈 때문에 created 엔티티를 인식하지 못하는 것이 아닌가 싶어서 질문드립니다! 확인해봤는데 create 함수에는 문제가 없습니다.
감사합니다(__)

[PFCJeong]

지금 겪고 계신 현상은 트랜잭션 격리 수준과 관련이 있습니다. 흠,, 세미나에서 다룰 내용은 아닌 것 같아요.

CustomPlaylistServiceTest에서 데이터 롤백을 위해 트랜잭션을 적용하지 말고, @BeforeEach등 다른 방법을 사용하시면 좋을 것 같습니다.

[DoohyunHwang97]

감사합니다! 해결했습니다:)