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

feat: 统计无限令牌的已用额度 (close #308)

Browse Source
This commit is contained in:
CalciumIon 2024-07-05 20:28:17 +08:00
parent 752048dfb4
commit 7011083201
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
model/token.go
View File

@ -250,11 +250,9 @@ func PreConsumeTokenQuota(tokenId int, quota int) (userQuota int, err error) {
if userQuota < quota { if userQuota < quota {
return 0, errors.New(fmt.Sprintf("用户额度不足，剩余额度为 %d", userQuota)) return 0, errors.New(fmt.Sprintf("用户额度不足，剩余额度为 %d", userQuota))
} }
if !token.UnlimitedQuota { err = DecreaseTokenQuota(tokenId, quota)
err = DecreaseTokenQuota(tokenId, quota) if err != nil {
if err != nil { return 0, err
return 0, err
}
} }
err = DecreaseUserQuota(token.UserId, quota) err = DecreaseUserQuota(token.UserId, quota)
return userQuota - quota, err return userQuota - quota, err