From 5e31a32f6ce2572cb01388150ddb325724ff15b8 Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期五, 28 六月 2019 19:29:20 +0800 Subject: [PATCH] Merge branch 'master' of http://192.168.1.14:10010/r/webserver --- middlewares/auth/jwt.go | 40 ++++++++++++++++++++-------------------- 1 files changed, 20 insertions(+), 20 deletions(-) diff --git a/middlewares/auth/jwt.go b/middlewares/auth/jwt.go index 01fdab9..0187b2f 100644 --- a/middlewares/auth/jwt.go +++ b/middlewares/auth/jwt.go @@ -13,14 +13,14 @@ type jwtAuthManager struct { secret string expire time.Duration - alg string + alg string } -func NewJwtAuthDriver() *jwtAuthManager{ +func NewJwtAuthDriver() *jwtAuthManager { return &jwtAuthManager{ - secret:TokenKey, - expire:time.Hour*8, - alg:"HS256", + secret: TokenKey, + expire: time.Hour * 8, + alg: "HS256", } } @@ -34,41 +34,41 @@ if len(t) < 2 { return false } - var keyFunc = func(token *jwtLib.Token) (interface{},error) { + var keyFunc = func(token *jwtLib.Token) (interface{}, error) { b := []byte(jwtAuth.secret) - return b,nil + return b, nil } - authJwtToken, err:= request.ParseFromRequest(c.Request,request.OAuth2Extractor,keyFunc) - if err !=nil { + authJwtToken, err := request.ParseFromRequest(c.Request, request.OAuth2Extractor, keyFunc) + if err != nil { return false } - c.Set("User",map[string]interface{}{ - "token":authJwtToken, + c.Set("User", map[string]interface{}{ + "token": authJwtToken, }) return authJwtToken.Valid } -func (jwtAuth *jwtAuthManager) User(c *gin.Context) interface{}{ +func (jwtAuth *jwtAuthManager) User(c *gin.Context) interface{} { var jwtToken *jwtLib.Token - if jwtUser, exist := c.Get("User");!exist{ - tokenStr :=strings.Replace(c.Request.Header.Get("Authorization"),"Bearer ","",-1) - if tokenStr == ""{ + if jwtUser, exist := c.Get("User"); !exist { + tokenStr := strings.Replace(c.Request.Header.Get("Authorization"), "Bearer ", "", -1) + if tokenStr == "" { return map[interface{}]interface{}{} } var err error - jwtToken,err = jwtLib.Parse(tokenStr, func(token *jwtLib.Token) (interface{}, error) { - b :=[]byte(jwtAuth.secret) - return b,nil + jwtToken, err = jwtLib.Parse(tokenStr, func(token *jwtLib.Token) (interface{}, error) { + b := []byte(jwtAuth.secret) + return b, nil }) - if err !=nil { + if err != nil { return nil } } else { jwtToken = jwtUser.(map[string]interface{})["token"].(*jwtLib.Token) } - if claims,ok :=jwtToken.Claims.(jwtLib.MapClaims);ok && jwtToken.Valid{ + if claims, ok := jwtToken.Claims.(jwtLib.MapClaims); ok && jwtToken.Valid { var user map[string]interface{} if err := json.Unmarshal([]byte(claims["user"].(string)), &user); err != nil { return nil -- Gitblit v1.8.0