oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: Improve quota error messages with formatted quota display

Browse Source
This commit is contained in:
1808837298@qq.com
2025-02-21 16:42:48 +08:00
parent 94736407a0
commit 2d42145b66
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
service/quota.go
View File

@@ -95,11 +95,11 @@ func PreWssConsumeQuota(ctx *gin.Context, relayInfo *relaycommon.RelayInfo, usag
quota := calculateAudioQuota(quotaInfo) quota := calculateAudioQuota(quotaInfo)
if userQuota < quota { if userQuota < quota {
return errors.New(fmt.Sprintf("用户额度不足，剩余额度为 %d", userQuota)) return fmt.Errorf("user quota is not enough, user quota: %s, need quota: %s", common.FormatQuota(userQuota), common.FormatQuota(quota))
} }
if !token.UnlimitedQuota && token.RemainQuota < quota { if !token.UnlimitedQuota && token.RemainQuota < quota {
return errors.New(fmt.Sprintf("令牌额度不足，剩余额度为 %d", token.RemainQuota)) return fmt.Errorf("token quota is not enough, token remain quota: %s, need quota: %s", common.FormatQuota(token.RemainQuota), common.FormatQuota(quota))
} }
err = PostConsumeQuota(relayInfo, quota, 0, false) err = PostConsumeQuota(relayInfo, quota, 0, false)
@@ -262,7 +262,7 @@ func PreConsumeTokenQuota(relayInfo *relaycommon.RelayInfo, quota int) error {
return err return err
} }
if !relayInfo.TokenUnlimited && token.RemainQuota < quota { if !relayInfo.TokenUnlimited && token.RemainQuota < quota {
return errors.New("令牌额度不足") return fmt.Errorf("token quota is not enough, token remain quota: %s, need quota: %s", common.FormatQuota(token.RemainQuota), common.FormatQuota(quota))
} }
err = model.DecreaseTokenQuota(relayInfo.TokenId, relayInfo.TokenKey, quota) err = model.DecreaseTokenQuota(relayInfo.TokenId, relayInfo.TokenKey, quota)
if err != nil { if err != nil {