fix: aiproxy sql constraint conv to hook (#5219)

* fix: constraint conv to hook * fix: conv json serializer to fastjson * fix: conv json serializer to fastjson
labring · Nov 20, 2024 · af26adb · af26adb
1 parent f5569ff
commit af26adb
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 9 deletions.
diff --git a/service/aiproxy/model/channel.go b/service/aiproxy/model/channel.go
@@ -31,12 +31,12 @@ type Channel struct {
 	TestAt           time.Time         `json:"test_at"`
 	BalanceUpdatedAt time.Time         `json:"balance_updated_at"`
 	ModelMapping     map[string]string `gorm:"serializer:fastjson;type:text" json:"model_mapping"`
-	Config           ChannelConfig     `gorm:"serializer:json;type:text"     json:"config"`
+	Config           ChannelConfig     `gorm:"serializer:fastjson;type:text" json:"config"`
 	Other            string            `json:"other"`
 	Key              string            `gorm:"type:text;index"               json:"key"`
 	Name             string            `gorm:"index"                         json:"name"`
 	BaseURL          string            `gorm:"index"                         json:"base_url"`
-	Models           []string          `gorm:"serializer:json;type:text"     json:"models"`
+	Models           []string          `gorm:"serializer:fastjson;type:text" json:"models"`
 	Balance          float64           `json:"balance"`
 	ResponseDuration int64             `gorm:"index"                         json:"response_duration"`
 	ID               int               `gorm:"primaryKey"                    json:"id"`

diff --git a/service/aiproxy/model/group.go b/service/aiproxy/model/group.go
@@ -25,12 +25,16 @@ const (
 type Group struct {
 	CreatedAt    time.Time `json:"created_at"`
 	AccessedAt   time.Time `json:"accessed_at"`
-	ID           string    `gorm:"primaryKey"                                                      json:"id"`
-	Tokens       []*Token  `gorm:"foreignKey:GroupID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE" json:"-"`
-	Status       int       `gorm:"default:1;index"                                                 json:"status"`
-	UsedAmount   float64   `gorm:"index"                                                           json:"used_amount"`
-	QPM          int64     `gorm:"index"                                                           json:"qpm"`
-	RequestCount int       `gorm:"index"                                                           json:"request_count"`
+	ID           string    `gorm:"primaryKey"         json:"id"`
+	Tokens       []*Token  `gorm:"foreignKey:GroupID" json:"-"`
+	Status       int       `gorm:"default:1;index"    json:"status"`
+	UsedAmount   float64   `gorm:"index"              json:"used_amount"`
+	QPM          int64     `gorm:"index"              json:"qpm"`
+	RequestCount int       `gorm:"index"              json:"request_count"`
+}
+
+func (g *Group) AfterDelete(tx *gorm.DB) (err error) {
+	return tx.Model(&Token{}).Where("group_id = ?", g.ID).Delete(&Token{}).Error
 }
 
 func (g *Group) MarshalJSON() ([]byte, error) {

diff --git a/service/aiproxy/model/token.go b/service/aiproxy/model/token.go
@@ -35,7 +35,7 @@ type Token struct {
 	Name         EmptyNullString `gorm:"index;uniqueIndex:idx_group_name;not null" json:"name"`
 	GroupID      string          `gorm:"index;uniqueIndex:idx_group_name"          json:"group"`
 	Subnet       string          `json:"subnet"`
-	Models       []string        `gorm:"serializer:json;type:text"                 json:"models"`
+	Models       []string        `gorm:"serializer:fastjson;type:text"             json:"models"`
 	Status       int             `gorm:"default:1;index"                           json:"status"`
 	ID           int             `gorm:"primaryKey"                                json:"id"`
 	Quota        float64         `json:"quota"`