blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-03 10:52:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

ws

Browse Source
This commit is contained in:
withchao 2023-12-29 11:01:55 +08:00
parent 8491023896
commit 16a3c97169
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/msggateway/n_ws_server.go
View File

@ -430,10 +430,11 @@ func (ws *WsServer) ParseWSArgs(r *http.Request) (args *WSArgs, err error) {
defer func() { defer func() {
args = &v args = &v
}() }()
query := r.URL.Query()
v.MsgResp, _ = strconv.ParseBool(query.Get(MsgResp))
if ws.onlineUserConnNum.Load() >= ws.wsMaxConnNum { if ws.onlineUserConnNum.Load() >= ws.wsMaxConnNum {
return nil, errs.ErrConnOverMaxNumLimit.Wrap("over max conn num limit") return nil, errs.ErrConnOverMaxNumLimit.Wrap("over max conn num limit")
} }
query := r.URL.Query()
if v.Token = query.Get(Token); v.Token == "" { if v.Token = query.Get(Token); v.Token == "" {
return nil, errs.ErrConnArgsErr.Wrap("token is empty") return nil, errs.ErrConnArgsErr.Wrap("token is empty")
} }
@ -458,7 +459,6 @@ func (ws *WsServer) ParseWSArgs(r *http.Request) (args *WSArgs, err error) {
if r.Header.Get(Compression) == GzipCompressionProtocol { if r.Header.Get(Compression) == GzipCompressionProtocol {
v.Compression = true v.Compression = true
} }
v.MsgResp, _ = strconv.ParseBool(query.Get(MsgResp))
m, err := ws.cache.GetTokensWithoutError(context.Background(), v.UserID, platformID) m, err := ws.cache.GetTokensWithoutError(context.Background(), v.UserID, platformID)
if err != nil { if err != nil {
return nil, err return nil, err