更新 JWT 校验

This commit is contained in:
Max 2021-09-30 09:09:36 +08:00
parent 1d9d8a1ce8
commit f84136fee9

View file

@ -6,6 +6,7 @@ import (
"github.com/dgrijalva/jwt-go"
"github.com/gin-gonic/gin"
"github.com/yaoapp/xiang/user"
)
// Guards 服务中间件
@ -13,15 +14,6 @@ var Guards = map[string]gin.HandlerFunc{
"bearer-jwt": bearerJWT, // JWT 鉴权
}
// JwtClaims JWT claims
type JwtClaims struct {
ID int
Type string
Mobile string
Name string
jwt.StandardClaims
}
// JWT 鉴权
func bearerJWT(c *gin.Context) {
tokenString := c.Request.Header.Get("Authorization")
@ -32,7 +24,7 @@ func bearerJWT(c *gin.Context) {
}
tokenString = strings.TrimSpace(strings.TrimPrefix(tokenString, "Bearer "))
token, err := jwt.ParseWithClaims(tokenString, &JwtClaims{}, func(token *jwt.Token) (interface{}, error) {
token, err := jwt.ParseWithClaims(tokenString, &user.JwtClaims{}, func(token *jwt.Token) (interface{}, error) {
return Conf.JWT.Secret, nil
})
@ -42,11 +34,10 @@ func bearerJWT(c *gin.Context) {
return
}
if claims, ok := token.Claims.(*JwtClaims); ok && token.Valid {
if claims, ok := token.Claims.(*user.JwtClaims); ok && token.Valid {
c.Set("id", claims.Subject)
c.Set("type", claims.Type)
c.Set("name", claims.Name)
c.Set("mobile", claims.Mobile)
c.Next()
return
}