loser/lckt-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 修改鉴权相关中间件

Browse Source
This commit is contained in:
loser
2025-05-09 11:16:18 +08:00
parent 44f1b5d4b5
commit ed3c15fbb6
3 changed files with 21 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
utils/claims.go
View File

@@ -19,9 +19,9 @@ func ClearToken(c *gin.Context) {
}
if net.ParseIP(host) != nil {
c.SetCookie("app-token", "", -1, "/", "", false, false)
c.SetCookie("x-token", "", -1, "/", "", false, false)
} else {
c.SetCookie("app-token", "", -1, "/", host, false, false)
c.SetCookie("x-token", "", -1, "/", host, false, false)
}
}
@@ -33,17 +33,17 @@ func SetToken(c *gin.Context, token string, maxAge int) {
}
if net.ParseIP(host) != nil {
c.SetCookie("app-token", token, maxAge, "/", "", false, false)
c.SetCookie("x-token", token, maxAge, "/", "", false, false)
} else {
c.SetCookie("app-token", token, maxAge, "/", host, false, false)
c.SetCookie("x-token", token, maxAge, "/", host, false, false)
}
}
func GetToken(c *gin.Context) string {
token := c.Request.Header.Get("app-token")
token := c.Request.Header.Get("x-token")
if token == "" {
j := NewJWT()
token, _ = c.Cookie("app-token")
token, _ = c.Cookie("x-token")
claims, err := j.ParseToken(token)
if err != nil {
global.GVA_LOG.Error("重新写入cookie token失败,未能成功解析token,请检查请求头是否存在x-token且claims是否为规定结构")

20
utils/user_jwt/claims.go
View File

@@ -12,41 +12,41 @@ import (
)
func ClearToken(c *gin.Context) {
// 增加cookie app-token 向来源的web添加
// 增加cookie Authorization 向来源的web添加
host, _, err := net.SplitHostPort(c.Request.Host)
if err != nil {
host = c.Request.Host
}
if net.ParseIP(host) != nil {
c.SetCookie("app-token", "", -1, "/", "", false, false)
c.SetCookie("Authorization", "", -1, "/", "", false, false)
} else {
c.SetCookie("app-token", "", -1, "/", host, false, false)
c.SetCookie("Authorization", "", -1, "/", host, false, false)
}
}
func SetToken(c *gin.Context, token string, maxAge int) {
// 增加cookie app-token 向来源的web添加
// 增加cookie Authorization 向来源的web添加
host, _, err := net.SplitHostPort(c.Request.Host)
if err != nil {
host = c.Request.Host
}
if net.ParseIP(host) != nil {
c.SetCookie("app-token", token, maxAge, "/", "", false, false)
c.SetCookie("Authorization", token, maxAge, "/", "", false, false)
} else {
c.SetCookie("app-token", token, maxAge, "/", host, false, false)
c.SetCookie("Authorization", token, maxAge, "/", host, false, false)
}
}
func GetToken(c *gin.Context) string {
token := c.Request.Header.Get("app-token")
token := c.Request.Header.Get("Authorization")
if token == "" {
j := NewUserJWT()
token, _ = c.Cookie("app-token")
token, _ = c.Cookie("Authorization")
claims, err := j.ParseToken(token)
if err != nil {
global.GVA_LOG.Error("重新写入cookie token失败,未能成功解析token,请检查请求头是否存在app-token且claims是否为规定结构")
global.GVA_LOG.Error("重新写入cookie token失败,未能成功解析token,请检查请求头是否存在Authorization且claims是否为规定结构")
return token
}
SetToken(c, token, int((claims.ExpiresAt.Unix()-time.Now().Unix())/60))
@@ -59,7 +59,7 @@ func GetClaims(c *gin.Context) (*systemReq.CustomClaims, error) {
j := NewUserJWT()
claims, err := j.ParseToken(token)
if err != nil {
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析信息失败, 请检查请求头是否存在app-token且claims是否为规定结构")
global.GVA_LOG.Error("从Gin的Context中获取从jwt解析信息失败, 请检查请求头是否存在Authorization且claims是否为规定结构")
}
return claims, err
}