채팅 토이 프로젝트를 진행하면서 공부했던 STOMP 관련 내용 정리 - dnwls16071/Backend_Summary GitHub Wiki

@Slf4j
@Configuration
@EnableWebSocket
@EnableWebSocketMessageBroker
@RequiredArgsConstructor
public class StompWebSocketConfig implements WebSocketMessageBrokerConfigurer {

	private final StompHandler stompHandler;

	@Override
	public void registerStompEndpoints(StompEndpointRegistry registry) {
		// Register STOMP endpoints mapping each to a specific URL and (optionally) enabling and configuring SockJS fallback options.
		registry.addEndpoint("/connect")
				.setAllowedOrigins("http://localhost:3000")
				.withSockJS();
	}

	@Override
	public void configureMessageBroker(MessageBrokerRegistry registry) {
		// Configure message broker options.
		// @MessageMapping이 있는 컨트롤러 메서드
		// 클라이언트 → 서버(애플리케이션)로 메시지를 보낼 때 사용되는 목적지 접두사를 설정합니다.
		registry.setApplicationDestinationPrefixes("/publish");

		// 메모리 기반의 내장 메시지 브로커
		// 서버 → 클라이언트(브로드캐스트)로 메시지를 보낼 때 사용되는 목적지 접두사를 설정합니다.
		registry.enableSimpleBroker("/topic");
	}

	@Override
	public void configureClientInboundChannel(ChannelRegistration registration) {
		// 클라이언트 메시지가 실제 핸들러에 도달하기 전에 메시지를 가로채고 처리
		// 클라이언트가 보낸 메시지에 대한 인증, 권한 부여, 로깅 등을 처리
		log.info("Interceptor - OK...");
		registration.interceptors(stompHandler);
	}
}

@Component
@RequiredArgsConstructor
public class StompHandler implements ChannelInterceptor {

	private final JwtDecoder jwtDecoder;

	@Override
	public Message<?> preSend(Message<?> message, MessageChannel channel) {
		// 메시지가 실제로 전송되기 전에 호출되는 사전 처리 단계를 의미
		final StompHeaderAccessor accessor = StompHeaderAccessor.wrap(message);

		System.out.println("StompHandler.preSend() called - Command: " + accessor.getCommand());

		if (StompCommand.CONNECT == accessor.getCommand()) {
			final String bearerToken = accessor.getFirstNativeHeader("Authorization");

			if (bearerToken == null) {
				System.out.println("No Authorization header found!");
				throw new IllegalArgumentException("Authorization header is missing");
			}

			final String token = bearerToken.substring(7);

			try {
				jwtDecoder.decode(token);
			} catch (Exception e) {
				throw new IllegalArgumentException("Invalid JWT token");
			}
		}
		return ChannelInterceptor.super.preSend(message, channel);
	}
}

@Controller
@RequiredArgsConstructor
public class ChatController {

	private final SimpMessagingTemplate messagingTemplate;

	// 클라이언트로부터 메시지를 수신하고 처리합니다.
	// /publish/chat/{roomId}로 메시지를 발행합니다.
	@MessageMapping("/chat/{roomId}")
	public void sendMessage(@DestinationVariable(value = "roomId") Long roomId, ChatRequest request) {
		messagingTemplate.convertAndSend("/topic/" + roomId, request);
	}
}

@Slf4j
@Component
public class StompEventListener {

	private final Set<String> sessions = ConcurrentHashMap.newKeySet();  // 동시성 이슈를 고려한 ConcurrentHashMap 사용

	@EventListener
	public void handleConnect(SessionConnectEvent event) {
		String sessionId = event.getMessage().getHeaders().get("simpSessionId").toString();
		sessions.add(sessionId);
		log.info("New WebSocket connection established. Session ID: {}. Total sessions: {}", sessionId, sessions.size());
	}

	@EventListener
	public void handleDisconnect(SessionDisconnectEvent event) {
		String sessionId = event.getSessionId();
		sessions.remove(sessionId);
		log.info("WebSocket connection closed. Session ID: {}. Total sessions: {}", sessionId, sessions.size());
	}
}