xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-18 00:16:37 +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

2
model/token.go
View File

@ -250,12 +250,10 @@ 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
} }