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

chore: Update token.go to allow admin to add or subtract used quota

Browse Source
This commit is contained in:
Laisky.Cai 2024-06-24 05:52:23 +00:00
parent 0164cdbcf2
commit 0e35db82d5
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
controller/token.go
View File

@ -288,7 +288,19 @@ func UpdateToken(c *gin.Context) {
cleanToken.Subnet = token.Subnet cleanToken.Subnet = token.Subnet
} }
// let admin to add or subtract used quota,
// make it possible to aggregate billings from different sources.
if tokenPatch.AddUsedQuota != 0 { if tokenPatch.AddUsedQuota != 0 {
if cleanToken.RemainQuota < int64(tokenPatch.AddUsedQuota) {
c.JSON(http.StatusOK, gin.H{
"success": false,
"message": "剩余额度不足",
})
return
}
cleanToken.UsedQuota += int64(tokenPatch.AddUsedQuota)
cleanToken.RemainQuota -= int64(tokenPatch.AddUsedQuota)
model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("外部(%s)消耗 %s", tokenPatch.AddReason, common.LogQuota(int64(tokenPatch.AddUsedQuota)))) model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("外部(%s)消耗 %s", tokenPatch.AddReason, common.LogQuota(int64(tokenPatch.AddUsedQuota))))
} }