Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 fix: Update channel results in data loss (#108)

Browse Source
This commit is contained in:
Martial BE 2024-03-13 18:34:37 +08:00
parent 05347bc9a1
commit 3aae5d262c
No known key found for this signature in database
GPG Key ID: D06C32DF0EDB9084
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
controller/channel.go
View File

@ -132,7 +132,11 @@ func UpdateChannel(c *gin.Context) {
}) })
return return
} }
err = channel.Update() if channel.Models == "" {
err = channel.Update(false)
} else {
err = channel.Update(true)
}
if err != nil { if err != nil {
c.JSON(http.StatusOK, gin.H{ c.JSON(http.StatusOK, gin.H{
"success": false, "success": false,

9
model/channel.go
View File

@ -152,7 +152,7 @@ func BatchDelModelChannels(params *BatchChannelsParams) (int64, error) {
} }
channel.Models = strings.Join(modelsSlice, ",") channel.Models = strings.Join(modelsSlice, ",")
channel.Update() channel.Update(false)
count++ count++
} }
@ -190,9 +190,14 @@ func (channel *Channel) Insert() error {
return err return err
} }
func (channel *Channel) Update() error { func (channel *Channel) Update(overwrite bool) error {
var err error var err error
if overwrite {
err = DB.Model(channel).Select("*").Omit("UsedQuota").Updates(channel).Error err = DB.Model(channel).Select("*").Omit("UsedQuota").Updates(channel).Error
} else {
err = DB.Model(channel).Omit("UsedQuota").Updates(channel).Error
}
if err != nil { if err != nil {
return err return err
} }