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