47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
|
// 中间件层
|
||
|
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/1340691923/xwl_bi/platform-basic-libs/jwt"
|
||
|
|
"github.com/1340691923/xwl_bi/platform-basic-libs/my_error"
|
||
|
|
"github.com/1340691923/xwl_bi/platform-basic-libs/response"
|
||
|
|
"github.com/1340691923/xwl_bi/platform-basic-libs/service/gm_user"
|
||
|
|
"github.com/1340691923/xwl_bi/platform-basic-libs/util"
|
||
|
|
fiber "github.com/gofiber/fiber/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
var res response.Response
|
||
|
|
|
||
|
|
func JwtMiddleware(c *fiber.Ctx) error {
|
||
|
|
|
||
|
|
var err error
|
||
|
|
var claims *jwt.Claims
|
||
|
|
token := util.GetToken(c)
|
||
|
|
if _, logoff := util.TokenBucket.Load(token); logoff {
|
||
|
|
err = my_error.NewBusiness(TOKEN_ERROR, ERROR_AUTH_CHECK_TOKEN_FAIL)
|
||
|
|
return res.Error(c, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if util.GetToken(c) == "" {
|
||
|
|
err = my_error.NewBusiness(TOKEN_ERROR, INVALID_PARAMS)
|
||
|
|
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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return c.Next()
|
||
|
|
}
|