Feat :: 룸 삭제 완성

API-Team-Organization · Jun 13, 2024 · 047fe28 · 047fe28
1 parent 63165c9
commit 047fe28
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 6 deletions.
diff --git a/src/main/kotlin/com/teamapi/palette/controller/RoomController.kt b/src/main/kotlin/com/teamapi/palette/controller/RoomController.kt
@@ -1,13 +1,10 @@
 package com.teamapi.palette.controller
 
+import com.azure.core.annotation.PathParam
 import com.teamapi.palette.response.Response
 import com.teamapi.palette.response.ResponseBody
 import com.teamapi.palette.service.RoomService
-import org.springframework.web.bind.annotation.GetMapping
-import org.springframework.web.bind.annotation.PatchMapping
-import org.springframework.web.bind.annotation.PostMapping
-import org.springframework.web.bind.annotation.RequestMapping
-import org.springframework.web.bind.annotation.RestController
+import org.springframework.web.bind.annotation.*
 import reactor.core.publisher.Flux
 
 @RestController
@@ -22,4 +19,8 @@ class RoomController (
     @GetMapping("/list")
     fun getRoomList() = roomService.getRoomList()
         .map { ResponseBody.ok("룸 조회 성공", it) }
+
+    @DeleteMapping("/{roomId}")
+    fun deleteRoom(@PathVariable roomId: Long) = roomService.deleteRoom(roomId)
+        .thenReturn(Response.ok("룸 삭제 완료"))
 }
diff --git a/src/main/kotlin/com/teamapi/palette/response/ErrorCode.kt b/src/main/kotlin/com/teamapi/palette/response/ErrorCode.kt
@@ -15,7 +15,9 @@ enum class ErrorCode(
     INVALID_PARAMETER(HttpStatus.BAD_REQUEST, "잘못된 파라미터"),
     METHOD_NOT_ALLOWED(HttpStatus.METHOD_NOT_ALLOWED, "허용되지 않은 요청 메소드"),
     MEDIA_TYPE_NOT_SUPPORTED(HttpStatus.BAD_REQUEST, "허용되지 않은 미디어 자료형"),
-    BAD_REQUEST(HttpStatus.BAD_REQUEST, "올바르지 않은 요청")
+    BAD_REQUEST(HttpStatus.BAD_REQUEST, "올바르지 않은 요청"),
+    ROOM_NOT_FOUND(HttpStatus.NOT_FOUND, "해당 룸을 찾을 수 없음"),
+    FORBIDDEN(HttpStatus.FORBIDDEN, "권한 없음")
     ;
 
     override fun getName() = name

diff --git a/src/main/kotlin/com/teamapi/palette/service/RoomService.kt b/src/main/kotlin/com/teamapi/palette/service/RoomService.kt
@@ -4,9 +4,12 @@ import com.teamapi.palette.dto.room.RoomResponse
 import com.teamapi.palette.entity.Room
 import com.teamapi.palette.repository.RoomRepository
 import com.teamapi.palette.repository.UserRepository
+import com.teamapi.palette.response.ErrorCode
+import com.teamapi.palette.response.exception.CustomException
 import com.teamapi.palette.util.findUser
 import org.springframework.stereotype.Service
 import reactor.core.publisher.Mono
+import reactor.kotlin.core.publisher.switchIfEmpty
 
 @Service
 class RoomService (
@@ -33,4 +36,22 @@ class RoomService (
             }
             .collectList()
     }
+
+    fun deleteRoom(roomId: Long): Mono<Void> {
+        return sessionHolder
+            .me()
+            .findUser(userRepository)
+            .flatMap { user ->
+                roomRepository.findById(roomId)
+                    .switchIfEmpty {
+                        Mono.error(CustomException(ErrorCode.ROOM_NOT_FOUND))
+                    }
+                    .flatMap { room ->
+                        if (room.userId == user.id)
+                            roomRepository.delete(room)
+                        else
+                            Mono.error(CustomException(ErrorCode.FORBIDDEN))
+                    }
+            }
+    }
 }