代码优化

This commit is contained in:
1340691923@qq.com 2022-03-15 15:43:37 +08:00
parent 0cce4ee276
commit 600a2d700e
4 changed files with 22 additions and 21 deletions

View File

@ -51,9 +51,7 @@ func WechatSpider(handle fasthttp.RequestHandler) fasthttp.RequestHandler {
"msg": "该UserAgent禁止访问接口", "msg": "该UserAgent禁止访问接口",
}) })
return return
} else { }
handle(ctx) handle(ctx)
} }
}
} }

View File

@ -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 var service gm_user.GmUserService
claims, err = jwt.ParseToken(token) claims, err = jwt.ParseToken(token)
if err != nil { if err != nil {
err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_FAIL) err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_FAIL)
return res.Error(c, err) return res.Error(c, err)
} else if time.Now().Unix() > claims.ExpiresAt { }
err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT) if time.Now().Unix() > claims.ExpiresAt {
return res.Error(c, err)
} else if !service.IsExitUser(claims) {
err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT) err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_TIMEOUT)
return res.Error(c, err) 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()
} }

View File

@ -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
} }

View File

@ -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))
} }