tt/OneKeyCallVideoTablet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化平板连接websocket

Browse Source
This commit is contained in:
tongtongstudio
2025-10-10 11:10:56 +08:00
parent 1c8fc0c811
commit b9d0deddd8
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/onekeycall/videotablet/config/WebSocketConfig.java
View File

@@ -21,7 +21,7 @@ public class WebSocketConfig implements WebSocketConfigurer {
@Override @Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(webSocketHandler, "/ws") // 客户端连接端点 registry.addHandler(webSocketHandler, "/ws/tablet_ws") // 客户端连接端点
.setAllowedOrigins("*") // 允许跨域 .setAllowedOrigins("*") // 允许跨域
.addInterceptors(new AuthHandshakeInterceptor(jwtUtil)); // 握手拦截器（如JWT校验） .addInterceptors(new AuthHandshakeInterceptor(jwtUtil)); // 握手拦截器（如JWT校验）
} }

8
src/main/java/com/onekeycall/videotablet/interceptor/AuthHandshakeInterceptor.java
View File

@@ -24,7 +24,6 @@ public class AuthHandshakeInterceptor 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 {
String authHeader = request.getHeaders().getFirst("Authorization"); String authHeader = request.getHeaders().getFirst("Authorization");
String deviceId = request.getHeaders().getFirst("Device-ID"); String deviceId = request.getHeaders().getFirst("Device-ID");
String userId = request.getHeaders().getFirst("User-ID");
if (request instanceof ServletServerHttpRequest) { if (request instanceof ServletServerHttpRequest) {
ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request; ServletServerHttpRequest servletRequest = (ServletServerHttpRequest) request;
@@ -32,13 +31,14 @@ public class AuthHandshakeInterceptor implements HandshakeInterceptor {
String sn = httpRequest.getParameter("sn"); String sn = httpRequest.getParameter("sn");
attributes.put("sn", sn); attributes.put("sn", sn);
log.info("Intercepted - sn: " + sn); log.info("Intercepted - sn: " + sn);
}
if (authHeader == null || !authHeader.startsWith("Bearer ") || userId == null) { if (authHeader == null || !authHeader.startsWith("Bearer ")) {
return false; return false;
} }
String token = authHeader.substring(7); // 去掉 "Bearer " 前缀 String token = authHeader.substring(7); // 去掉 "Bearer " 前缀
return jwtUtil.validateAccessToken(userId, token, deviceId); // 自定义校验逻辑 return jwtUtil.validateDeviceToken(token, deviceId, sn); // 自定义校验逻辑
}
return false;
} }
@Override @Override