ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-15 23:54:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: return quota to user when delete token (close #37)

Browse Source
This commit is contained in:
JustSong
2023-05-04 10:20:39 +08:00
parent 4fed003f1a
commit 331177d97e
4 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
model/user.go
View File

@@ -225,6 +225,11 @@ func GetUserQuota(id int) (quota int, err error) {
return quota, err
}
func IncreaseUserQuota(id int, quota int) (err error) {
err = DB.Model(&User{}).Where("id = ?", id).Update("quota", gorm.Expr("quota + ?", quota)).Error
return err
}
func DecreaseUserQuota(id int, quota int) (err error) {
err = DB.Model(&User{}).Where("id = ?", id).Update("quota", gorm.Expr("quota - ?", quota)).Error
return err