zhuyujun/go-desk-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改websockt服务鉴权

Browse Source
This commit is contained in:
zyj
2025-09-16 17:54:44 +08:00
parent c1b601de69
commit 25e0620a3f
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/websocks.go
View File

@@ -35,7 +35,7 @@ func (*Websocks) Init(ctx *gin.Context) {
// 判断当前连接是否合法 // 判断当前连接是否合法
key := http.CanonicalHeaderKey("sec-websocket-protocol") key := http.CanonicalHeaderKey("sec-websocket-protocol")
protos := ctx.Request.Header[key] protos := ctx.Request.Header[key]
if len(protos) > 0 { if len(protos) <= 0 {
ctx.JSON(http.StatusInternalServerError, gin.H{"code": WebsocktFailed.Code, "data": WebsocktFailed.Data, "msg": WebsocktFailed.Msg}) ctx.JSON(http.StatusInternalServerError, gin.H{"code": WebsocktFailed.Code, "data": WebsocktFailed.Data, "msg": WebsocktFailed.Msg})
return return
} }
@@ -60,7 +60,6 @@ func (*Websocks) Init(ctx *gin.Context) {
if err != nil { if err != nil {
log.Printf("WebSocket 升级失败: %v", err) log.Printf("WebSocket 升级失败: %v", err)
ctx.JSON(http.StatusInternalServerError, gin.H{"code": WebsocktFailed.Code, "data": WebsocktFailed.Data, "msg": WebsocktFailed.Msg}) ctx.JSON(http.StatusInternalServerError, gin.H{"code": WebsocktFailed.Code, "data": WebsocktFailed.Data, "msg": WebsocktFailed.Msg})
return return
} }