This commit is contained in:
wenxu12345 2022-06-21 17:15:24 +08:00
parent 08d9551230
commit 9b76eb6b55

View File

@ -12,7 +12,6 @@ import (
go_redis "github.com/go-redis/redis/v8"
"gopkg.in/errgo.v2/errors"
"net/http"
"reflect"
"sync"
"time"
@ -291,7 +290,7 @@ func (ws *WServer) headerCheck(w http.ResponseWriter, r *http.Request, operation
query := r.URL.Query()
if len(query["token"]) != 0 && len(query["sendID"]) != 0 && len(query["platformID"]) != 0 {
if ok, err, msg := token_verify.WsVerifyToken(query["token"][0], query["sendID"][0], query["platformID"][0], operationID); !ok {
switch err {
switch errors.Cause(err) {
case constant.ErrTokenExpired:
status = int(constant.ErrTokenExpired.ErrCode)
case constant.ErrTokenInvalid:
@ -309,7 +308,7 @@ func (ws *WServer) headerCheck(w http.ResponseWriter, r *http.Request, operation
case constant.ErrTokenDifferentUserID:
status = int(constant.ErrTokenDifferentUserID.ErrCode)
}
log.Error(operationID, "Token verify failed ", "query ", query, msg, err.Error(), "type: ", reflect.TypeOf(errors.Cause(err)))
log.Error(operationID, "Token verify failed ", "query ", query, msg, err.Error())
w.Header().Set("Sec-Websocket-Version", "13")
w.Header().Set("ws_err_msg", err.Error())
http.Error(w, err.Error(), status)