李伟乐 1 year ago
parent 8bd3aa665e
commit a71e3d102c
  1. 8
      middleware/jwt/default.go
  2. 2
      middleware/jwt/jwt.go

@ -8,10 +8,10 @@ import (
"strings" "strings"
) )
type jwtDefault struct { type JwtDefault struct {
} }
func (j *jwtDefault) GetToken(ctx context.Context, key string) (tokenStr string) { func (j *JwtDefault) GetToken(ctx context.Context, key string) (tokenStr string) {
arr := strings.Split(key, ":") arr := strings.Split(key, ":")
if len(arr) != 2 { if len(arr) != 2 {
return "" return ""
@ -39,7 +39,7 @@ func (j *jwtDefault) GetToken(ctx context.Context, key string) (tokenStr string)
return "" return ""
} }
func (j *jwtDefault) ParseToken(ctx context.Context, key string, token string) (*UserInfo, error) { func (j *JwtDefault) ParseToken(ctx context.Context, key string, token string) (*UserInfo, error) {
userInfo, err := Parse(key, token) userInfo, err := Parse(key, token)
if err != nil { if err != nil {
return nil, err return nil, err
@ -47,7 +47,7 @@ func (j *jwtDefault) ParseToken(ctx context.Context, key string, token string) (
return userInfo, nil return userInfo, nil
} }
func (j *jwtDefault) Validate(ctx context.Context, permission string, permissions []string) error { func (j *JwtDefault) Validate(ctx context.Context, permission string, permissions []string) error {
allowPers := strings.Split(permission, "|") allowPers := strings.Split(permission, "|")
allowMap := make(map[string]bool, len(allowPers)) allowMap := make(map[string]bool, len(allowPers))
for _, v := range allowPers { for _, v := range allowPers {

@ -43,7 +43,7 @@ func JWT(opts ...JwtOption) middleware.Middleware {
var cfg = &options{ var cfg = &options{
jwtKey: "JssLx22bjQwnyqby", jwtKey: "JssLx22bjQwnyqby",
fromKey: "header:token", fromKey: "header:token",
validate: &jwtDefault{}, validate: &JwtDefault{},
} }
for _, o := range opts { for _, o := range opts {
o(cfg) o(cfg)

Loading…
Cancel
Save