[java] 자바 웹소켓과 멀티스레드 처리

11 Dec 2023

java

이번에는 자바를 사용하여 웹소켓 통신을 구현하고 멀티스레드를 효율적으로 처리하는 방법에 대해 알아보겠습니다. 웹소켓은 실시간 양방향 통신을 지원하며, 멀티스레드를 사용하여 다수의 클라이언트 요청을 동시에 처리할 수 있습니다.

자바 웹소켓 구현

우선 웹소켓을 구현하기 위해 javax.websocket 패키지를 사용하겠습니다. 다음은 간단한 웹소켓 엔드포인트의 예시입니다.

import javax.websocket.OnMessage;
import javax.websocket.server.ServerEndpoint;
import javax.websocket.OnOpen;
import javax.websocket.OnClose;
import javax.websocket.Session;
import java.io.IOException;

@ServerEndpoint("/websocket")
public class WebSocketEndpoint {
    @OnOpen
    public void onOpen(Session session) {
        // 클라이언트 연결시 작업
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        // 메시지 수신시 작업
    }

    @OnClose
    public void onClose(Session session) {
        // 클라이언트 연결 종료시 작업
    }
}

위의 코드는 @ServerEndpoint 어노테이션을 사용하여 웹소켓 엔드포인트를 정의하고, 연결(OnOpen), 메시지 수신(OnMessage), 연결 종료(OnClose) 시의 동작을 정의하고 있습니다.

멀티스레드 처리

여러 클라이언트의 동시 요청을 처리하기 위해서는 멀티스레드로 작업을 처리해야 합니다. 자바에서 멀티스레드를 처리하는 방법 중 하나는 ExecutorService를 사용하는 것입니다. 아래는 간단한 예제 코드입니다.

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class WebSocketHandler {

    private final ExecutorService executor = Executors.newFixedThreadPool(10);

    public void processRequest(Session session, String message) {
        executor.submit(() -> {
            // 멀티스레드로 처리할 작업 수행
        });
    }
}

위의 코드에서 ExecutorService를 사용하여 고정 크기의 스레드 풀을 생성하고, processRequest 메소드를 호출할 때마다 해당 작업을 스레드 풀에 제출합니다.

이제 자바 웹소켓을 구현하고 멀티스레드를 효율적으로 처리하는 방법에 대해 알아보았습니다. 실무 환경에서는 더 복잡한 상황에 대비하여 보다 신중한 설계가 필요할 것입니다.