db

type CostFunc func(model string, inputTokens, outputTokens int) float64

type GlobalStats struct {
	TotalInput   int64
	TotalOutput  int64
	TotalTokens  int64 // = TotalInput + TotalOutput
	RequestCount int64
	ErrorCount   int64
}

type Group struct {
	ID                string `gorm:"primarykey"`
	Name              string `gorm:"uniqueIndex;not null"`
	DailyTokenLimit   *int64 // NULL = 无限制
	MonthlyTokenLimit *int64 // NULL = 无限制
	RequestsPerMinute *int   // NULL = 无限制（每分钟请求数 RPM）
	CreatedAt         time.Time
}

type GroupRepo struct {
	// contains filtered or unexported fields
}

type Peer struct {
	ID           string `gorm:"primarykey"` // e.g. "sp-2"
	Addr         string `gorm:"uniqueIndex;not null"`
	Weight       int    `gorm:"default:50"`
	IsActive     bool   `gorm:"default:true"`
	RegisteredAt time.Time
	LastSeen     *time.Time
}

type RefreshToken struct {
	JTI       string    `gorm:"primarykey"`
	UserID    string    `gorm:"not null;index"`
	ExpiresAt time.Time `gorm:"not null"`
	Revoked   bool      `gorm:"default:false"`
	CreatedAt time.Time
}

type RefreshTokenRepo struct {
	// contains filtered or unexported fields
}

type UsageFilter struct {
	UserID  string
	GroupID string
	From    *time.Time
	To      *time.Time
	Model   string
	Limit   int
	Offset  int
}

type UsageLog struct {
	ID           uint   `gorm:"primarykey;autoIncrement"`
	RequestID    string `gorm:"uniqueIndex;not null"` // 幂等防重
	UserID       string `gorm:"not null;index"`
	Model        string
	InputTokens  int  `gorm:"default:0"`
	OutputTokens int  `gorm:"default:0"`
	TotalTokens  int  `gorm:"default:0"`
	IsStreaming  bool `gorm:"default:false"`
	UpstreamURL  string
	StatusCode   int
	DurationMs   int64
	CostUSD      float64   `gorm:"default:0"`           // 估算费用（USD）
	SourceNode   string    `gorm:"default:'local'"`     // 数据来源节点 ID
	Synced       bool      `gorm:"default:false;index"` // sp-2+ 用：是否已上报给 sp-1
	CreatedAt    time.Time `gorm:"index"`
}

type UsageRecord struct {
	RequestID    string
	UserID       string
	Model        string
	InputTokens  int
	OutputTokens int
	IsStreaming  bool
	UpstreamURL  string
	StatusCode   int
	DurationMs   int64
	SourceNode   string
	CreatedAt    time.Time
}

type UsageRepo struct {
	// contains filtered or unexported fields
}

type UsageWriter struct {
	// contains filtered or unexported fields
}

type User struct {
	ID           string  `gorm:"primarykey"`
	Username     string  `gorm:"uniqueIndex;not null"`
	PasswordHash string  `gorm:"not null"`
	GroupID      *string `gorm:"index"` // NULL 表示未分配分组
	Group        Group   `gorm:"foreignKey:GroupID"`
	IsActive     bool    `gorm:"default:true"`
	CreatedAt    time.Time
	LastLoginAt  *time.Time
}

type UserRepo struct {
	// contains filtered or unexported fields
}

type UserStatRow struct {
	UserID       string
	TotalInput   int64
	TotalOutput  int64
	RequestCount int64
}