admin/timeline-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: 修改 XUserIdHandshakeInterceptor 以支持 Bearer 前缀
All checks were successful
test/timeline-server/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
jianghao
2026-02-11 15:11:08 +08:00
parent 482c32a59c
commit 7972fa775a
1 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
timeline-user-service/src/main/java/com/timeline/user/config/XUserIdHandshakeInterceptor.java
View File

@@ -20,8 +20,6 @@ import java.util.Map;
@Slf4j
public class XUserIdHandshakeInterceptor implements HandshakeInterceptor {
private static final org.slf4j.Logger log = org.slf4j.LoggerFactory.getLogger(XUserIdHandshakeInterceptor.class);
// 从配置中获取JWT密钥
@Value("${jwt.secret:timelineSecretKey}")
private String jwtSecret = "timelineSecretKey";
@@ -35,9 +33,9 @@ public class XUserIdHandshakeInterceptor implements HandshakeInterceptor {
@Override
public boolean beforeHandshake(@NonNull ServerHttpRequest request,
@NonNull ServerHttpResponse response,
@NonNull WebSocketHandler wsHandler,
@NonNull Map<String, Object> attributes) {
@NonNull ServerHttpResponse response,
@NonNull WebSocketHandler wsHandler,
@NonNull Map<String, Object> attributes) {
if (request instanceof ServletServerHttpRequest servletRequest) {
HttpServletRequest httpServletRequest = servletRequest.getServletRequest();
@@ -75,6 +73,12 @@ public class XUserIdHandshakeInterceptor implements HandshakeInterceptor {
try {
// URL 解码
token = java.net.URLDecoder.decode(token, java.nio.charset.StandardCharsets.UTF_8);
// 移除 Bearer 前缀（如果存在）
if (token.startsWith("Bearer ")) {
token = token.substring(7);
}
Claims claims = JwtUtils.parseToken(token, jwtSecret);
if (claims != null) {
userId = claims.get("userId", String.class);
@@ -109,9 +113,9 @@ public class XUserIdHandshakeInterceptor implements HandshakeInterceptor {
@Override
public void afterHandshake(@NonNull ServerHttpRequest request,
@org.springframework.lang.Nullable ServerHttpResponse response,
@NonNull WebSocketHandler wsHandler,
@org.springframework.lang.Nullable Exception exception) {
@org.springframework.lang.Nullable ServerHttpResponse response,
@NonNull WebSocketHandler wsHandler,
@org.springframework.lang.Nullable Exception exception) {
// 在握手完成后，检查userId是否已设置
if (request instanceof ServletServerHttpRequest servletRequest) {
String userId = (String) servletRequest.getServletRequest().getAttribute("userId");
@@ -123,7 +127,6 @@ public class XUserIdHandshakeInterceptor implements HandshakeInterceptor {
}
}
public static class UserPrincipal implements Principal {
private final String name;