diff --git a/vital-signs/src/main/java/com/rax/vital/interceptor/WebSocketInterceptors.java b/vital-signs/src/main/java/com/rax/vital/interceptor/WebSocketInterceptors.java index 1c79f16..6c28b7a 100644 --- a/vital-signs/src/main/java/com/rax/vital/interceptor/WebSocketInterceptors.java +++ b/vital-signs/src/main/java/com/rax/vital/interceptor/WebSocketInterceptors.java @@ -1,8 +1,10 @@ package com.rax.vital.interceptor; +import jakarta.servlet.http.HttpSession; import org.springframework.http.server.ServerHttpRequest; import org.springframework.http.server.ServerHttpResponse; import org.springframework.http.server.ServletServerHttpRequest; +import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.server.HandshakeInterceptor; @@ -13,7 +15,10 @@ public class WebSocketInterceptors implements HandshakeInterceptor { public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map attributes) throws Exception { if (request instanceof ServletServerHttpRequest) { ServletServerHttpRequest servletServerHttpRequest = (ServletServerHttpRequest) request; - servletServerHttpRequest. + HttpSession session = servletServerHttpRequest.getServletRequest().getSession(false); + if (session != null) { + SecurityContextHolder.getContext().getAuthentication().getPrincipal(); + } } return false; }