bi/middleware/jwt.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)
	}

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