Feat/54 chatting

.gitignore

@@ -37,4 +37,7 @@ out/
 .vscode/
 
 #####
-env.properties
+env.properties
+
+### 초기 데이터 ###
+initDB.java

build.gradle

@@ -29,6 +29,7 @@ dependencies {
 	implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
 	implementation 'org.springframework.boot:spring-boot-starter-security'
 	implementation 'org.springframework.boot:spring-boot-starter-web'
+	implementation 'org.springframework.boot:spring-boot-starter-websocket'
 	implementation 'org.springframework.boot:spring-boot-starter-validation'
 	compileOnly 'org.projectlombok:lombok'
 	annotationProcessor 'org.projectlombok:lombok'

src/main/java/com/sptp/backend/artwork/repository/ArtWork.java

@@ -0,0 +1,21 @@
+package com.sptp.backend.artwork.repository;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+import javax.persistence.*;
+
+@Entity
+@Getter
+@NoArgsConstructor
+@AllArgsConstructor
+@Builder
+public class ArtWork {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    @Column(name = "art_work_id")
+    private Long id;
+}

src/main/java/com/sptp/backend/chat_room/repository/ChatRoom.java

@@ -0,0 +1,35 @@
+package com.sptp.backend.chat_room.repository;
+
+import com.sptp.backend.artwork.repository.ArtWork;
+import com.sptp.backend.member.repository.Member;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+import javax.persistence.*;
+
+@Entity
+@Getter
+@NoArgsConstructor
+@AllArgsConstructor
+@Builder
+public class ChatRoom {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    @Column(name = "chat_room_id")
+    private Long id;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "artist_id")
+    private Member artist;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "collector_id")
+    private Member collector;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "art_work_id")
+    private ArtWork artWork;
+}

src/main/java/com/sptp/backend/chat_room/repository/ChatRoomRepository.java

@@ -0,0 +1,6 @@
+package com.sptp.backend.chat_room.repository;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface ChatRoomRepository extends JpaRepository<ChatRoom, Long> {
+}

src/main/java/com/sptp/backend/common/config/WebSocketConfig.java

@@ -0,0 +1,29 @@
+package com.sptp.backend.common.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.messaging.simp.config.MessageBrokerRegistry;
+import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
+import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
+import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
+
+@Configuration
+@EnableWebSocketMessageBroker
+public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
+
+    @Override
+    public void configureMessageBroker(MessageBrokerRegistry registry) {
+        // 전송자가 '/app/**' 경로로 메세지 전송
+        registry.setApplicationDestinationPrefixes("/app");
+
+        // '/queue/**' or '/topic/**' 경로로 송신되었을 때, 스프링 내장의 simplebroker가 메세지 처리
+        // queue는 1:1, topic은 1:n 관계 시 관습적으로 사용
+        registry.enableSimpleBroker("/queue", "/topic");
+    }
+
+    @Override
+    public void registerStompEndpoints(StompEndpointRegistry registry) {
+        registry.addEndpoint("/ws-connection")
+                .setAllowedOrigins("*")
+                .withSockJS(); // websocket 미지원 브라우저를 위해 추가
+    }
+}

src/main/java/com/sptp/backend/common/exception/ErrorCode.java

@@ -25,7 +25,10 @@ public enum ErrorCode {
 
     //토큰 예외
     TOKEN_EXPIRED(HttpStatus.UNAUTHORIZED, "토큰이 만료되었습니다."),
-    TOKEN_INVALID(HttpStatus.NOT_FOUND, "토큰이 유효하지 않습니다.");
+    TOKEN_INVALID(HttpStatus.NOT_FOUND, "토큰이 유효하지 않습니다."),
+
+    //채팅 예외
+    NOT_FOUND_CHAT_ROOM(HttpStatus.NOT_FOUND, "존재하지 않는 채팅방입니다.");
 
     private final HttpStatus httpStatus;
     private final String detail;

src/main/java/com/sptp/backend/message/repository/Message.java

@@ -0,0 +1,35 @@
+package com.sptp.backend.message.repository;
+
+import com.sptp.backend.chat_room.repository.ChatRoom;
+import com.sptp.backend.member.repository.Member;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+import javax.persistence.*;
+import java.time.LocalDateTime;
+
+@Entity
+@Getter
+@NoArgsConstructor
+@AllArgsConstructor
+@Builder
+public class Message {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    @Column(name = "message_id")
+    private Long id;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "sender_id")
+    private Member sender;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "artist_id")
+    private ChatRoom chatRoom;
+
+    private String message;
+//    private LocalDateTime sendTime;
+}

src/main/java/com/sptp/backend/message/repository/MessageRepository.java

@@ -0,0 +1,6 @@
+package com.sptp.backend.message.repository;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface MessageRepository extends JpaRepository<Message, Long> {
+}

src/main/java/com/sptp/backend/message/service/MessageService.java

@@ -0,0 +1,41 @@
+package com.sptp.backend.message.service;
+
+import com.sptp.backend.chat_room.repository.ChatRoom;
+import com.sptp.backend.chat_room.repository.ChatRoomRepository;
+import com.sptp.backend.common.exception.CustomException;
+import com.sptp.backend.common.exception.ErrorCode;
+import com.sptp.backend.member.repository.Member;
+import com.sptp.backend.member.repository.MemberRepository;
+import com.sptp.backend.message.repository.Message;
+import com.sptp.backend.message.repository.MessageRepository;
+import com.sptp.backend.message.web.dto.MessageRequest;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+@Service
+@Slf4j
+@RequiredArgsConstructor
+public class MessageService {
+
+    private final MessageRepository messageRepository;
+    private final ChatRoomRepository chatRoomRepository;
+    private final MemberRepository memberRepository;
+
+    public void saveMessage(Long loginMemberId, MessageRequest messageRequest) {
+
+        Member sender = memberRepository.findById(loginMemberId)
+                .orElseThrow(() -> new CustomException(ErrorCode.NOT_FOUND_MEMBER));
+
+        ChatRoom chatRoom = chatRoomRepository.findById(messageRequest.getChatRoomId())
+                .orElseThrow(() -> new CustomException(ErrorCode.NOT_FOUND_CHAT_ROOM));
+
+        Message message = Message.builder()
+                .sender(sender)
+                .chatRoom(chatRoom)
+                .message(messageRequest.getMessage())
+                .build();
+
+        messageRepository.save(message);
+    }
+}

src/main/java/com/sptp/backend/message/web/MessageController.java

@@ -0,0 +1,29 @@
+package com.sptp.backend.message.web;
+
+import com.sptp.backend.jwt.service.dto.CustomUserDetails;
+import com.sptp.backend.message.service.MessageService;
+import com.sptp.backend.message.web.dto.MessageRequest;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.simp.SimpMessagingTemplate;
+import org.springframework.security.core.annotation.AuthenticationPrincipal;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.validation.Valid;
+
+@Slf4j
+@RestController
+@RequiredArgsConstructor
+public class MessageController {
+
+    private final MessageService messageService;
+    private final SimpMessagingTemplate simpMessagingTemplate;
+
+    @MessageMapping("/send")
+    public void chat(@Valid MessageRequest messageRequest, @AuthenticationPrincipal CustomUserDetails userDetails) {
+        messageService.saveMessage(userDetails.getMember().getId(), messageRequest);
+        simpMessagingTemplate.convertAndSend("/queue/chat-rooms/" + messageRequest.getChatRoomId(),
+                messageRequest.getMessage());
+    }
+}

src/main/java/com/sptp/backend/message/web/dto/MessageRequest.java

@@ -0,0 +1,20 @@
+package com.sptp.backend.message.web.dto;
+
+import lombok.*;
+
+import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.NotNull;
+
+@Getter
+@Setter
+@NoArgsConstructor
+@AllArgsConstructor
+@Builder
+public class MessageRequest {
+
+    @NotNull(message = "채팅방 고유 번호는 필수입니다.")
+    private Long chatRoomId;
+
+    @NotBlank(message = "채팅 메세지를 입력해주세요.")
+    private String message;
+}

src/test/java/com/sptp/backend/message/service/MessageServiceTest.java

@@ -0,0 +1,135 @@
+package com.sptp.backend.message.service;
+
+import com.sptp.backend.artwork.repository.ArtWork;
+import com.sptp.backend.chat_room.repository.ChatRoom;
+import com.sptp.backend.chat_room.repository.ChatRoomRepository;
+import com.sptp.backend.common.exception.CustomException;
+import com.sptp.backend.common.exception.ErrorCode;
+import com.sptp.backend.member.repository.Member;
+import com.sptp.backend.member.repository.MemberRepository;
+import com.sptp.backend.message.repository.Message;
+import com.sptp.backend.message.repository.MessageRepository;
+import com.sptp.backend.message.web.dto.MessageRequest;
+import org.assertj.core.api.Assertions;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Nested;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import java.util.Optional;
+
+import static org.assertj.core.api.Assertions.*;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyLong;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+class MessageServiceTest {
+
+    @InjectMocks
+    MessageService messageService;
+
+    @Mock
+    ChatRoomRepository chatRoomRepository;
+
+    @Mock
+    MemberRepository memberRepository;
+
+    @Mock
+    MessageRepository messageRepository;
+
+
+    @Nested
+    class saveMessageTest {
+        long senderId = 1L;
+        long receiverId = 2L;
+        long artWorkId = 4L;
+        long chatRoomId = 4L;
+        Member sender;
+        Member receiver;
+        ArtWork artWork;
+        ChatRoom chatRoom;
+        MessageRequest messageRequest;
+
+        @BeforeEach
+        void init() {
+            sender = Member.builder()
+                    .id(senderId)
+                    .userId("sender")
+                    .password("12345678")
+                    .build();
+
+            receiver = Member.builder()
+                    .id(receiverId)
+                    .userId("receiver")
+                    .password("12345678")
+                    .build();
+
+            artWork = ArtWork.builder()
+                    .id(artWorkId)
+                    .build();
+
+            chatRoom = ChatRoom.builder()
+                    .id(chatRoomId)
+                    .artist(sender)
+                    .collector(receiver)
+                    .artWork(artWork)
+                    .build();
+
+            messageRequest = MessageRequest.builder()
+                    .chatRoomId(chatRoomId)
+                    .message("test")
+                    .build();
+        }
+
+        @Test
+        void success() {
+            //given
+            when(memberRepository.findById(senderId))
+                    .thenReturn(Optional.of(sender));
+
+            when(chatRoomRepository.findById(chatRoomId))
+                    .thenReturn(Optional.of(chatRoom));
+
+            when(messageRepository.save(any(Message.class)))
+                    .thenReturn(Message.builder().build());
+
+            //when
+            //then
+            assertThatNoException().isThrownBy(() -> messageService.saveMessage(senderId, messageRequest));
+        }
+
+        @Test
+        void failByNotFoundMember() {
+            //given
+            when(memberRepository.findById(senderId))
+                    .thenReturn(Optional.empty());
+
+            //when
+            //then
+            assertThatThrownBy(() -> messageService.saveMessage(senderId, messageRequest))
+                    .isInstanceOf(CustomException.class)
+                    .message().isEqualTo(ErrorCode.NOT_FOUND_MEMBER.getDetail());
+        }
+
+        @Test
+        void failByNotFoundChatRoom() {
+            //given
+            when(memberRepository.findById(senderId))
+                    .thenReturn(Optional.of(sender));
+
+            when(chatRoomRepository.findById(chatRoomId))
+                    .thenReturn(Optional.empty());
+
+            //when
+            //then
+            assertThatThrownBy(() -> messageService.saveMessage(senderId, messageRequest))
+                    .isInstanceOf(CustomException.class)
+                    .message().isEqualTo(ErrorCode.NOT_FOUND_CHAT_ROOM.getDetail());
+        }
+    }
+}