From 263c7b18ce7f18f1222318f6e0e499e907895a52 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期三, 13 十一月 2019 19:33:39 +0800
Subject: [PATCH] fix

---
 middlewares/auth/jwt.go |   42 +++++++++++++++++++++---------------------
 1 files changed, 21 insertions(+), 21 deletions(-)

diff --git a/middlewares/auth/jwt.go b/middlewares/auth/jwt.go
index 01fdab9..37b4fc3 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) map[string]interface{} {
 	var jwtToken *jwtLib.Token
-	if jwtUser, exist := c.Get("User");!exist{
-		tokenStr :=strings.Replace(c.Request.Header.Get("Authorization"),"Bearer ","",-1)
-		if tokenStr == ""{
-			return map[interface{}]interface{}{}
+	if jwtUser, exist := c.Get("User"); !exist {
+		tokenStr := strings.Replace(c.Request.Header.Get("Authorization"), "Bearer ", "", -1)
+		if tokenStr == "" {
+			return nil
 		}
 		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