xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-20 02:26:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: enhance token status handling in UpdateToken function with additional cases

Browse Source
This commit is contained in:
Laisky.Cai 2024-11-20 03:02:21 +00:00
parent 0d9c79be35
commit f322f8029d
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
controller/token.go
View File

@ -350,7 +350,8 @@ func UpdateToken(c *gin.Context) {
return return
} }
if token.Status == model.TokenStatusEnabled { switch token.Status {
case model.TokenStatusEnabled:
if cleanToken.Status == model.TokenStatusExpired && if cleanToken.Status == model.TokenStatusExpired &&
cleanToken.ExpiredTime <= helper.GetTimestamp() && cleanToken.ExpiredTime != -1 && cleanToken.ExpiredTime <= helper.GetTimestamp() && cleanToken.ExpiredTime != -1 &&
token.ExpiredTime != -1 && token.ExpiredTime < helper.GetTimestamp() { token.ExpiredTime != -1 && token.ExpiredTime < helper.GetTimestamp() {
@ -369,6 +370,14 @@ func UpdateToken(c *gin.Context) {
}) })
return return
} }
case model.TokenStatusExhausted:
if token.RemainQuota > 0 || token.UnlimitedQuota {
token.Status = model.TokenStatusEnabled
}
case model.TokenStatusExpired:
if token.ExpiredTime == -1 || token.ExpiredTime > helper.GetTimestamp() {
token.Status = model.TokenStatusEnabled
}
} }
if statusOnly != "" { if statusOnly != "" {