#30 웹소켓을 이용한 1:1 채팅 구현

src/main/java/com/market/server/controller/ChattingController.java

@@ -3,9 +3,9 @@
 import com.market.server.service.ChattingService;
 import io.swagger.annotations.Api;
 import lombok.extern.log4j.Log4j2;
+import org.springframework.ui.Model;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.servlet.ModelAndView;
-import java.util.HashMap;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -40,19 +40,22 @@ public ModelAndView room() {
  */
  @PostMapping("/rooms")
  @ResponseBody
- public List<RoomDTO> createRoom(@RequestParam HashMap<String, String> params) {
- String roomName = (String) params.get("roomName");
- RoomDTO roomDTO = null;
+ public List<RoomDTO> createRoom(RoomDTO roomDTO, Model model) {
+ model.addAttribute("roomDTO", roomDTO);
+ String roomName = roomDTO.getRoomName();
+
+ RoomDTO room = null;
  if (roomName != null && !roomName.trim().equals("")) {
- int roomNumber = chattingService.getLastRoomNumber();
- roomDTO = RoomDTO.builder()
+ int roomNumber = 0;
+ roomNumber = chattingService.getLastRoomNumber();
+ room = RoomDTO.builder()
  .roomNumber(++roomNumber)
  .roomName(roomName)
  .build();
- chattingService.register(roomDTO);
+ chattingService.register(room);
  }
 
- return chattingService.getAllRooms(roomDTO);
+ return chattingService.getAllRooms(room);
  }
 
  /**
@@ -73,19 +76,19 @@ public List<RoomDTO> getRoom(@RequestBody RoomDTO roomDTO) {
  * @return
  */
  @GetMapping("/chatting")
- public ModelAndView moveRoom(@RequestParam HashMap<String, String> params) {
+ public ModelAndView moveRoom(RoomDTO roomDTO, Model model) {
+ model.addAttribute("roomDTO", roomDTO);
  ModelAndView mv = new ModelAndView();
- int roomNumber = Integer.parseInt((String) params.get("roomNumber"));
+ int roomNumber = roomDTO.getRoomNumber();
 
  List<RoomDTO> new_list = chattingService.getAllRooms(null).stream().filter(o -> o.getRoomNumber() == roomNumber).collect(Collectors.toList());
  if (new_list != null && new_list.size() > 0) {
- mv.addObject("roomName", params.get("roomName"));
- mv.addObject("roomNumber", params.get("roomNumber"));
+ mv.addObject("roomName", roomDTO.getRoomName());
+ mv.addObject("roomNumber", roomDTO.getRoomNumber());
  mv.setViewName("chat");
  } else {
  mv.setViewName("room");
  }
  return mv;
  }
-
 }

src/main/java/com/market/server/mapper/ChattingMapper.java

@@ -9,5 +9,5 @@
 public interface ChattingMapper {
  public int register(RoomDTO roomDTO);
  public List<RoomDTO> selectRooms(String sortStatus, int searchCount, int pagingStartOffset);
- public int getLastRoomNumber();
+ public Integer getLastRoomNumber();
 }

src/main/java/com/market/server/service/ChattingService.java

@@ -28,14 +28,18 @@ public void register(RoomDTO roomDTO) {
  public List<RoomDTO> getAllRooms(RoomDTO roomDTO) {
  List<RoomDTO> roomDTOList = null;
  if (roomDTO == null)
- roomDTOList = chattingMapper.selectRooms("NEWEST", 20, 0);
+ roomDTOList = chattingMapper.selectRooms("NEWEST", 30, 0);
  else
  roomDTOList = chattingMapper.selectRooms(roomDTO.getSortStatus().toString(), roomDTO.getSearchCount(), roomDTO.getPagingStartOffset());
  return roomDTOList;
  }
 
- public int getLastRoomNumber() {
- return chattingMapper.getLastRoomNumber();
+ public Integer getLastRoomNumber() {
+ Integer result = -2;
+ result = chattingMapper.getLastRoomNumber();
+ if (result == null)
+ result = 0;
+ return result;
  }
 
 }

src/main/resources/banner.txt

@@ -0,0 +1,8 @@
+
+ ____. _________.__ __ .________
+ | |__ __ ____ / _____/| |__ ____ ____ | | __ | ____/
+ | | | \/ \ \_____ \ | | \ / _ \_/ ___\| |/ / |____ \
+/\__| | | / | \/ \| Y ( <_> ) \___| < / \
+\________|____/|___| /_______ /|___| /\____/ \___ >__|_ \/______ /
+ \/ \/ \/ \/ \/ \/
+application.title:${application.titlke}