112 lines
5.2 KiB
Go
112 lines
5.2 KiB
Go
package models
|
|
|
|
import (
|
|
"gpt-load/internal/types"
|
|
"time"
|
|
|
|
"gorm.io/datatypes"
|
|
)
|
|
|
|
// Key状态
|
|
const (
|
|
KeyStatusActive = "active"
|
|
KeyStatusInvalid = "invalid"
|
|
)
|
|
|
|
// SystemSetting 对应 system_settings 表
|
|
type SystemSetting struct {
|
|
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
|
|
SettingKey string `gorm:"type:varchar(255);not null;unique" json:"setting_key"`
|
|
SettingValue string `gorm:"type:text;not null" json:"setting_value"`
|
|
Description string `gorm:"type:varchar(512)" json:"description"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// GroupConfig 存储特定于分组的配置
|
|
type GroupConfig struct {
|
|
RequestTimeout *int `json:"request_timeout,omitempty"`
|
|
IdleConnTimeout *int `json:"idle_conn_timeout,omitempty"`
|
|
ConnectTimeout *int `json:"connect_timeout,omitempty"`
|
|
MaxIdleConns *int `json:"max_idle_conns,omitempty"`
|
|
MaxIdleConnsPerHost *int `json:"max_idle_conns_per_host,omitempty"`
|
|
ResponseHeaderTimeout *int `json:"response_header_timeout,omitempty"`
|
|
MaxRetries *int `json:"max_retries,omitempty"`
|
|
BlacklistThreshold *int `json:"blacklist_threshold,omitempty"`
|
|
KeyValidationIntervalMinutes *int `json:"key_validation_interval_minutes,omitempty"`
|
|
}
|
|
|
|
// Group 对应 groups 表
|
|
type Group struct {
|
|
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
|
|
EffectiveConfig types.SystemSettings `gorm:"-" json:"effective_config,omitempty"`
|
|
Name string `gorm:"type:varchar(255);not null;unique" json:"name"`
|
|
Endpoint string `gorm:"-" json:"endpoint"`
|
|
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 表
|
|
type APIKey struct {
|
|
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
|
|
KeyValue string `gorm:"type:varchar(512);not null;uniqueIndex:idx_group_key" json:"key_value"`
|
|
GroupID uint `gorm:"not null;uniqueIndex:idx_group_key" json:"group_id"`
|
|
Status string `gorm:"type:varchar(50);not null;default:'active'" json:"status"`
|
|
RequestCount int64 `gorm:"not null;default:0" json:"request_count"`
|
|
FailureCount int64 `gorm:"not null;default:0" json:"failure_count"`
|
|
LastUsedAt *time.Time `json:"last_used_at"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// RequestLog 对应 request_logs 表
|
|
type RequestLog struct {
|
|
ID string `gorm:"type:varchar(36);primaryKey" json:"id"`
|
|
Timestamp time.Time `gorm:"type:datetime(3);not null;index" json:"timestamp"`
|
|
GroupID uint `gorm:"not null;index" json:"group_id"`
|
|
KeyID uint `gorm:"not null;index" json:"key_id"`
|
|
IsSuccess bool `gorm:"not null" json:"is_success"`
|
|
SourceIP string `gorm:"type:varchar(45)" json:"source_ip"`
|
|
StatusCode int `gorm:"not null" json:"status_code"`
|
|
RequestPath string `gorm:"type:varchar(1024)" json:"request_path"`
|
|
Duration int64 `gorm:"not null" json:"duration_ms"`
|
|
ErrorMessage string `gorm:"type:text" json:"error_message"`
|
|
UserAgent string `gorm:"type:varchar(512)" json:"user_agent"`
|
|
Retries int `gorm:"not null" json:"retries"`
|
|
}
|
|
|
|
// GroupRequestStat 用于表示每个分组的请求统计
|
|
type GroupRequestStat struct {
|
|
DisplayName string `json:"display_name"`
|
|
RequestCount int64 `json:"request_count"`
|
|
}
|
|
|
|
// DashboardStats 用于仪表盘的统计数据
|
|
type DashboardStats struct {
|
|
TotalRequests int64 `json:"total_requests"`
|
|
SuccessRequests int64 `json:"success_requests"`
|
|
SuccessRate float64 `json:"success_rate"`
|
|
GroupStats []GroupRequestStat `json:"group_stats"`
|
|
}
|
|
|
|
// GroupHourlyStat 对应 group_hourly_stats 表，用于存储每个分组每小时的请求统计
|
|
type GroupHourlyStat struct {
|
|
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
|
|
Time time.Time `gorm:"type:datetime;not null;uniqueIndex:idx_group_time" json:"time"` // 整点时间
|
|
GroupID uint `gorm:"not null;uniqueIndex:idx_group_time" json:"group_id"`
|
|
SuccessCount int64 `gorm:"not null;default:0" json:"success_count"`
|
|
FailureCount int64 `gorm:"not null;default:0" json:"failure_count"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|