From 0d2e0a1af8e63c5fdf170f8562e992d725403ce7 Mon Sep 17 00:00:00 2001 From: Klaus Date: Tue, 28 Nov 2023 16:07:07 +0900 Subject: [PATCH] =?UTF-8?q?=EB=A3=B0=EB=A0=9B=20=EB=8F=8C=EB=A6=AC?= =?UTF-8?q?=EA=B8=B0=20API=20=EC=B6=94=EA=B0=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../co/vividnext/sodalive/can/CanService.kt | 2 +- .../co/vividnext/sodalive/can/use/CanUsage.kt | 3 +- .../live/roulette/RouletteController.kt | 10 ++++ .../sodalive/live/roulette/RouletteService.kt | 55 ++++++++++++++++++- .../live/roulette/SpinRouletteRequest.kt | 6 ++ 5 files changed, 72 insertions(+), 4 deletions(-) create mode 100644 src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/SpinRouletteRequest.kt diff --git a/src/main/kotlin/kr/co/vividnext/sodalive/can/CanService.kt b/src/main/kotlin/kr/co/vividnext/sodalive/can/CanService.kt index ea1fe34..1c0ca4b 100644 --- a/src/main/kotlin/kr/co/vividnext/sodalive/can/CanService.kt +++ b/src/main/kotlin/kr/co/vividnext/sodalive/can/CanService.kt @@ -53,7 +53,7 @@ class CanService(private val repository: CanRepository) { } .map { val title: String = when (it.canUsage) { - CanUsage.DONATION -> { + CanUsage.DONATION, CanUsage.SPIN_ROULETTE -> { if (it.room != null) { "[라이브 후원] ${it.room!!.member!!.nickname}" } else if (it.audioContent != null) { diff --git a/src/main/kotlin/kr/co/vividnext/sodalive/can/use/CanUsage.kt b/src/main/kotlin/kr/co/vividnext/sodalive/can/use/CanUsage.kt index 5e53fdc..58bbde9 100644 --- a/src/main/kotlin/kr/co/vividnext/sodalive/can/use/CanUsage.kt +++ b/src/main/kotlin/kr/co/vividnext/sodalive/can/use/CanUsage.kt @@ -4,5 +4,6 @@ enum class CanUsage { LIVE, DONATION, CHANGE_NICKNAME, - ORDER_CONTENT + ORDER_CONTENT, + SPIN_ROULETTE } diff --git a/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteController.kt b/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteController.kt index f7ec72f..69dd64d 100644 --- a/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteController.kt +++ b/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteController.kt @@ -39,4 +39,14 @@ class RouletteController(private val service: RouletteService) { return ApiResponse.ok(response) } + + @PostMapping("/spin") + fun spinRoulette( + @RequestBody request: SpinRouletteRequest, + @AuthenticationPrincipal(expression = "#this == 'anonymousUser' ? null : member") member: Member? + ) = run { + if (member == null) throw SodaException("로그인 정보를 확인해주세요.") + + ApiResponse.ok(service.spinRoulette(request = request, memberId = member.id!!)) + } } diff --git a/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteService.kt b/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteService.kt index 866a090..4a90880 100644 --- a/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteService.kt +++ b/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/RouletteService.kt @@ -1,11 +1,24 @@ package kr.co.vividnext.sodalive.live.roulette +import kr.co.vividnext.sodalive.can.payment.CanPaymentService +import kr.co.vividnext.sodalive.can.use.CanUsage import kr.co.vividnext.sodalive.common.SodaException +import kr.co.vividnext.sodalive.live.room.LiveRoomRepository +import kr.co.vividnext.sodalive.member.MemberRole import org.springframework.data.repository.findByIdOrNull import org.springframework.stereotype.Service +import org.springframework.transaction.annotation.Transactional +import java.util.concurrent.locks.ReentrantReadWriteLock +import kotlin.concurrent.write @Service -class RouletteService(private val repository: RouletteRepository) { +class RouletteService( + private val canPaymentService: CanPaymentService, + private val repository: RouletteRepository, + private val roomRepository: LiveRoomRepository +) { + private val tokenLocks: MutableMap = mutableMapOf() + fun createOrUpdateRoulette(memberId: Long, request: CreateOrUpdateRouletteRequest) { rouletteValidate(request) @@ -16,7 +29,10 @@ class RouletteService(private val repository: RouletteRepository) { items = request.items ) - repository.save(roulette) + val lock = getOrCreateLock(memberId = memberId) + lock.write { + repository.save(roulette) + } } private fun rouletteValidate(request: CreateOrUpdateRouletteRequest) { @@ -46,4 +62,39 @@ class RouletteService(private val repository: RouletteRepository) { return GetRouletteResponse(can = roulette.can, items = roulette.items) } + + @Transactional + fun spinRoulette(request: SpinRouletteRequest, memberId: Long): GetRouletteResponse { + // STEP 1 - 라이브 정보 가져오기 + val room = roomRepository.findByIdOrNull(request.roomId) + ?: throw SodaException("해당하는 라이브가 없습니다.") + + val host = room.member ?: throw SodaException("잘못된 요청입니다.") + + if (host.role != MemberRole.CREATOR) { + throw SodaException("비비드넥스트와 계약한\n크리에이터에게만 후원을 하실 수 있습니다.") + } + + // STEP 2 - 룰렛 데이터 가져오기 + val roulette = repository.findByIdOrNull(id = host.id!!) + + if (roulette == null || !roulette.isActive || roulette.items.isEmpty()) { + throw SodaException("룰렛을 사용할 수 없습니다.") + } + + // STEP 3 - 캔 사용 + canPaymentService.spendCan( + memberId = memberId, + needCan = roulette.can, + canUsage = CanUsage.SPIN_ROULETTE, + liveRoom = room, + container = request.container + ) + + return GetRouletteResponse(can = roulette.can, items = roulette.items) + } + + private fun getOrCreateLock(memberId: Long): ReentrantReadWriteLock { + return tokenLocks.computeIfAbsent(memberId) { ReentrantReadWriteLock() } + } } diff --git a/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/SpinRouletteRequest.kt b/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/SpinRouletteRequest.kt new file mode 100644 index 0000000..2cdbfa5 --- /dev/null +++ b/src/main/kotlin/kr/co/vividnext/sodalive/live/roulette/SpinRouletteRequest.kt @@ -0,0 +1,6 @@ +package kr.co.vividnext.sodalive.live.roulette + +data class SpinRouletteRequest( + val roomId: Long, + val container: String +)