更新 JWT 校验
This commit is contained in:
parent
1d9d8a1ce8
commit
f84136fee9
1 changed files with 3 additions and 12 deletions
|
|
@ -6,6 +6,7 @@ import (
|
||||||
|
|
||||||
"github.com/dgrijalva/jwt-go"
|
"github.com/dgrijalva/jwt-go"
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/yaoapp/xiang/user"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Guards 服务中间件
|
// Guards 服务中间件
|
||||||
|
|
@ -13,15 +14,6 @@ var Guards = map[string]gin.HandlerFunc{
|
||||||
"bearer-jwt": bearerJWT, // JWT 鉴权
|
"bearer-jwt": bearerJWT, // JWT 鉴权
|
||||||
}
|
}
|
||||||
|
|
||||||
// JwtClaims JWT claims
|
|
||||||
type JwtClaims struct {
|
|
||||||
ID int
|
|
||||||
Type string
|
|
||||||
Mobile string
|
|
||||||
Name string
|
|
||||||
jwt.StandardClaims
|
|
||||||
}
|
|
||||||
|
|
||||||
// JWT 鉴权
|
// JWT 鉴权
|
||||||
func bearerJWT(c *gin.Context) {
|
func bearerJWT(c *gin.Context) {
|
||||||
tokenString := c.Request.Header.Get("Authorization")
|
tokenString := c.Request.Header.Get("Authorization")
|
||||||
|
|
@ -32,7 +24,7 @@ func bearerJWT(c *gin.Context) {
|
||||||
}
|
}
|
||||||
|
|
||||||
tokenString = strings.TrimSpace(strings.TrimPrefix(tokenString, "Bearer "))
|
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
|
return Conf.JWT.Secret, nil
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
@ -42,11 +34,10 @@ func bearerJWT(c *gin.Context) {
|
||||||
return
|
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("id", claims.Subject)
|
||||||
c.Set("type", claims.Type)
|
c.Set("type", claims.Type)
|
||||||
c.Set("name", claims.Name)
|
c.Set("name", claims.Name)
|
||||||
c.Set("mobile", claims.Mobile)
|
|
||||||
c.Next()
|
c.Next()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue