diff --git a/src/main/java/kr/co/fastcampus/yanabada/common/config/WebSockConfig.java b/src/main/java/kr/co/fastcampus/yanabada/common/config/WebSockConfig.java index 0f19a118..adc8f7b7 100644 --- a/src/main/java/kr/co/fastcampus/yanabada/common/config/WebSockConfig.java +++ b/src/main/java/kr/co/fastcampus/yanabada/common/config/WebSockConfig.java @@ -1,15 +1,21 @@ package kr.co.fastcampus.yanabada.common.config; +import kr.co.fastcampus.yanabada.common.handler.StompHandler; +import lombok.RequiredArgsConstructor; import org.springframework.context.annotation.Configuration; +import org.springframework.messaging.simp.config.ChannelRegistration; import org.springframework.messaging.simp.config.MessageBrokerRegistry; import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker; import org.springframework.web.socket.config.annotation.StompEndpointRegistry; import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer; @Configuration +@RequiredArgsConstructor @EnableWebSocketMessageBroker public class WebSockConfig implements WebSocketMessageBrokerConfigurer { + private final StompHandler stompHandler; + @Override public void configureMessageBroker(MessageBrokerRegistry config) { config.enableSimpleBroker("/sub"); @@ -20,4 +26,9 @@ public void configureMessageBroker(MessageBrokerRegistry config) { public void registerStompEndpoints(StompEndpointRegistry registry) { registry.addEndpoint("/ws-stomp").setAllowedOriginPatterns("*"); } + + @Override + public void configureClientInboundChannel(ChannelRegistration registration) { + registration.interceptors(stompHandler); + } }