service/library/middleware/jwt_authenticator.go

package middleware

import (
	"fmt"
	"service/api/base"
	"service/library/logger"

	"github.com/gin-gonic/gin"
)

var (
	DefaultJwtAuthenticator gin.HandlerFunc
)

func InitJwtAuthenticator(verifyFunc func(*gin.Context, string) (int64, error)) {
	DefaultJwtAuthenticator = func(ctx *gin.Context) {

		//1.获取token
		baseRequest, ok := ctx.MustGet("client_req").(base.BaseRequestAccessible)
		if !ok {
			base.CheckBadRequest(ctx, fmt.Errorf("BaseRequest acquire failed"))
			return
		}
		tokenString := baseRequest.GetBaseRequest().Token
		logger.Info("token : %v", tokenString)

		//2.校验
		id, err := verifyFunc(ctx, tokenString)
		if !base.CheckBadRequest(ctx, err) {
			return
		}

		ctx.Set("token_uuid", id)
		ctx.Next()
	}
}

func JwtAuthenticator() gin.HandlerFunc {
	return DefaultJwtAuthenticator
}