fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 密钥管理

Browse Source
This commit is contained in:
tbphp
2025-07-04 21:19:15 +08:00
parent 7c10474d19
commit 01b86f7e30
23 changed files with 1427 additions and 250 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
internal/models/types.go
View File

@@ -1,9 +1,6 @@
package models
import (
"database/sql/driver"
"encoding/json"
"errors"
"time"
"gorm.io/datatypes"
@@ -19,52 +16,36 @@ type SystemSetting struct {
UpdatedAt time.Time `json:"updated_at"`
}
// Upstreams 是一个上游地址的切片，可以被 GORM 正确处理
type Upstreams []string
// Value 实现 driver.Valuer 接口，用于将 Upstreams 类型转换为数据库值
func (u Upstreams) Value() (driver.Value, error) {
if len(u) == 0 {
return "[]", nil
}
return json.Marshal(u)
}
// Scan 实现 sql.Scanner 接口，用于将数据库值扫描到 Upstreams 类型
func (u *Upstreams) Scan(value interface{}) error {
bytes, ok := value.([]byte)
if !ok {
return errors.New("type assertion to []byte failed")
}
return json.Unmarshal(bytes, u)
}
// GroupConfig 存储特定于分组的配置
type GroupConfig struct {
BlacklistThreshold *int `json:"blacklist_threshold,omitempty"`
MaxRetries *int `json:"max_retries,omitempty"`
ServerReadTimeout *int `json:"server_read_timeout,omitempty"`
ServerWriteTimeout *int `json:"server_write_timeout,omitempty"`
ServerIdleTimeout *int `json:"server_idle_timeout,omitempty"`
BlacklistThreshold *int `json:"blacklist_threshold,omitempty"`
MaxRetries *int `json:"max_retries,omitempty"`
ServerReadTimeout *int `json:"server_read_timeout,omitempty"`
ServerWriteTimeout *int `json:"server_write_timeout,omitempty"`
ServerIdleTimeout *int `json:"server_idle_timeout,omitempty"`
ServerGracefulShutdownTimeout *int `json:"server_graceful_shutdown_timeout,omitempty"`
RequestTimeout *int `json:"request_timeout,omitempty"`
ResponseTimeout *int `json:"response_timeout,omitempty"`
IdleConnTimeout *int `json:"idle_conn_timeout,omitempty"`
RequestTimeout *int `json:"request_timeout,omitempty"`
ResponseTimeout *int `json:"response_timeout,omitempty"`
IdleConnTimeout *int `json:"idle_conn_timeout,omitempty"`
KeyValidationIntervalMinutes *int `json:"key_validation_interval_minutes,omitempty"`
}
// Group 对应 groups 表
type Group struct {
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
Name string `gorm:"type:varchar(255);not null;unique" json:"name"`
DisplayName string `gorm:"type:varchar(255)" json:"display_name"`
Description string `gorm:"type:varchar(512)" json:"description"`
Upstreams Upstreams `gorm:"type:json;not null" json:"upstreams"`
ChannelType string `gorm:"type:varchar(50);not null" json:"channel_type"`
Sort int `gorm:"default:0" json:"sort"`
Config datatypes.JSONMap `gorm:"type:json" json:"config"`
APIKeys []APIKey `gorm:"foreignKey:GroupID" json:"api_keys"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
Name string `gorm:"type:varchar(255);not null;unique" json:"name"`
DisplayName string `gorm:"type:varchar(255)" json:"display_name"`
Description string `gorm:"type:varchar(512)" json:"description"`
Upstreams datatypes.JSON `gorm:"type:json;not null" json:"upstreams"`
ChannelType string `gorm:"type:varchar(50);not null" json:"channel_type"`
Sort int `gorm:"default:0" json:"sort"`
TestModel string `gorm:"type:varchar(255);not null" json:"test_model"`
ParamOverrides datatypes.JSONMap `gorm:"type:json" json:"param_overrides"`
Config datatypes.JSONMap `gorm:"type:json" json:"config"`
APIKeys []APIKey `gorm:"foreignKey:GroupID" json:"api_keys"`
LastValidatedAt *time.Time `json:"last_validated_at"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
// APIKey 对应 api_keys 表
@@ -94,7 +75,7 @@ type RequestLog struct {
// GroupRequestStat 用于表示每个分组的请求统计
type GroupRequestStat struct {
DisplayName string `json:"display_name"`
DisplayName string `json:"display_name"`
RequestCount int64 `json:"request_count"`
}