Skip to content

Commit b09574b

Browse files
committed
Update authorizationmiddleware.go
1 parent 6962441 commit b09574b

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

go_ws_sh/authorizationmiddleware.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"context"
55
"encoding/base64"
66
"encoding/json"
7+
"strings"
78

89
"github.com/cloudwego/hertz/pkg/app"
910
"github.com/cloudwego/hertz/pkg/protocol/consts"
@@ -12,14 +13,14 @@ import (
1213
"github.com/masx200/go_ws_sh/types"
1314
)
1415

15-
1616
// AuthorizationMiddleware 定义身份验证中间件
1717
func AuthorizationMiddleware(credentialdb *gorm.DB, tokendb *gorm.DB, sessiondb *gorm.DB) types.HertzMiddleWare {
1818
return func(c context.Context, r *app.RequestContext, next types.HertzNext) {
1919

2020
bearertoken := r.Request.Header.Get("authorization")
2121

2222
if bearertoken != "" {
23+
bearertoken=strings.TrimPrefix(bearertoken, "Bearer ")
2324
decoded, err := base64.StdEncoding.DecodeString(bearertoken)
2425
if err != nil {
2526
r.AbortWithStatusJSON(consts.StatusUnauthorized, map[string]string{

0 commit comments

Comments
 (0)