[SpringBoot] 웹소켓을 이용한 채팅 구현

HTTP vs 웹소켓

HTTP는 요청과 응답 구조로 통신이 이뤄진다. 

실시간 서비스를 제공하기 위해서는 매번 새로운 요청이 필요해 자원이 많이 필요하다.

 

반면, 웹소켓은 한 번 열결 맺은 뒤 지속적으로 통신이 가능하다.

실시간성 서비스에 웹소켓을 이용하면 적은 자원으로 통신이 가능하다. 

http:// 가 아닌 ws:// 로 요청을 보낸다.

 

---

 

바로 구현해보자!

 

 

 

프로젝트 설정 및 의존성 추가

implementation 'org.springframework.boot:spring-boot-starter-websocket'

 

WebSocketConfig 클래스 생성

import lombok.RequiredArgsConstructor;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebSocket
@RequiredArgsConstructor
public class WebSocketConfig implements WebSocketConfigurer {

    private final WebSocketHandler webSocketHandler;

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(webSocketHandler, "/chat")
        .setAllowedOrigins("*");
    }
}

 

WebSocket을 활성화하며

'/chat' 경로로 들어오는 WebSocket 연결을 처리할 핸들러(webSocketHandler)를 등록하고 설정한다.

 

WebSocketHandler 클래스 생성

클라이언트가 발송한 메세지를 받아 처리해줄 Handler이다. 

텍스트 기반의 WebScoket 메세지를 처리하는 TextWebSocketHandler를 상속받아 작성한다.

import org.springframework.stereotype.Component;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

@Component
public class WebSocketHandler extends TextWebSocketHandler {

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception{
        String payload = message.getPayload();
        TextMessage textMessage = new TextMessage("Welcome chatting server");
        session.sendMessage(textMessage);
    }

}

 

handleTextMessage는 메세지를 보낼 때 사용할 클라이언트의 WebSocketSession과

수신된 메세지(TextMessage)를 매개 변수로 받는다.

 

session.sendMessage()를 통해서는 메세지를 발신할 수 있다.

 

 

테스트

postman에서 WebSocket 테스트를 할 수 있다.

왼쪽 상단에 New를 누르고 WebSocket을 선택하면 된다.  

 

안녕!!! 메세지를 보내니 기존 session.sendMessage() 안에 들어가 있었던 welcom chatting server가 출력되었다.

정상적으로 작동된다👍👍