From afe8b26eeeb63503a21897d5f4b0899c2e9bd73c Mon Sep 17 00:00:00 2001
From: laodaming <11058467+laudamine@user.noreply.gitee.com>
Date: Wed, 11 Oct 2023 18:37:47 +0800
Subject: [PATCH] fix

---
 server/websocket/internal/logic/datatransferlogic.go | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/server/websocket/internal/logic/datatransferlogic.go b/server/websocket/internal/logic/datatransferlogic.go
index 9050f898..beb5b5c8 100644
--- a/server/websocket/internal/logic/datatransferlogic.go
+++ b/server/websocket/internal/logic/datatransferlogic.go
@@ -129,12 +129,14 @@ func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request)
 			return
 		}
 		//有效token
-		if s[0] != "empty_token" {
-			r.Header.Set("Authorization", "Bearer "+s[0])
+		token := strings.Trim(s[0], " ")
+		if token != "empty_token" {
+			r.Header.Set("Authorization", "Bearer "+token)
 		}
 		//有效wid
-		if s[1] != "empty_wid" {
-			oldWid = s[1]
+		wid := strings.Trim(s[1], " ")
+		if wid != "empty_wid" {
+			oldWid = wid
 		}
 		//设置Sec-Websocket-Protocol
 		upgrader.Subprotocols = []string{secWebsocketProtocol}
@@ -197,7 +199,7 @@ func (l *DataTransferLogic) setConnPool(conn *websocket.Conn, userInfo *auth.Use
 		//解析传入的wid是不是属于自己的用户的
 		decryptionWid, err := encryption_decryption.CBCDecrypt(oldWid)
 		if err != nil {
-			logx.Error(err, ":", oldWid)
+			logx.Error(err)
 			return wsConnectItem{}, errors.New("解码wid失败")
 		}
 		lendecryptionWid := len(decryptionWid)