代码优化
This commit is contained in:
		
							parent
							
								
									0cce4ee276
								
							
						
					
					
						commit
						600a2d700e
					
				@ -51,9 +51,7 @@ func WechatSpider(handle fasthttp.RequestHandler) fasthttp.RequestHandler {
 | 
				
			|||||||
				"msg":  "该UserAgent禁止访问接口!",
 | 
									"msg":  "该UserAgent禁止访问接口!",
 | 
				
			||||||
			})
 | 
								})
 | 
				
			||||||
			return
 | 
								return
 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			handle(ctx)
 | 
					 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							handle(ctx)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
@ -27,20 +27,22 @@ func JwtMiddleware(c *fiber.Ctx) error {
 | 
				
			|||||||
	if util.GetToken(c) == "" {
 | 
						if util.GetToken(c) == "" {
 | 
				
			||||||
		err = my_error.NewBusiness(TOKEN_ERROR, INVALID_PARAMS)
 | 
							err = my_error.NewBusiness(TOKEN_ERROR, INVALID_PARAMS)
 | 
				
			||||||
		return res.Error(c, err)
 | 
							return res.Error(c, err)
 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		var service gm_user.GmUserService
 | 
					 | 
				
			||||||
		claims, err = jwt.ParseToken(token)
 | 
					 | 
				
			||||||
		if err != nil {
 | 
					 | 
				
			||||||
			err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_FAIL)
 | 
					 | 
				
			||||||
			return res.Error(c, err)
 | 
					 | 
				
			||||||
		} else if time.Now().Unix() > claims.ExpiresAt {
 | 
					 | 
				
			||||||
			err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT)
 | 
					 | 
				
			||||||
			return res.Error(c, err)
 | 
					 | 
				
			||||||
		} else if !service.IsExitUser(claims) {
 | 
					 | 
				
			||||||
			err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT)
 | 
					 | 
				
			||||||
			return res.Error(c, err)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var service gm_user.GmUserService
 | 
				
			||||||
 | 
						claims, err = jwt.ParseToken(token)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_FAIL)
 | 
				
			||||||
 | 
							return res.Error(c, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if time.Now().Unix() > claims.ExpiresAt {
 | 
				
			||||||
 | 
							err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT)
 | 
				
			||||||
 | 
							return res.Error(c, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if !service.IsExitUser(claims) {
 | 
				
			||||||
 | 
							err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT)
 | 
				
			||||||
 | 
							return res.Error(c, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return c.Next()
 | 
						return c.Next()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
@ -22,17 +22,19 @@ type CheckConfigStruct struct {
 | 
				
			|||||||
func (this Request) CheckParameter(checkConfig []CheckConfigStruct, ctx *fiber.Ctx) (err error) {
 | 
					func (this Request) CheckParameter(checkConfig []CheckConfigStruct, ctx *fiber.Ctx) (err error) {
 | 
				
			||||||
	method := strings.ToUpper(ctx.Method())
 | 
						method := strings.ToUpper(ctx.Method())
 | 
				
			||||||
	for _, config := range checkConfig {
 | 
						for _, config := range checkConfig {
 | 
				
			||||||
		if method == "GET" {
 | 
							switch method {
 | 
				
			||||||
 | 
							case "GET":
 | 
				
			||||||
			if ctx.FormValue(config.Key) == "" {
 | 
								if ctx.FormValue(config.Key) == "" {
 | 
				
			||||||
				err = my_error.NewBusiness(ErrorMap, config.Code)
 | 
									err = my_error.NewBusiness(ErrorMap, config.Code)
 | 
				
			||||||
				return
 | 
									return
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		} else if method == "POST" {
 | 
							case "POST":
 | 
				
			||||||
			if !gjson.GetBytes(ctx.Body(), config.Key).Exists() {
 | 
								if !gjson.GetBytes(ctx.Body(), config.Key).Exists() {
 | 
				
			||||||
				err = my_error.NewBusiness(ErrorMap, config.Code)
 | 
									err = my_error.NewBusiness(ErrorMap, config.Code)
 | 
				
			||||||
				return
 | 
									return
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return
 | 
						return
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
@ -107,9 +107,8 @@ func GetDims(database, table string, excludedColumns []string, conn *sqlx.DB,onl
 | 
				
			|||||||
			if jsonErr == nil {
 | 
								if jsonErr == nil {
 | 
				
			||||||
				dimsCacheMap.Store(dimsCachekey,dims)
 | 
									dimsCacheMap.Store(dimsCachekey,dims)
 | 
				
			||||||
				return dims,err
 | 
									return dims,err
 | 
				
			||||||
			} else {
 | 
					 | 
				
			||||||
				logs.Logger.Error("jsonErr", zap.Error(jsonErr))
 | 
					 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
								logs.Logger.Error("jsonErr", zap.Error(jsonErr))
 | 
				
			||||||
		}else{
 | 
							}else{
 | 
				
			||||||
			logs.Logger.Error("GzipUnCompressByte Err", zap.Error(err))
 | 
								logs.Logger.Error("GzipUnCompressByte Err", zap.Error(err))
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user