라이브 방 정보 - ReentrantReadWriteLock 추가

src/main/kotlin/kr/co/vividnext/sodalive/live/room/LiveRoomService.kt

@@ -53,6 +53,8 @@ import java.time.LocalDateTime
 import java.time.ZoneId
 import java.time.format.DateTimeFormatter
 import java.util.Date
+import java.util.concurrent.locks.ReentrantReadWriteLock
+import kotlin.concurrent.write
 
 @Service
 @Transactional(readOnly = true)
@@ -88,6 +90,8 @@ class LiveRoomService(
     @Value("\${cloud.aws.cloud-front.host}")
     private val cloudFrontHost: String
 ) {
+    private val tokenLocks: MutableMap<Long, ReentrantReadWriteLock> = mutableMapOf()
+
     fun getRoomList(
         dateString: String?,
         status: LiveRoomStatus,
@@ -415,49 +419,52 @@ class LiveRoomService(
             throw SodaException("비밀번호가 일치하지 않습니다.\n다시 확인 후 입력해주세요.")
         }
 
-        var roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+        val lock = getOrCreateLock(memberId = member.id!!)
-        if (roomInfo == null) {
+        lock.write {
-            roomInfo = roomInfoRepository.save(LiveRoomInfo(roomId = request.roomId))
+            var roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
-        }
+            if (roomInfo == null) {
-
+                roomInfo = roomInfoRepository.save(LiveRoomInfo(roomId = request.roomId))
-        if (roomInfo.speakerCount + roomInfo.listenerCount + roomInfo.managerCount >= room.numberOfPeople) {
-            throw SodaException("방이 가득찼습니다.")
-        }
-
-        if (
-            room.price > 0 &&
-            room.member!!.id!! != member.id!! &&
-            canRepository.isExistPaidLiveRoom(memberId = member.id!!, roomId = request.roomId) == null
-        ) {
-            val findMember = memberRepository.findByIdOrNull(id = member.id!!)
-                ?: throw SodaException("로그인 정보를 확인해 주세요.")
-
-            val totalCan = findMember.getChargeCan(request.container) + findMember.getRewardCan(request.container)
-            if (totalCan < room.price) {
-                throw SodaException("${room.price - totalCan}캔이 부족합니다. 충전 후 이용해 주세요.")
             }
 
-            canPaymentService.spendCan(
+            if (roomInfo.speakerCount + roomInfo.listenerCount + roomInfo.managerCount >= room.numberOfPeople) {
-                memberId = member.id!!,
+                throw SodaException("방이 가득찼습니다.")
-                needCan = room.price,
+            }
-                canUsage = CanUsage.LIVE,
+
-                liveRoom = room,
+            if (
-                container = request.container
+                room.price > 0 &&
-            )
+                room.member!!.id!! != member.id!! &&
+                canRepository.isExistPaidLiveRoom(memberId = member.id!!, roomId = request.roomId) == null
+            ) {
+                val findMember = memberRepository.findByIdOrNull(id = member.id!!)
+                    ?: throw SodaException("로그인 정보를 확인해 주세요.")
+
+                val totalCan = findMember.getChargeCan(request.container) + findMember.getRewardCan(request.container)
+                if (totalCan < room.price) {
+                    throw SodaException("${room.price - totalCan}캔이 부족합니다. 충전 후 이용해 주세요.")
+                }
+
+                canPaymentService.spendCan(
+                    memberId = member.id!!,
+                    needCan = room.price,
+                    canUsage = CanUsage.LIVE,
+                    liveRoom = room,
+                    container = request.container
+                )
+            }
+
+            roomInfo.removeListener(member)
+            roomInfo.removeSpeaker(member)
+            roomInfo.removeManager(member)
+
+            if (room.member!!.id == member.id) {
+                roomInfo.addSpeaker(member, cloudFrontHost)
+            } else {
+                roomInfo.addListener(member, cloudFrontHost)
+            }
+
+            roomInfoRepository.save(roomInfo)
+            roomVisitService.roomVisit(room, member)
         }
-
-        roomInfo.removeListener(member)
-        roomInfo.removeSpeaker(member)
-        roomInfo.removeManager(member)
-
-        if (room.member!!.id == member.id) {
-            roomInfo.addSpeaker(member, cloudFrontHost)
-        } else {
-            roomInfo.addListener(member, cloudFrontHost)
-        }
-
-        roomInfoRepository.save(roomInfo)
-        roomVisitService.roomVisit(room, member)
     }
 
     fun getRecentRoomInfo(member: Member): GetRecentRoomInfoResponse {
@@ -621,18 +628,21 @@ class LiveRoomService(
     }
 
     fun deleteDonationMessage(request: DeleteLiveRoomDonationMessage, member: Member) {
-        val room = repository.findByIdOrNull(request.roomId)
+        val lock = getOrCreateLock(memberId = member.id!!)
-            ?: throw SodaException("해당하는 라이브가 없습니다.")
+        lock.write {
+            val room = repository.findByIdOrNull(request.roomId)
+                ?: throw SodaException("해당하는 라이브가 없습니다.")
 
-        if (member.id!! != room.member!!.id!!) {
+            if (member.id!! != room.member!!.id!!) {
-            throw SodaException("잘못된 요청입니다.")
+                throw SodaException("잘못된 요청입니다.")
+            }
+
+            val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+                ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
+
+            roomInfo.removeDonationMessage(request.messageUUID)
+            roomInfoRepository.save(roomInfo)
         }
-
-        val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
-            ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
-
-        roomInfo.removeDonationMessage(request.messageUUID)
-        roomInfoRepository.save(roomInfo)
     }
 
     fun getUserProfile(roomId: Long, userId: Long, member: Member): GetLiveRoomUserProfileResponse {
@@ -701,64 +711,73 @@ class LiveRoomService(
     }
 
     fun setSpeaker(request: SetManagerOrSpeakerOrAudienceRequest) {
-        val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+        val lock = getOrCreateLock(memberId = request.memberId)
-            ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
+        lock.write {
+            val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+                ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
 
-        val account = memberRepository.findByIdOrNull(request.accountId)
+            val member = memberRepository.findByIdOrNull(request.memberId)
-            ?: throw SodaException("로그인 정보를 확인해 주세요.")
+                ?: throw SodaException("로그인 정보를 확인해 주세요.")
 
-        if (roomInfo.speakerCount > 9) {
+            if (roomInfo.speakerCount > 9) {
-            throw SodaException("스피커 정원이 초과하였습니다.")
+                throw SodaException("스피커 정원이 초과하였습니다.")
+            }
+
+            roomInfo.removeListener(member)
+            roomInfo.removeManager(member)
+            roomInfo.addSpeaker(member, cloudFrontHost)
+
+            roomInfoRepository.save(roomInfo)
         }
-
-        roomInfo.removeListener(account)
-        roomInfo.removeManager(account)
-        roomInfo.addSpeaker(account, cloudFrontHost)
-
-        roomInfoRepository.save(roomInfo)
     }
 
     fun setListener(request: SetManagerOrSpeakerOrAudienceRequest) {
-        val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+        val lock = getOrCreateLock(memberId = request.memberId)
-            ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
+        lock.write {
+            val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+                ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
 
-        val member = memberRepository.findByIdOrNull(request.accountId)
+            val member = memberRepository.findByIdOrNull(request.memberId)
-            ?: throw SodaException("로그인 정보를 확인해 주세요.")
+                ?: throw SodaException("로그인 정보를 확인해 주세요.")
 
-        roomInfo.removeSpeaker(member)
+            roomInfo.removeSpeaker(member)
-        roomInfo.removeManager(member)
+            roomInfo.removeManager(member)
-        roomInfo.addListener(member, cloudFrontHost)
+            roomInfo.addListener(member, cloudFrontHost)
 
-        roomInfoRepository.save(roomInfo)
+            roomInfoRepository.save(roomInfo)
+        }
     }
 
     fun setManager(request: SetManagerOrSpeakerOrAudienceRequest, member: Member) {
-        val room = repository.getLiveRoom(request.roomId) ?: throw SodaException("잘못된 요청입니다.")
+        val lock = getOrCreateLock(memberId = member.id!!)
-        if (room.member!!.id!! != member.id!!) {
+        lock.write {
-            throw SodaException("권한이 없습니다.")
+            val room = repository.getLiveRoom(request.roomId) ?: throw SodaException("잘못된 요청입니다.")
+            if (room.member!!.id!! != member.id!!) {
+                throw SodaException("권한이 없습니다.")
+            }
+
+            val user = memberRepository.findByIdOrNull(request.memberId) ?: throw SodaException("해당하는 유저가 없습니다.")
+            val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
+                ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
+
+            val roomAccountResponse = LiveRoomMember(member = user, cloudFrontHost)
+            if (roomInfo.managerList.contains(roomAccountResponse)) {
+                throw SodaException("이미 매니저 입니다.")
+            }
+
+            if (
+                !roomInfo.speakerList.contains(roomAccountResponse) &&
+                !roomInfo.listenerList.contains(roomAccountResponse)
+            ) {
+                throw SodaException("해당하는 유저가 없습니다.")
+            }
+
+            roomInfo.removeListener(user)
+            roomInfo.removeSpeaker(user)
+            roomInfo.addManager(user, cloudFrontHost)
+
+            roomInfoRepository.save(roomInfo)
         }
-
-        val user = memberRepository.findByIdOrNull(request.accountId) ?: throw SodaException("해당하는 유저가 없습니다.")
-        val roomInfo = roomInfoRepository.findByIdOrNull(request.roomId)
-            ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
-
-        val roomAccountResponse = LiveRoomMember(member = user, cloudFrontHost)
-        if (roomInfo.managerList.contains(roomAccountResponse)) {
-            throw SodaException("이미 매니저 입니다.")
-        }
-
-        if (
-            !roomInfo.speakerList.contains(roomAccountResponse) &&
-            !roomInfo.listenerList.contains(roomAccountResponse)
-        ) {
-            throw SodaException("해당하는 유저가 없습니다.")
-        }
-
-        roomInfo.removeListener(user)
-        roomInfo.removeSpeaker(user)
-        roomInfo.addManager(user, cloudFrontHost)
-
-        roomInfoRepository.save(roomInfo)
     }
 
     @Transactional
@@ -781,16 +800,19 @@ class LiveRoomService(
         )
 
         if (request.message.isNotBlank()) {
-            val roomInfo = roomInfoRepository.findByIdOrNull(room.id!!)
+            val lock = getOrCreateLock(memberId = member.id!!)
-                ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
+            lock.write {
+                val roomInfo = roomInfoRepository.findByIdOrNull(room.id!!)
+                    ?: throw SodaException("해당하는 라이브의 정보가 없습니다.")
 
-            roomInfo.addDonationMessage(
+                roomInfo.addDonationMessage(
-                nickname = member.nickname,
+                    nickname = member.nickname,
-                can = request.can,
+                    can = request.can,
-                donationMessage = request.message
+                    donationMessage = request.message
-            )
+                )
 
-            roomInfoRepository.save(roomInfo)
+                roomInfoRepository.save(roomInfo)
+            }
         }
     }
 
@@ -848,20 +870,27 @@ class LiveRoomService(
 
         val roomId = roomVisit?.room?.id
         if (roomId != null) {
-            val roomInfo = roomInfoRepository.findByIdOrNull(roomId)
+            val lock = getOrCreateLock(memberId = member.id!!)
-            if (roomInfo != null) {
+            lock.write {
-                val room = repository.getLiveRoom(roomId) ?: return
+                val roomInfo = roomInfoRepository.findByIdOrNull(roomId)
-                if (room.member != null && room.member!! == member) {
+                if (roomInfo != null) {
-                    room.isActive = false
+                    val room = repository.getLiveRoom(roomId) ?: return
-                    kickOutService.deleteKickOutData(roomId = room.id!!)
+                    if (room.member != null && room.member!! == member) {
-                    roomInfoRepository.deleteById(roomInfo.roomId)
+                        room.isActive = false
-                } else {
+                        kickOutService.deleteKickOutData(roomId = room.id!!)
-                    roomInfo.removeSpeaker(member)
+                        roomInfoRepository.deleteById(roomInfo.roomId)
-                    roomInfo.removeListener(member)
+                    } else {
-                    roomInfo.removeManager(member)
+                        roomInfo.removeSpeaker(member)
-                    roomInfoRepository.save(roomInfo)
+                        roomInfo.removeListener(member)
+                        roomInfo.removeManager(member)
+                        roomInfoRepository.save(roomInfo)
+                    }
                 }
             }
         }
     }
+
+    private fun getOrCreateLock(memberId: Long): ReentrantReadWriteLock {
+        return tokenLocks.computeIfAbsent(memberId) { ReentrantReadWriteLock() }
+    }
 }

src/main/kotlin/kr/co/vividnext/sodalive/live/room/SetManagerOrSpeakerOrAudienceRequest.kt

@@ -2,5 +2,5 @@ package kr.co.vividnext.sodalive.live.room
 
 data class SetManagerOrSpeakerOrAudienceRequest(
     val roomId: Long,
-    val accountId: Long
+    val memberId: Long
 )