xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: 优化按次计费的数据库查询次数

Browse Source
This commit is contained in:
CaIon 2024-04-04 20:10:30 +08:00
parent aaf3a1f07b
commit f3f36dafbd
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
relay/relay-text.go
View File

@ -288,11 +288,13 @@ func postConsumeQuota(ctx *gin.Context, relayInfo *relaycommon.RelayInfo, textRe
// logContent += fmt.Sprintf("，敏感词：%s", strings.Join(sensitiveResp.SensitiveWords, ", ")) // logContent += fmt.Sprintf("，敏感词：%s", strings.Join(sensitiveResp.SensitiveWords, ", "))
//} //}
quotaDelta := quota - preConsumedQuota quotaDelta := quota - preConsumedQuota
err := model.PostConsumeTokenQuota(relayInfo.TokenId, userQuota, quotaDelta, preConsumedQuota, true) if quotaDelta != 0 {
if err != nil { err := model.PostConsumeTokenQuota(relayInfo.TokenId, userQuota, quotaDelta, preConsumedQuota, true)
common.LogError(ctx, "error consuming token remain quota: "+err.Error()) if err != nil {
common.LogError(ctx, "error consuming token remain quota: "+err.Error())
}
} }
err = model.CacheUpdateUserQuota(relayInfo.UserId) err := model.CacheUpdateUserQuota(relayInfo.UserId)
if err != nil { if err != nil {
common.LogError(ctx, "error update user quota cache: "+err.Error()) common.LogError(ctx, "error update user quota cache: "+err.Error())
} }