fix :: missed controller edition

API-Team-Organization · Sep 10, 2024 · 9578068 · 9578068
1 parent a4810f1
commit 9578068
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/main/kotlin/com/teamapi/palette/controller/InfoController.kt b/src/main/kotlin/com/teamapi/palette/controller/InfoController.kt
@@ -2,9 +2,11 @@ package com.teamapi.palette.controller
 
 import com.teamapi.palette.annotations.SwaggerRequireAuthorize
 import com.teamapi.palette.dto.user.UpdateRequest
+import com.teamapi.palette.dto.user.UserResponse
 import com.teamapi.palette.response.Response
 import com.teamapi.palette.response.ResponseBody
 import com.teamapi.palette.service.UserService
+import org.springframework.http.ResponseEntity
 import org.springframework.web.bind.annotation.*
 
 @RestController
@@ -14,12 +16,16 @@ class InfoController(
     private val userService: UserService
 ) {
     @GetMapping("/me")
-    fun myInfo() = userService.me()
-        .map { ResponseBody.ok("유저 조회 성공", it) }
+    suspend fun myInfo(): ResponseEntity<ResponseBody<UserResponse>> {
+        val data = userService.me()
+        return ResponseBody.ok("유저 조회 성공", data)
+    }
 
     @PatchMapping("/me")
-    fun updateInfo(
+    suspend fun updateInfo(
         @RequestBody request: UpdateRequest
-    ) = userService.update(request)
-        .thenReturn(Response.ok("유저 수정 성공"))
+    ): ResponseEntity<Response> {
+        userService.update(request)
+        return Response.ok("유저 수정 성공")
+    }
 }