From f6547ecc30f8a5c51cb32ada62819d5fe8115769 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期三, 06 十一月 2019 14:46:44 +0800
Subject: [PATCH] fix auth

---
 controllers/user.go      |    7 ++-----
 middlewares/auth/auth.go |   11 +++++------
 middlewares/auth/jwt.go  |    2 +-
 3 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/controllers/user.go b/controllers/user.go
index a78d271..6f5fc2f 100644
--- a/controllers/user.go
+++ b/controllers/user.go
@@ -2,7 +2,6 @@
 
 import (
 	"basic.com/dbapi.git"
-	"basic.com/valib/logger.git"
 	"github.com/gin-gonic/gin"
 	"net/http"
 	"time"
@@ -104,10 +103,8 @@
 // @Router /data/api-u/users/findAllUser [get]
 func (uc UserController) FindAllUser(c *gin.Context) {
 	authDriver := auth.GenerateAuthDriver()
-	user := (*authDriver).User(c)
-	logger.Debug("cur userInfo:",user)
-	loginM := user.(map[string]interface{})
-	userId := loginM["id"].(string)
+	userM := (*authDriver).User(c)
+	userId := userM["id"].(string)
 	var api dbapi.UserApi
 	b,d := api.FindAllUser(userId)
 	if b {
diff --git a/middlewares/auth/auth.go b/middlewares/auth/auth.go
index 7df2c82..6d80437 100644
--- a/middlewares/auth/auth.go
+++ b/middlewares/auth/auth.go
@@ -15,7 +15,7 @@
 
 type Auth interface {
 	Check(c *gin.Context)bool
-	User(c *gin.Context)interface{}
+	User(c *gin.Context)map[string]interface{}
 	Login(http *http.Request,w http.ResponseWriter,user map[string]interface{})interface{}
 	Logout(http *http.Request,w http.ResponseWriter) bool
 }
@@ -62,14 +62,13 @@
 				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
 				c.Abort()
 			}
-			user := (*jwtDriver).User(c)
-			if user == nil {
+			userM := (*jwtDriver).User(c)
+			if userM == nil {
 				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
 				c.Abort()
 				return
 			}
-			loginM := user.(map[string]interface{})
-			userId := loginM["id"].(string)
+			userId := userM["id"].(string)
 			if OutUser(userId) {
 				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
 				c.Abort()
@@ -82,5 +81,5 @@
 }
 
 func GetCurUser(c *gin.Context)map[string]interface{}{
-	return (*GenerateAuthDriver()).User(c).(map[string]interface{})
+	return (*GenerateAuthDriver()).User(c)
 }
\ No newline at end of file
diff --git a/middlewares/auth/jwt.go b/middlewares/auth/jwt.go
index aadecf4..37b4fc3 100644
--- a/middlewares/auth/jwt.go
+++ b/middlewares/auth/jwt.go
@@ -50,7 +50,7 @@
 	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)

--
Gitblit v1.8.0