Feat: chat entity 설계

Author: skyeong42

build.gradle.kts

@@ -55,7 +55,10 @@ dependencies {
 
 	//Swagger
 	implementation("org.springdoc:springdoc-openapi-starter-webmvc-ui:2.7.0")
+
+	//Web Socket
 	implementation("org.java-websocket:Java-WebSocket:1.5.2")
+	implementation ("org.springframework:spring-messaging")
 
 	// JWT
 	implementation("io.jsonwebtoken:jjwt-api:0.12.6")

src/main/java/cmf/commitField/domain/chat/chatMessage/controller/ChatController.java

@@ -0,0 +1,55 @@
+package cmf.commitField.domain.chat.chatMessage.controller;
+
+import cmf.commitField.domain.chat.chatMessage.entity.ChatMessage;
+import cmf.commitField.domain.chat.chatMessage.repository.ChatMessageRepository;
+import cmf.commitField.domain.chat.chatRoom.entity.ChatRoom;
+import cmf.commitField.domain.chat.chatRoom.repository.ChatRoomRepository;
+import cmf.commitField.domain.user.entity.CustomOAuth2User;
+import cmf.commitField.domain.user.entity.User;
+import cmf.commitField.domain.user.repository.UserRepository;
+import lombok.RequiredArgsConstructor;
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.stereotype.Controller;
+
+import java.util.Optional;
+
+@Controller
+@RequiredArgsConstructor
+public class ChatController {
+
+    private final ChatMessageRepository chatMessageRepository;
+    private final ChatRoomRepository chatRoomRepository;
+    private final UserRepository userRepository;
+    private final SimpMessagingTemplate messagingTemplate; // 특정 채널로 메시지를 보내기 위함
+
+    @MessageMapping("/chat.sendMessage")
+    public void sendMessage(ChatMessage chatMessage) {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+
+        if (authentication != null && authentication.isAuthenticated() &&
+                authentication.getPrincipal() instanceof CustomOAuth2User) {
+
+            CustomOAuth2User oauth2User = (CustomOAuth2User) authentication.getPrincipal();
+            Optional<User> userOptional = userRepository.findByUsername(oauth2User.getName());
+
+            if (userOptional.isPresent()) {
+                User user = userOptional.get();
+                Optional<ChatRoom> chatRoomOptional = chatRoomRepository.findById(chatMessage.getChatRoom().getId());
+
+                if (chatRoomOptional.isPresent()) {
+                    ChatRoom chatRoom = chatRoomOptional.get();
+                    chatMessage.setUser(user);
+                    chatMessage.setChatRoom(chatRoom);
+
+                    ChatMessage savedMessage = chatMessageRepository.save(chatMessage);
+
+                    // 채팅방 구독자들에게 메시지 전송
+                    messagingTemplate.convertAndSend("/sub/chat/room/" + chatRoom.getId(), savedMessage);
+                }
+            }
+        }
+    }
+}

src/main/java/cmf/commitField/domain/chat/chatMessage/entity/ChatMessage.java

@@ -7,23 +7,20 @@
 import jakarta.persistence.FetchType;
 import jakarta.persistence.JoinColumn;
 import jakarta.persistence.ManyToOne;
-import lombok.AllArgsConstructor;
-import lombok.Getter;
-import lombok.NoArgsConstructor;
+import lombok.*;
 import lombok.experimental.SuperBuilder;
 
-import java.time.LocalDateTime;
-
 @Entity
 @SuperBuilder
 @Getter
-@NoArgsConstructor
-@AllArgsConstructor
+@Setter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@AllArgsConstructor(access = AccessLevel.PROTECTED)
 public class ChatMessage extends BaseEntity {
 
     private String message;
 
-    private LocalDateTime sendTime;
+//    private LocalDateTime sendTime;
 
     @ManyToOne(fetch = FetchType.LAZY)
     @JoinColumn(name = "user_id")

src/main/java/cmf/commitField/domain/chat/chatMessage/repository/ChatMessageRepository.java

@@ -0,0 +1,9 @@
+package cmf.commitField.domain.chat.chatMessage.repository;
+
+import cmf.commitField.domain.chat.chatMessage.entity.ChatMessage;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface ChatMessageRepository extends JpaRepository<ChatMessage, Long> {
+}

src/main/java/cmf/commitField/domain/chat/chatRoom/entity/ChatRoom.java

@@ -5,9 +5,7 @@
 import cmf.commitField.domain.user.entity.User;
 import cmf.commitField.global.jpa.BaseEntity;
 import jakarta.persistence.*;
-import lombok.AllArgsConstructor;
-import lombok.Getter;
-import lombok.NoArgsConstructor;
+import lombok.*;
 import lombok.experimental.SuperBuilder;
 
 import java.time.LocalDateTime;
@@ -16,11 +14,13 @@
 @Entity
 @SuperBuilder
 @Getter
-@NoArgsConstructor
-@AllArgsConstructor
+@Setter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@AllArgsConstructor(access = AccessLevel.PROTECTED)
 public class ChatRoom extends BaseEntity {
 
     private String title;
+    private String tier;
 
     private LocalDateTime deletedAt;
 

src/main/java/cmf/commitField/domain/chat/chatRoom/repository/ChatRoomRepository.java

@@ -0,0 +1,9 @@
+package cmf.commitField.domain.chat.chatRoom.repository;
+
+import cmf.commitField.domain.chat.chatRoom.entity.ChatRoom;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface ChatRoomRepository extends JpaRepository<ChatRoom, Long> {
+}

src/main/java/cmf/commitField/domain/chat/userChatRoom/entity/UserChatRoom.java

@@ -7,23 +7,22 @@
 import jakarta.persistence.FetchType;
 import jakarta.persistence.JoinColumn;
 import jakarta.persistence.ManyToOne;
-import lombok.AllArgsConstructor;
-import lombok.Getter;
-import lombok.NoArgsConstructor;
+import lombok.*;
 import lombok.experimental.SuperBuilder;
 
 @Entity
 @Getter
+@Setter
 @SuperBuilder
-@NoArgsConstructor
-@AllArgsConstructor
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@AllArgsConstructor(access = AccessLevel.PROTECTED)
 public class UserChatRoom extends BaseEntity {
 
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "user_id")
+    @JoinColumn(name = "user_id",nullable = false)
     private User user;
 
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "chat_room_id")
+    @JoinColumn(name = "chat_room_id",nullable = false)
     private ChatRoom chatRoom;
 }