feat: 채팅 기능 기본 구성 추가 (#112)

Author: suubinkim

build.gradle

@@ -64,7 +64,10 @@ dependencies {
     testImplementation 'org.springframework.boot:spring-boot-starter-test'
     testImplementation 'org.springframework.security:spring-security-test'
     developmentOnly 'org.springframework.boot:spring-boot-devtools'
+    //웹소켓
     implementation 'org.springframework.boot:spring-boot-starter-websocket'
+    implementation 'org.webjars:stomp-websocket:2.3.3-1'
+    implementation 'org.webjars:sockjs-client:1.1.2'
 }
 
 jar {

src/main/java/shop/fevertime/backend/api/AlarmApiController.java renamed to src/main/java/shop/fevertime/backend/api/ChatApiController.java

@@ -5,17 +5,20 @@
 import org.springframework.messaging.simp.SimpMessageSendingOperations;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.RequestBody;
+import shop.fevertime.backend.dto.ChatMessageDto;
 import shop.fevertime.backend.dto.MessageDto;
 
 @Controller
 @RequiredArgsConstructor
-public class AlarmApiController {
+public class ChatApiController {
 
     private final SimpMessageSendingOperations messagingTemplate;
 
-    @MessageMapping("/post")
-    public void addUser(@RequestBody MessageDto dto) {
-        messagingTemplate.convertAndSend("/topic/feed/"+ dto.getUserId(), dto);
+    @MessageMapping("/chat/message")
+    public void addUser(@RequestBody ChatMessageDto messageDto) {
+        if (ChatMessageDto.MessageType.ENTER.equals(messageDto.getType()))
+            messageDto.setMessage(messageDto.getSender() + "님이 입장하셨습니다.");
+        messagingTemplate.convertAndSend("/sub/chat/room/" + messageDto.getRoomId(), messageDto);
     }
 
 }

src/main/java/shop/fevertime/backend/api/ChatRoomApiController.java

@@ -0,0 +1,53 @@
+package shop.fevertime.backend.api;
+
+import lombok.RequiredArgsConstructor;
+import org.springframework.security.core.annotation.AuthenticationPrincipal;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.*;
+import shop.fevertime.backend.domain.ChatRoom;
+import shop.fevertime.backend.dto.ChatRoomDto;
+import shop.fevertime.backend.dto.response.ResultResponseDto;
+import shop.fevertime.backend.repository.ChatRoomRepository;
+import shop.fevertime.backend.security.UserDetailsImpl;
+import shop.fevertime.backend.service.ChatRoomService;
+
+import java.util.List;
+
+@RequiredArgsConstructor
+@RestController
+@RequestMapping("/chat")
+public class ChatRoomApiController {
+
+    private final ChatRoomRepository chatRoomRepository;
+    private final ChatRoomService chatRoomService;
+
+    /**
+     * 채팅방 목록 조회 API
+     */
+    //반환값 dto로 변경해야함
+    @GetMapping("/rooms")
+    public List<ChatRoom> getRooms() {
+        return chatRoomService.getAllRooms();
+    }
+
+    /**
+     * 채팅방 생성 API
+     */
+    @PostMapping("/room")
+    public ResultResponseDto createRoom(@RequestBody ChatRoomDto chatRoomDto,
+                                        @AuthenticationPrincipal UserDetailsImpl userDetails) {
+        chatRoomService.createRoom(chatRoomDto, userDetails.getUser());
+        return new ResultResponseDto("success", "채팅 방이 생성되었습니다.");
+    }
+
+    /**
+     * 채팅방 삭제 API
+     */
+    @DeleteMapping("/room/{roomId}")
+    public ResultResponseDto deleteRoom(@PathVariable Long roomId,
+                           @AuthenticationPrincipal UserDetailsImpl userDetails) {
+        chatRoomService.deleteRoom(roomId, userDetails.getUser());
+        return new ResultResponseDto("success", "채팅 방이 삭제되었습니다.");
+    }
+
+}

src/main/java/shop/fevertime/backend/config/StompHandler.java

@@ -13,21 +13,14 @@
 @EnableWebSocketMessageBroker // EnableWebsocket 알아보기. , 자료구조 Queue, STOMP 검색해서 통신 방법 알아보기
 public class WebSocketConfig implements WebSocketMessageBrokerConfigurer { // websoket.io, 만약 댓글을 10만명이 달게 된다면?, port를 잡는다. 확장이 어렵다.
 
-    private final StompHandler stompHandler; // jwt 토큰 인증 핸들러
-
     @Override
     public void configureMessageBroker(MessageBrokerRegistry registry) {
-        registry.enableSimpleBroker("/topic"); // 메모리 기반 메세지 브로커가 해당 api를 구독하고 있는 클라이언트에게 메세지를 전달한다.
+        registry.enableSimpleBroker("/sub"); // 메모리 기반 메세지 브로커가 해당 api를 구독하고 있는 클라이언트에게 메세지를 전달한다.
         registry.setApplicationDestinationPrefixes("/pub"); // 서버에서 클라이언트로부터의 메세지를 받을 api를 prefix 설정한다.
     }
 
     @Override
     public void registerStompEndpoints(StompEndpointRegistry registry) { // 클라이언트에서 websoket을 연결할 api를 설정한다.
-        registry.addEndpoint("/websocket").setAllowedOrigins("https://api.fevertime.shop", "https://www.fevertime.shop", "http://localhost:8080", "http://localhost:63342").withSockJS();
-    }
-
-    @Override
-    public void configureClientInboundChannel(ChannelRegistration registration) {// channel: 데이터가 흘러 들어오는 공간, 채널에 대한 필터, 패턴 매칭, 조건 확인
-        registration.interceptors(stompHandler); //핸들러 등록
+        registry.addEndpoint("/ws-stomp").setAllowedOrigins("https://api.fevertime.shop", "https://www.fevertime.shop", "http://localhost:8080", "http://localhost:63342").withSockJS();
     }
 }

src/main/java/shop/fevertime/backend/config/WebSocketConfig.java

@@ -0,0 +1,31 @@
+package shop.fevertime.backend.domain;
+
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import shop.fevertime.backend.dto.ChatRoomDto;
+
+import javax.persistence.*;
+
+@Getter
+@NoArgsConstructor
+@Entity
+public class ChatRoom extends BaseTimeEntity {
+
+    @Id // ID 값, Primary Key로 사용하겠다는 뜻입니다.
+    @GeneratedValue(strategy = GenerationType.AUTO) // 자동 증가 명령입니다.
+    @Column(name = "room_id")
+    private Long id;
+
+    @Column // 컬럼 값이고 반드시 값이 존재해야 함을 나타냅니다.
+    private String name;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "user_id")
+    private User user;
+
+    public ChatRoom(ChatRoomDto chatRoomDto, User user) {
+        this.name = chatRoomDto.getName();
+        this.user = user;
+    }
+
+}

src/main/java/shop/fevertime/backend/domain/ChatRoom.java

@@ -0,0 +1,17 @@
+package shop.fevertime.backend.dto;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class ChatMessageDto {
+    //메세지 타입 : 입장, 채팅
+    public enum MessageType {
+        ENTER, TALK
+    }
+    private MessageType type; //메세지 타입
+    private String roomId;
+    private String sender;
+    private String message;
+}

src/main/java/shop/fevertime/backend/dto/ChatMessageDto.java

@@ -0,0 +1,10 @@
+package shop.fevertime.backend.dto;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class ChatRoomDto {
+    private String name;
+}

src/main/java/shop/fevertime/backend/dto/ChatRoomDto.java

@@ -0,0 +1,11 @@
+package shop.fevertime.backend.repository;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+import shop.fevertime.backend.domain.ChatRoom;
+import shop.fevertime.backend.domain.User;
+
+@Repository
+public interface ChatRoomRepository extends JpaRepository<ChatRoom, Long> {
+    ChatRoom findByIdAndUser(Long roomId, User user);
+}

src/main/java/shop/fevertime/backend/repository/ChatRoomRepository.java

@@ -0,0 +1,38 @@
+package shop.fevertime.backend.service;
+
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.data.domain.Sort;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import shop.fevertime.backend.domain.ChatRoom;
+import shop.fevertime.backend.domain.User;
+import shop.fevertime.backend.dto.ChatRoomDto;
+import shop.fevertime.backend.repository.ChatRoomRepository;
+
+import java.util.List;
+
+@Slf4j
+@RequiredArgsConstructor
+@Service
+public class ChatRoomService {
+
+    private final ChatRoomRepository chatRoomRepository;
+
+    //반환값 dto로 바꿔야함
+    public List<ChatRoom> getAllRooms() {
+        return chatRoomRepository.findAll(Sort.by(Sort.Direction.DESC, "createdDate"));
+    }
+
+    @Transactional
+    public void createRoom(ChatRoomDto chatRoomDto, User user) {
+        chatRoomRepository.save(new ChatRoom(chatRoomDto, user));
+    }
+
+    @Transactional
+    public void deleteRoom(Long roomId, User user) {
+        ChatRoom chatRoom = chatRoomRepository.findByIdAndUser(roomId, user);
+        chatRoomRepository.delete(chatRoom);
+    }
+
+}