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

chore: decrease token quota in ConsumeToken function

Browse Source 
This commit modifies the ConsumeToken function in the token.go file to decrease the token quota by the specified amount. It replaces the previous code that directly updated the quota values with a call to the DecreaseTokenQuota function from the model package. This change improves code clarity and maintainability.

Closes #123
This commit is contained in:
Laisky.Cai 2024-06-25 06:15:07 +00:00
parent 671039f050
commit 4fd0b247d2
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
controller/token.go
View File

@ -279,8 +279,14 @@ func ConsumeToken(c *gin.Context) {
return return
} }
cleanToken.UsedQuota += int64(tokenPatch.AddUsedQuota) if err = model.DecreaseTokenQuota(cleanToken.Id, int64(tokenPatch.AddUsedQuota)); err != nil {
cleanToken.RemainQuota -= int64(tokenPatch.AddUsedQuota) c.JSON(http.StatusOK, gin.H{
"success": false,
"message": err.Error(),
})
return
}
model.RecordConsumeLog(c.Request.Context(), model.RecordConsumeLog(c.Request.Context(),
userID, 0, 0, 0, tokenPatch.AddReason, cleanToken.Name, userID, 0, 0, 0, tokenPatch.AddReason, cleanToken.Name,
int64(tokenPatch.AddUsedQuota), int64(tokenPatch.AddUsedQuota),