This commit is contained in:
@@ -39,6 +39,49 @@ func SetToken(c *gin.Context, token string, maxAge int) {
|
||||
}
|
||||
}
|
||||
|
||||
func ClearUserToken(c *gin.Context) {
|
||||
// 增加cookie user-token 向来源的web添加
|
||||
host, _, err := net.SplitHostPort(c.Request.Host)
|
||||
if err != nil {
|
||||
host = c.Request.Host
|
||||
}
|
||||
|
||||
if net.ParseIP(host) != nil {
|
||||
c.SetCookie("user-token", "", -1, "/", "", false, false)
|
||||
} else {
|
||||
c.SetCookie("user-token", "", -1, "/", host, false, false)
|
||||
}
|
||||
}
|
||||
|
||||
func SetUserToken(c *gin.Context, token string, maxAge int) {
|
||||
// 增加cookie user-token 向来源的web添加
|
||||
host, _, err := net.SplitHostPort(c.Request.Host)
|
||||
if err != nil {
|
||||
host = c.Request.Host
|
||||
}
|
||||
|
||||
if net.ParseIP(host) != nil {
|
||||
c.SetCookie("user-token", token, maxAge, "/", "", false, false)
|
||||
} else {
|
||||
c.SetCookie("user-token", token, maxAge, "/", host, false, false)
|
||||
}
|
||||
}
|
||||
|
||||
func GetUserToken(c *gin.Context) string {
|
||||
token := c.Request.Header.Get("user-token")
|
||||
if token == "" {
|
||||
j := NewJWT()
|
||||
token, _ = c.Cookie("user-token")
|
||||
claims, err := j.ParseToken(token)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("重新写入cookie token失败,未能成功解析token,请检查请求头是否存在user-token且claims是否为规定结构")
|
||||
return token
|
||||
}
|
||||
SetUserToken(c, token, int((claims.ExpiresAt.Unix()-time.Now().Unix())/60))
|
||||
}
|
||||
return token
|
||||
}
|
||||
|
||||
func GetToken(c *gin.Context) string {
|
||||
token := c.Request.Header.Get("x-token")
|
||||
if token == "" {
|
||||
@@ -146,3 +189,134 @@ func LoginToken(user system.Login) (token string, claims systemReq.CustomClaims,
|
||||
token, err = j.CreateToken(claims)
|
||||
return
|
||||
}
|
||||
|
||||
// AppUserLogin 小程序用户登录接口
|
||||
type AppUserLogin interface {
|
||||
GetUUID() uuid.UUID
|
||||
GetUserId() uint
|
||||
GetOpenID() string
|
||||
GetUnionID() string
|
||||
GetNickname() string
|
||||
GetAvatar() string
|
||||
GetPhone() string
|
||||
GetGender() int
|
||||
GetCity() string
|
||||
GetProvince() string
|
||||
GetCountry() string
|
||||
}
|
||||
|
||||
// AppUserLoginToken 创建小程序用户登录token
|
||||
func AppUserLoginToken(user AppUserLogin) (token string, claims systemReq.AppUserClaims, err error) {
|
||||
j := NewJWT()
|
||||
claims = j.CreateAppUserClaims(systemReq.AppBaseClaims{
|
||||
UUID: user.GetUUID(),
|
||||
ID: user.GetUserId(),
|
||||
OpenID: user.GetOpenID(),
|
||||
UnionID: user.GetUnionID(),
|
||||
NickName: user.GetNickname(),
|
||||
Avatar: user.GetAvatar(),
|
||||
Phone: user.GetPhone(),
|
||||
Gender: user.GetGender(),
|
||||
City: user.GetCity(),
|
||||
Province: user.GetProvince(),
|
||||
Country: user.GetCountry(),
|
||||
})
|
||||
token, err = j.CreateAppUserToken(claims)
|
||||
return
|
||||
}
|
||||
|
||||
// 用户端专用Claims获取函数
|
||||
|
||||
func GetAppUserClaims(c *gin.Context) (*systemReq.AppUserClaims, error) {
|
||||
token := GetUserToken(c)
|
||||
j := NewJWT()
|
||||
claims, err := j.ParseAppUserToken(token)
|
||||
if err != nil {
|
||||
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析信息失败, 请检查请求头是否存在user-token且claims是否为规定结构")
|
||||
}
|
||||
return claims, err
|
||||
}
|
||||
|
||||
// GetAppUserID 从Gin的Context中获取从jwt解析出来的用户ID
|
||||
func GetAppUserID(c *gin.Context) uint {
|
||||
if claims, exists := c.Get("user_claims"); !exists {
|
||||
if cl, err := GetAppUserClaims(c); err != nil {
|
||||
return 0
|
||||
} else {
|
||||
return cl.AppBaseClaims.ID
|
||||
}
|
||||
} else {
|
||||
waitUse := claims.(*systemReq.AppUserClaims)
|
||||
return waitUse.AppBaseClaims.ID
|
||||
}
|
||||
}
|
||||
|
||||
// GetAppUserUuid 从Gin的Context中获取从jwt解析出来的用户UUID
|
||||
func GetAppUserUuid(c *gin.Context) uuid.UUID {
|
||||
if claims, exists := c.Get("user_claims"); !exists {
|
||||
if cl, err := GetAppUserClaims(c); err != nil {
|
||||
return uuid.UUID{}
|
||||
} else {
|
||||
return cl.AppBaseClaims.UUID
|
||||
}
|
||||
} else {
|
||||
waitUse := claims.(*systemReq.AppUserClaims)
|
||||
return waitUse.AppBaseClaims.UUID
|
||||
}
|
||||
}
|
||||
|
||||
// GetAppUserInfo 从Gin的Context中获取从jwt解析出来的用户信息
|
||||
func GetAppUserInfo(c *gin.Context) *systemReq.AppUserClaims {
|
||||
if claims, exists := c.Get("user_claims"); !exists {
|
||||
if cl, err := GetAppUserClaims(c); err != nil {
|
||||
return nil
|
||||
} else {
|
||||
return cl
|
||||
}
|
||||
} else {
|
||||
waitUse := claims.(*systemReq.AppUserClaims)
|
||||
return waitUse
|
||||
}
|
||||
}
|
||||
|
||||
// GetAppUserOpenID 从Gin的Context中获取从jwt解析出来的微信OpenID
|
||||
func GetAppUserOpenID(c *gin.Context) string {
|
||||
if claims, exists := c.Get("user_claims"); !exists {
|
||||
if cl, err := GetAppUserClaims(c); err != nil {
|
||||
return ""
|
||||
} else {
|
||||
return cl.AppBaseClaims.OpenID
|
||||
}
|
||||
} else {
|
||||
waitUse := claims.(*systemReq.AppUserClaims)
|
||||
return waitUse.AppBaseClaims.OpenID
|
||||
}
|
||||
}
|
||||
|
||||
// GetAppUserNickName 从Gin的Context中获取从jwt解析出来的用户昵称
|
||||
func GetAppUserNickName(c *gin.Context) string {
|
||||
if claims, exists := c.Get("user_claims"); !exists {
|
||||
if cl, err := GetAppUserClaims(c); err != nil {
|
||||
return ""
|
||||
} else {
|
||||
return cl.AppBaseClaims.NickName
|
||||
}
|
||||
} else {
|
||||
waitUse := claims.(*systemReq.AppUserClaims)
|
||||
return waitUse.AppBaseClaims.NickName
|
||||
}
|
||||
}
|
||||
|
||||
// GetAppUserAvatar 从Gin的Context中获取从jwt解析出来的用户头像
|
||||
func GetAppUserAvatar(c *gin.Context) string {
|
||||
if claims, exists := c.Get("user_claims"); !exists {
|
||||
if cl, err := GetAppUserClaims(c); err != nil {
|
||||
return ""
|
||||
} else {
|
||||
return cl.AppBaseClaims.Avatar
|
||||
}
|
||||
} else {
|
||||
waitUse := claims.(*systemReq.AppUserClaims)
|
||||
return waitUse.AppBaseClaims.Avatar
|
||||
}
|
||||
}
|
||||
|
||||
@@ -103,3 +103,64 @@ func SetRedisJWT(jwt string, userName string) (err error) {
|
||||
err = global.GVA_REDIS.Set(context.Background(), userName, jwt, timer).Err()
|
||||
return err
|
||||
}
|
||||
|
||||
// 用户端专用JWT函数
|
||||
|
||||
// CreateAppUserClaims 创建小程序用户Claims
|
||||
func (j *JWT) CreateAppUserClaims(baseClaims request.AppBaseClaims) request.AppUserClaims {
|
||||
bf, _ := ParseDuration(global.GVA_CONFIG.JWT.BufferTime)
|
||||
ep, _ := ParseDuration(global.GVA_CONFIG.JWT.ExpiresTime)
|
||||
claims := request.AppUserClaims{
|
||||
AppBaseClaims: baseClaims,
|
||||
BufferTime: int64(bf / time.Second), // 缓冲时间1天 缓冲时间内会获得新的token刷新令牌
|
||||
RegisteredClaims: jwt.RegisteredClaims{
|
||||
Audience: jwt.ClaimStrings{"GVA-APP"}, // 受众
|
||||
NotBefore: jwt.NewNumericDate(time.Now().Add(-1000)), // 签名生效时间
|
||||
ExpiresAt: jwt.NewNumericDate(time.Now().Add(ep)), // 过期时间 7天 配置文件
|
||||
Issuer: global.GVA_CONFIG.JWT.Issuer, // 签名的发行者
|
||||
},
|
||||
}
|
||||
return claims
|
||||
}
|
||||
|
||||
// CreateAppUserToken 创建小程序用户token
|
||||
func (j *JWT) CreateAppUserToken(claims request.AppUserClaims) (string, error) {
|
||||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||
return token.SignedString(j.SigningKey)
|
||||
}
|
||||
|
||||
// CreateAppUserTokenByOldToken 旧token换新token（小程序用户专用）
|
||||
func (j *JWT) CreateAppUserTokenByOldToken(oldToken string, claims request.AppUserClaims) (string, error) {
|
||||
v, err, _ := global.GVA_Concurrency_Control.Do("APP-JWT:"+oldToken, func() (interface{}, error) {
|
||||
return j.CreateAppUserToken(claims)
|
||||
})
|
||||
return v.(string), err
|
||||
}
|
||||
|
||||
// ParseAppUserToken 解析小程序用户token
|
||||
func (j *JWT) ParseAppUserToken(tokenString string) (*request.AppUserClaims, error) {
|
||||
token, err := jwt.ParseWithClaims(tokenString, &request.AppUserClaims{}, func(token *jwt.Token) (i interface{}, e error) {
|
||||
return j.SigningKey, nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
switch {
|
||||
case errors.Is(err, jwt.ErrTokenExpired):
|
||||
return nil, TokenExpired
|
||||
case errors.Is(err, jwt.ErrTokenMalformed):
|
||||
return nil, TokenMalformed
|
||||
case errors.Is(err, jwt.ErrTokenSignatureInvalid):
|
||||
return nil, TokenSignatureInvalid
|
||||
case errors.Is(err, jwt.ErrTokenNotValidYet):
|
||||
return nil, TokenNotValidYet
|
||||
default:
|
||||
return nil, TokenInvalid
|
||||
}
|
||||
}
|
||||
if token != nil {
|
||||
if claims, ok := token.Claims.(*request.AppUserClaims); ok && token.Valid {
|
||||
return claims, nil
|
||||
}
|
||||
}
|
||||
return nil, TokenValid
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user