xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-12-25 08:05:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 状态码复写

Browse Source
This commit is contained in:
CaIon
2024-04-20 21:05:23 +08:00
parent b8fb351fd8
commit 6b71db7ce2
5 changed files with 70 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
model/channel.go
View File

@@ -25,6 +25,7 @@ type Channel struct {
Group string `json:"group" gorm:"type:varchar(64);default:'default'"`
UsedQuota int64 `json:"used_quota" gorm:"bigint;default:0"`
ModelMapping *string `json:"model_mapping" gorm:"type:varchar(1024);default:''"`
StatusCodeMapping *string `json:"status_code_mapping" gorm:"type:varchar(1024);default:''"`
Priority *int64 `json:"priority" gorm:"bigint;default:0"`
AutoBan *int `json:"auto_ban" gorm:"default:1"`
}
@@ -153,6 +154,13 @@ func (channel *Channel) GetModelMapping() string {
return *channel.ModelMapping
}
func (channel *Channel) GetStatusCodeMapping() string {
if channel.StatusCodeMapping == nil {
return ""
}
return *channel.StatusCodeMapping
}
func (channel *Channel) Insert() error {
var err error
err = DB.Create(channel).Error