#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.List;
 import java.util.stream.Collectors;
 
@@ -40,14 +40,12 @@ public ModelAndView room() {
  */
  @PostMapping("/rooms")
  @ResponseBody
- public List<RoomDTO> createRoom(RoomDTO roomDTO, Model model) {
- model.addAttribute("roomDTO", roomDTO);
+ public List<RoomDTO> createRoom(RoomDTO roomDTO) {
  String roomName = roomDTO.getRoomName();
-
  RoomDTO room = null;
+
  if (roomName != null && !roomName.trim().equals("")) {
- int roomNumber = 0;
- roomNumber = chattingService.getLastRoomNumber();
+ int roomNumber = chattingService.getLastRoomNumber();
  room = RoomDTO.builder()
  .roomNumber(++roomNumber)
  .roomName(roomName)
@@ -76,8 +74,7 @@ public List<RoomDTO> getRoom(@RequestBody RoomDTO roomDTO) {
  * @return
  */
  @GetMapping("/chatting")
- public ModelAndView moveRoom(RoomDTO roomDTO, Model model) {
- model.addAttribute("roomDTO", roomDTO);
+ public ModelAndView moveRoom(RoomDTO roomDTO) {
  ModelAndView mv = new ModelAndView();
  int roomNumber = roomDTO.getRoomNumber();
 

src/main/java/com/market/server/handler/SocketHandler.java

@@ -26,6 +26,8 @@
 public class SocketHandler extends TextWebSocketHandler {
 
  private List<HashMap<String, Object>> rls = new ArrayList<>(); //웹소켓 세션을 담아둘 리스트 ---roomListSessions
+ private static final int DEFAULT_FILE_UPLOAD_INDEX = 0;
+ private static final String FILE_DIRECTORY_NAME = System.getProperty("user.dir") + File.separator + "attachFile" + File.separator;
 
  @Override
  public void handleTextMessage(WebSocketSession session, TextMessage message) {
@@ -65,31 +67,31 @@ public void handleTextMessage(WebSocketSession session, TextMessage message) {
  @SneakyThrows
  @Override
  public void handleBinaryMessage(WebSocketSession session, BinaryMessage message) {
- int fileUploadIdx = 0;
  //바이너리 메시지 발송
  ByteBuffer byteBuffer = message.getPayload();
  String fileName = DateUtil.getNowTimeToyyyyMMddHHmm(new Date(), ".jpg");
- String fileDirectoryName = System.getProperty("user.dir") + File.separator + "attachFile" + File.separator;
- File file = new File(fileDirectoryName, fileName);
+ File file = new File(FILE_DIRECTORY_NAME, fileName);
+ File dir = new File(FILE_DIRECTORY_NAME);
 
- File dir = new File(fileDirectoryName);
  if (!dir.exists()) {
  dir.mkdirs();
  }
 
- try (FileOutputStream out = new FileOutputStream(file, true); //생성을 위해 OutputStream을 연다.
- FileChannel outChannel = out.getChannel(); //채널을 열고)
+ try (
+ FileOutputStream out = new FileOutputStream(file, true); //생성을 위해 OutputStream을 연다.
+ FileChannel outChannel = out.getChannel(); //채널을 열고
  ) {
  byteBuffer.flip(); //byteBuffer를 읽기 위해 세팅
  byteBuffer.compact(); //파일을 복사한다.
  outChannel.write(byteBuffer); //파일을 쓴다.
- } catch (Exception e) {
+ } catch (IOException e) {
+ e.printStackTrace();
  log.error("파일 전송 실패", e);
  }
 
  byteBuffer.position(0); //파일을 저장하면서 position값이 변경되었으므로 0으로 초기화한다.
  //파일쓰기가 끝나면 이미지를 발송한다.
- HashMap<String, Object> temp = rls.get(fileUploadIdx);
+ HashMap<String, Object> temp = rls.get(DEFAULT_FILE_UPLOAD_INDEX);
  for (String k : temp.keySet()) {
  if (k.equals("roomNumber")) {
  continue;
@@ -98,6 +100,7 @@ public void handleBinaryMessage(WebSocketSession session, BinaryMessage message)
  try {
  wss.sendMessage(new BinaryMessage(byteBuffer)); //초기화된 버퍼를 발송한다.
  } catch (IOException e) {
+ e.printStackTrace();
  log.error("sendMessage 실패", e);
  }
  }

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

@@ -12,12 +12,16 @@
 @Log4j2
 public class ChattingService {
  @Autowired
- private ChattingMapper chattingMapper;
+ private ChattingMapper chattingRoomMapper;
+
+ private static final String DEFAULT_ROOMS_SORT_STATUS = "NEWEST";
+ private static final int DEFAULT_SEARCH_COUNT = 30;
+ private static final int DEFAULT_PAGING_OFFSET = 0;
 
  // 채팅방 등록
  public void register(RoomDTO roomDTO) {
  if (roomDTO != null) {
- chattingMapper.register(roomDTO);
+ chattingRoomMapper.register(roomDTO);
  } else {
  log.error("register ERROR! {}", roomDTO);
  throw new RuntimeException("register ERROR! 채팅방 등록 메서드를 확인해주세요\n" + "Params : " + roomDTO);
@@ -28,15 +32,14 @@ public void register(RoomDTO roomDTO) {
  public List<RoomDTO> getAllRooms(RoomDTO roomDTO) {
  List<RoomDTO> roomDTOList = null;
  if (roomDTO == null)
- roomDTOList = chattingMapper.selectRooms("NEWEST", 30, 0);
+ roomDTOList = chattingRoomMapper.selectRooms(DEFAULT_ROOMS_SORT_STATUS, DEFAULT_SEARCH_COUNT, DEFAULT_PAGING_OFFSET);
  else
- roomDTOList = chattingMapper.selectRooms(roomDTO.getSortStatus().toString(), roomDTO.getSearchCount(), roomDTO.getPagingStartOffset());
+ roomDTOList = chattingRoomMapper.selectRooms(roomDTO.getSortStatus().toString(), roomDTO.getSearchCount(), roomDTO.getPagingStartOffset());
  return roomDTOList;
  }
 
  public Integer getLastRoomNumber() {
- Integer result = -2;
- result = chattingMapper.getLastRoomNumber();
+ Integer result = chattingRoomMapper.getLastRoomNumber();
  if (result == null)
  result = 0;
  return result;

src/main/java/com/market/server/utils/DateUtil.java

@@ -1,9 +1,12 @@
 package com.market.server.utils;
 
+import lombok.extern.log4j.Log4j2;
+
 import java.text.SimpleDateFormat;
 import java.util.Date;
 import java.text.ParseException;
 
+@Log4j2
 public class DateUtil {
 
  /**
@@ -13,8 +16,14 @@ public class DateUtil {
  * @return 처리 시간의 문자열 202009040159.jpg
  * @author topojs8
  */
- public static String getNowTimeToyyyyMMddHHmm(Date date, String fileType) {
- return new SimpleDateFormat("yyyyMMddHHmm").format(date) + fileType;
+ private static final ThreadLocal<SimpleDateFormat> tl = new ThreadLocal<SimpleDateFormat>();
+ public static final String getNowTimeToyyyyMMddHHmm(Date date, String fileType) {
+ SimpleDateFormat sdf = tl.get();
+ if(sdf == null) {
+ sdf = new SimpleDateFormat("yyyyMMddHHmm");
+ tl.set(sdf);
+ }
+ return sdf.format(date) + fileType;
  }
 
  /**
@@ -28,9 +37,10 @@ private static void parseDate(String dateStr) {
  try {
  SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMddHHmm");
  Date date = simpleDateFormat.parse(dateStr);
+ log.info("Successfully Parsed Date " + date);
  System.out.println("Successfully Parsed Date " + date);
  } catch (ParseException e) {
- System.out.println("ParseError " + e.getMessage());
+ log.error("ParseError", e);
  } catch (Exception e) {
  e.printStackTrace();
  }

src/main/resources/application-dev.properties

@@ -15,6 +15,7 @@ market.server.redis.port=6379
 market.server.redis.password=
 
 # expire
+# millisecond 600������ �� 10�ð�
 expire.defaultTime=36288000
 
 #JSP, HTML ModelAndView Path Setting