fix : 웹소켓 연결 헤더 검증에서 첫 메시지 페이로드로 검증으로 수정

Author: DongHyeonka

src/main/java/org/jungppo/bambooforest/chat/dto/ChatMessageDto.java

@@ -1,5 +1,7 @@
 package org.jungppo.bambooforest.chat.dto;
 
+import org.jungppo.bambooforest.chat.exception.ChatMessageValidationException;
+
 import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonProperty;
 
@@ -13,20 +15,47 @@
 @AllArgsConstructor(access = AccessLevel.PRIVATE)
 public class ChatMessageDto {
     @NotNull(message = "Message type cannot be null")
-    private MessageType type;
+    private ChatMessageType type;
 
     @NotBlank(message = "Message cannot be blank")
     private String message;
 
-    public enum MessageType {
-        ENTER, TALK, LEAVE //입장, 채팅, 퇴장
+    private String token;   // 인증 토큰
+
+    private String roomId;  // 룸 ID
+
+    private Long memberId;  // 멤버 ID
+
+    public enum ChatMessageType {
+        AUTH, ENTER, TALK, LEAVE //인증, 입장, 채팅, 퇴장
     }
 
     @JsonCreator
     public static ChatMessageDto of(
-        @JsonProperty("type") MessageType type,
-        @JsonProperty("message") String message
+        @JsonProperty("type") ChatMessageType type,
+        @JsonProperty("message") String message,
+        @JsonProperty("token") String token,
+        @JsonProperty("roomId") String roomId,
+        @JsonProperty("memberId") Long memberId
     ) {
-        return new ChatMessageDto(type, message);
+        ChatMessageDto dto = new ChatMessageDto(type, message, token, roomId, memberId);
+        dto.validate();
+        return dto;
+    }
+
+    private void validate() {
+        if (this.type == ChatMessageType.AUTH) {
+            validateField(this.token);
+            validateField(this.roomId);
+            validateField(this.memberId);
+        }
+    }
+
+    private void validateField(String field) {
+        if (field == null || field.isBlank()) throw new ChatMessageValidationException();
+    }
+
+    private void validateField(Long field) {
+        if (field == null) throw new ChatMessageValidationException();
     }
 }

src/main/java/org/jungppo/bambooforest/chat/exception/ChatMessageValidationException.java

@@ -0,0 +1,9 @@
+package org.jungppo.bambooforest.chat.exception;
+
+import static org.jungppo.bambooforest.global.exception.domain.ExceptionType.BIND_EXCEPTION;
+
+public class ChatMessageValidationException extends ChatBusinessException {
+    public ChatMessageValidationException() {
+        super(BIND_EXCEPTION);
+    }
+}