39 lines
773 B
Go
39 lines
773 B
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) 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.校验
|
||
|
|
err := verifyFunc(ctx, tokenString)
|
||
|
|
if !base.CheckBadRequest(ctx, err) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx.Next()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func JwtAuthenticator() gin.HandlerFunc {
|
||
|
|
return DefaultJwtAuthenticator
|
||
|
|
}
|