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

feat: add batch update support (close #414)

Browse Source
This commit is contained in:
JustSong
2023-09-03 14:58:20 +08:00
parent 04acdb1ccb
commit c3dc315e75
7 changed files with 136 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
model/channel.go
View File

@@ -141,6 +141,14 @@ func UpdateChannelStatusById(id int, status int) {
}
func UpdateChannelUsedQuota(id int, quota int) {
if common.BatchUpdateEnabled {
addNewRecord(BatchUpdateTypeChannelUsedQuota, id, quota)
return
}
updateChannelUsedQuota(id, quota)
}
func updateChannelUsedQuota(id int, quota int) {
err := DB.Model(&Channel{}).Where("id = ?", id).Update("used_quota", gorm.Expr("used_quota + ?", quota)).Error
if err != nil {
common.SysError("failed to update channel used quota: " + err.Error())