From c070c02f3fcb8c2c8dfa1ffcadf72a4fb827a43a Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期三, 20 十一月 2019 14:15:34 +0800
Subject: [PATCH] back

---
 middlewares/auth/auth.go |   16 ++++++++++------
 1 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/middlewares/auth/auth.go b/middlewares/auth/auth.go
index 61e40ca..d897b00 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
 }
@@ -56,15 +56,19 @@
 	return func(c *gin.Context) {
 		urlPath := c.Request.URL.Path
 
-		if strings.Contains(urlPath,"/data/api-v") && !strings.Contains(urlPath,"login"){
+		if strings.Contains(urlPath,"/data/api-") && !strings.Contains(urlPath,"login"){
 			jwtDriver :=NewJwtAuthDriver()
 			if !jwtDriver.Check(c) {
 				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
 				c.Abort()
 			}
-			user := (*jwtDriver).User(c)
-			loginM := user.(map[string]interface{})
-			userId := loginM["id"].(string)
+			userM := (*jwtDriver).User(c)
+			if userM == nil {
+				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
+				c.Abort()
+				return
+			}
+			userId := userM["id"].(string)
 			if OutUser(userId) {
 				util.ResponseFormat(c,code.TokenNotFound,"灏氭湭鐧诲綍锛岃鐧诲綍")
 				c.Abort()
@@ -77,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

--
Gitblit v1.8.0