55 lines
1004 B
Go
55 lines
1004 B
Go
package middleware
|
|
|
|
import (
|
|
"github.com/dgrijalva/jwt-go"
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"myschools.me/wyh/token.git/config"
|
|
"myschools.me/wyh/token.git/model"
|
|
)
|
|
|
|
const (
|
|
TokenName = "token"
|
|
)
|
|
|
|
//AuthUser 用户认证拦截器
|
|
func AuthUser() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
tokenString := c.GetHeader(TokenName)
|
|
|
|
claims := &model.UserClaims{}
|
|
tkn, err := jwt.ParseWithClaims(tokenString, claims, func(t *jwt.Token) (interface{}, error) {
|
|
return []byte(config.JwtKey), nil
|
|
})
|
|
if err != nil {
|
|
if err == jwt.ErrSignatureInvalid {
|
|
c.AbortWithStatusJSON(200, gin.H{
|
|
"code": "1",
|
|
"msg": "用户认证失败",
|
|
"data": "",
|
|
})
|
|
return
|
|
}
|
|
c.AbortWithStatusJSON(200, gin.H{
|
|
"code": "1",
|
|
"msg": "用户认证请求失败",
|
|
"data": "",
|
|
})
|
|
return
|
|
}
|
|
|
|
if !tkn.Valid {
|
|
c.AbortWithStatusJSON(200, gin.H{
|
|
"code": "1",
|
|
"msg": "验证失败",
|
|
"data": "",
|
|
})
|
|
return
|
|
}
|
|
|
|
c.Set("user", claims)
|
|
|
|
c.Next()
|
|
}
|
|
}
|