From feac84cdd8c1b27b213b1abb0423c8965b1d5d14 Mon Sep 17 00:00:00 2001
From: hxr <1490493387@qq.com>
Date: Mon, 13 May 2024 23:47:03 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20WebSocket=E8=A7=A3=E6=9E=90token?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/youlai/system/config/WebSocketConfig.java | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/youlai/system/config/WebSocketConfig.java b/src/main/java/com/youlai/system/config/WebSocketConfig.java
index 69245d7c..a128f13f 100644
--- a/src/main/java/com/youlai/system/config/WebSocketConfig.java
+++ b/src/main/java/com/youlai/system/config/WebSocketConfig.java
@@ -3,6 +3,7 @@ package com.youlai.system.config;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.jwt.JWTPayload;
 import cn.hutool.jwt.JWTUtil;
+import com.youlai.system.common.constant.SecurityConstants;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.http.HttpHeaders;
@@ -80,11 +81,8 @@ public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {
                     if (StrUtil.isNotBlank(bearerToken) && bearerToken.startsWith("Bearer ")) {
                         try {
                             // 移除 "Bearer " 前缀，从令牌中提取用户信息(username), 并设置到认证信息中
-                            
-                            // 这里不应该用"name"
-                            // String username = JwtUtils.parseToken(bearerToken).get("name").toString();
+                            bearerToken = bearerToken.substring(SecurityConstants.JWT_TOKEN_PREFIX.length());
                             String username = JWTUtil.parseToken(bearerToken).getPayloads().getStr(JWTPayload.SUBJECT);
-
                             if (StrUtil.isNotBlank(username)) {
                                 accessor.setUser(() -> username);
                                 return message;