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

chore: update token.go to record consumption log

Browse Source
This commit is contained in:
Laisky.Cai 2024-06-24 06:14:58 +00:00
parent 1c2e8f046b
commit a8cc96bf4a
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
controller/token.go
View File

@ -294,7 +294,11 @@ func UpdateToken(c *gin.Context) {
cleanToken.UsedQuota += int64(tokenPatch.AddUsedQuota) cleanToken.UsedQuota += int64(tokenPatch.AddUsedQuota)
cleanToken.RemainQuota -= 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.RecordConsumeLog(c.Request.Context(),
userId, 0, 0, 0, tokenPatch.AddReason, cleanToken.Name,
int64(tokenPatch.AddUsedQuota),
fmt.Sprintf("外部(%s)消耗 %s",
tokenPatch.AddReason, common.LogQuota(int64(tokenPatch.AddUsedQuota))))
default: default:
// If you add more fields, please also update token.Update() // If you add more fields, please also update token.Update()
cleanToken.Name = token.Name cleanToken.Name = token.Name