#199 feat: 메시지 전송 API

PSR-Co · Nov 26, 2023 · 12cba85 · 12cba85
1 parent bcfdff2
commit 12cba85
Show file tree

Hide file tree

Showing 5 changed files with 70 additions and 3 deletions.
diff --git a/src/main/kotlin/com/psr/psr/chat/controller/ChatController.kt b/src/main/kotlin/com/psr/psr/chat/controller/ChatController.kt
@@ -1,5 +1,6 @@
 package com.psr.psr.chat.controller
 
+import com.psr.psr.chat.dto.request.ChatMessageReq
 import com.psr.psr.chat.service.ChatService
 import com.psr.psr.global.dto.BaseResponse
 import com.psr.psr.global.jwt.UserAccount
@@ -11,6 +12,7 @@ import io.swagger.v3.oas.annotations.responses.ApiResponse
 import io.swagger.v3.oas.annotations.responses.ApiResponses
 import io.swagger.v3.oas.annotations.security.SecurityRequirement
 import io.swagger.v3.oas.annotations.tags.Tag
+import jakarta.validation.Valid
 import org.springframework.security.core.annotation.AuthenticationPrincipal
 import org.springframework.web.bind.annotation.*
 
@@ -62,5 +64,26 @@ class ChatController(
         return BaseResponse(chatService.leaveChatRoom(userAccount.getUser(), chatRoomId));
     }
 
+    /**
+     * 메시지 전송
+     */
+    @Operation(summary = "메시지 전송(박소정)", description = "채팅방에 메시지를 전송한다.")
+    @ApiResponses(
+        value = [
+            ApiResponse(responseCode = "200", description = "요청에 성공했습니다."),
+            ApiResponse(
+                responseCode = "404",
+                description = "해당 요청을 찾을 수 없습니다.",
+                content = arrayOf(Content(schema = Schema(implementation = BaseResponse::class)))
+            )]
+    )
+    @PostMapping("/{chatRoomId}")
+    fun createChatMessage(@AuthenticationPrincipal userAccount: UserAccount,
+                          @Parameter(description = "(Long) 채팅방 id", example = "1") @PathVariable chatRoomId: Long,
+                          @RequestBody @Valid request: ChatMessageReq
+    ): BaseResponse<Unit> {
+        return BaseResponse(chatService.createChatMessage(userAccount.getUser(), chatRoomId, request))
+    }
+
 
 }
diff --git a/src/main/kotlin/com/psr/psr/chat/dto/request/ChatMessageReq.kt b/src/main/kotlin/com/psr/psr/chat/dto/request/ChatMessageReq.kt
@@ -0,0 +1,14 @@
+package com.psr.psr.chat.dto.request
+
+
+import io.swagger.v3.oas.annotations.media.Schema
+import jakarta.validation.constraints.NotBlank
+
+data class ChatMessageReq(
+
+    @Schema(description = "메세지", example = "안녕하세요~")
+    @field:NotBlank(message = "메세지를 입력해주세요.")
+    val message: String
+
+)
+
diff --git a/src/main/kotlin/com/psr/psr/chat/entity/ChatMessage.kt b/src/main/kotlin/com/psr/psr/chat/entity/ChatMessage.kt
@@ -8,7 +8,7 @@ import org.jetbrains.annotations.NotNull
 @Entity
 data class ChatMessage(
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
-    var id: Long,
+    var id: Long? = null,
 
     @ManyToOne
     @JoinColumn(nullable = false, name = "sender_user_id")
@@ -21,4 +21,14 @@ data class ChatMessage(
     @NotNull
     var message: String
 
-) : BaseEntity()
+) : BaseEntity() {
+    companion object {
+        fun toEntity(user: User, chatRoom: ChatRoom, message: String): ChatMessage {
+            return ChatMessage(
+                senderUser = user,
+                chatRoom = chatRoom,
+                message = message
+            )
+        }
+    }
+}
diff --git a/src/main/kotlin/com/psr/psr/chat/repository/ChatMessageRepository.kt b/src/main/kotlin/com/psr/psr/chat/repository/ChatMessageRepository.kt
@@ -0,0 +1,9 @@
+package com.psr.psr.chat.repository
+
+import com.psr.psr.chat.entity.ChatMessage
+import org.springframework.data.jpa.repository.JpaRepository
+import org.springframework.stereotype.Repository
+
+@Repository
+interface ChatMessageRepository : JpaRepository<ChatMessage, Long> {
+}
diff --git a/src/main/kotlin/com/psr/psr/chat/service/ChatService.kt b/src/main/kotlin/com/psr/psr/chat/service/ChatService.kt
@@ -1,6 +1,9 @@
 package com.psr.psr.chat.service
 
+import com.psr.psr.chat.dto.request.ChatMessageReq
+import com.psr.psr.chat.entity.ChatMessage
 import com.psr.psr.chat.entity.ChatRoom
+import com.psr.psr.chat.repository.ChatMessageRepository
 import com.psr.psr.chat.repository.ChatRoomRepository
 import com.psr.psr.global.Constant
 import com.psr.psr.global.exception.BaseException
@@ -14,7 +17,8 @@ import org.springframework.transaction.annotation.Transactional
 @Service
 class ChatService(
     private val orderRepository: OrderRepository,
-    private val chatRoomRepository: ChatRoomRepository
+    private val chatRoomRepository: ChatRoomRepository,
+    private val chatMessageRepository: ChatMessageRepository
 ) {
     @Transactional
     fun createChatRoom(user: User, orderId: Long) {
@@ -35,4 +39,11 @@ class ChatService(
         if(chatRoom.senderUser==null && chatRoom.receiverUser==null)
             chatRoomRepository.delete(chatRoom)
     }
+
+    @Transactional
+    fun createChatMessage(user: User, chatRoomId: Long, request: ChatMessageReq) {
+        val chatRoom: ChatRoom = chatRoomRepository.findByIdAndStatus(chatRoomId, Constant.UserStatus.ACTIVE_STATUS)
+            ?: throw BaseException(BaseResponseCode.NOT_FOUND_CHATROOM)
+        chatMessageRepository.save(ChatMessage.toEntity(user, chatRoom, request.message))
+    }
 }