rax/rax-remote-v2
3
0
Fork 0
mirror of https://gitee.com/republicline/rax-remote-v2.git synced 2025-08-24 20:04:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

temp

This commit is contained in:
zhaoyz 2024-04-12 10:28:03 +08:00
parent 5ed80a5b82
commit 23e1470ea6
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vital-signs/src/main/java/com/rax/vital/interceptor/WebSocketInterceptors.java
View File

@ -1,8 +1,10 @@
package com.rax.vital.interceptor; package com.rax.vital.interceptor;
import jakarta.servlet.http.HttpSession;
import org.springframework.http.server.ServerHttpRequest; import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse; import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest; import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.HandshakeInterceptor; 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<String, Object> attributes) throws Exception { public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
if (request instanceof ServletServerHttpRequest) { if (request instanceof ServletServerHttpRequest) {
ServletServerHttpRequest servletServerHttpRequest = (ServletServerHttpRequest) request; ServletServerHttpRequest servletServerHttpRequest = (ServletServerHttpRequest) request;
servletServerHttpRequest. HttpSession session = servletServerHttpRequest.getServletRequest().getSession(false);
if (session != null) {
SecurityContextHolder.getContext().getAuthentication().getPrincipal();
}
} }
return false; return false;
} }