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