estel/gva
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feature:调整jwt续期功能，保证新的token返回和一致性

Browse Source
This commit is contained in:
piexlmax（奇淼
2024-03-27 10:39:19 +08:00
parent 619c5b5e1d
commit 2d884c83cc
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
server/middleware/jwt.go
View File

@@ -58,7 +58,6 @@ func JWTAuth() gin.HandlerFunc {
// c.Abort() // c.Abort()
//} //}
c.Set("claims", claims) c.Set("claims", claims)
c.Next()
if claims.ExpiresAt.Unix()-time.Now().Unix() < claims.BufferTime { if claims.ExpiresAt.Unix()-time.Now().Unix() < claims.BufferTime {
dr, _ := utils.ParseDuration(global.GVA_CONFIG.JWT.ExpiresTime) dr, _ := utils.ParseDuration(global.GVA_CONFIG.JWT.ExpiresTime)
claims.ExpiresAt = jwt.NewNumericDate(time.Now().Add(dr)) claims.ExpiresAt = jwt.NewNumericDate(time.Now().Add(dr))
@@ -78,5 +77,13 @@ func JWTAuth() gin.HandlerFunc {
_ = jwtService.SetRedisJWT(newToken, newClaims.Username) _ = jwtService.SetRedisJWT(newToken, newClaims.Username)
} }
} }
c.Next()
if newToken, exists := c.Get("new-token"); exists {
c.Header("new-token", newToken.(string))
}
if newExpiresAt, exists := c.Get("new-expires-at"); exists {
c.Header("new-expires-at", newExpiresAt.(string))
}
} }
} }