chiyiandDevin
b4efe6b192
- 新增 channels/channel_models:上游 URL + AES-GCM 加密 key +
并发/权重/本地统计,模型经渠道 /v1/models 导入绑定
- internal/upstream:vivid 协议客户端(NDJSON 流 + jobId 断流重连 +
uploads 预签名 + models + balance + 错误分类)
- service/v1 重写:权重分组调度、Redis 并发槽 conc:c:、故障转移,
站内/测试/对外统一走渠道执行器
- 对外端点换成 vivid 四件套:POST /v1/generate(NDJSON+jobId 重连)、
POST /v1/uploads(SigV4 预签名 PUT)、GET /v1/models、GET /v1/balance,
错误包 {"error":{"code","message"}}
- API key 新发改 vk_+48hex;删除全部内置 provider 与 token 池
- 管理端:渠道 CRUD/拉模型/导入/余额端点,前端账号页换渠道页,
概览/文档/测试弹窗适配
Generated with [Devin](https://devin.ai)
Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
304 lines
13 KiB
Go
304 lines
13 KiB
Go
package model
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"gorm.io/datatypes"
|
|
)
|
|
|
|
type User struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
Email string `gorm:"size:255;uniqueIndex;not null"`
|
|
Name string `gorm:"size:255"`
|
|
PasswordHash string `gorm:"size:255"`
|
|
Role string `gorm:"size:32;index;not null"`
|
|
Status string `gorm:"size:32;index;not null"`
|
|
Credits float64 `gorm:"not null;default:0"`
|
|
CreditsUsed float64 `gorm:"not null;default:0"` // 累计已消耗额度：扣费时加、退款时减，充值/发放不计入
|
|
Notes string `gorm:"type:text"`
|
|
ConcurrencyGroupID string `gorm:"size:32;index"`
|
|
AnnouncementSeen string `gorm:"size:32"` // version hash of the last announcement this user dismissed
|
|
RechargeTotal float64 `gorm:"not null;default:0"` // 累计充值金额(元)
|
|
InviteCode string `gorm:"size:32;uniqueIndex"`
|
|
InvitedBy *string `gorm:"size:32;index"`
|
|
InviteRewardDone bool `gorm:"not null;default:false"`
|
|
InviteRewardAt *time.Time
|
|
CheckinLast string `gorm:"size:32"`
|
|
CheckinStreak int `gorm:"not null;default:0"`
|
|
GenerationCount int64 `gorm:"not null;default:0"`
|
|
BannedWordHits int64 `gorm:"not null;default:0"` // 提示词命中违禁词被拦截的累计次数
|
|
LastLoginAt *time.Time
|
|
LastLoginIP string `gorm:"size:128"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
APIKeys []APIKey `gorm:"foreignKey:UserID"`
|
|
}
|
|
|
|
// BannedWord is an admin-managed prompt blocklist entry. Generation requests
|
|
// whose prompt contains Word (case-insensitive substring) are rejected before
|
|
// reaching any provider; Hits counts how many requests each word blocked.
|
|
type BannedWord struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
Word string `gorm:"size:255;uniqueIndex;not null"`
|
|
Hits int64 `gorm:"not null;default:0"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
// BannedWordHit records one blocked request: which word matched, who sent it,
|
|
// and when. Feeds the admin 违禁词触发列表.
|
|
type BannedWordHit struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
WordID string `gorm:"size:32;index"`
|
|
Word string `gorm:"size:255;index;not null"`
|
|
UserID string `gorm:"size:32;index"`
|
|
UserName string `gorm:"size:255"` // snapshot of name/email at hit time
|
|
Prompt string `gorm:"type:text"`
|
|
CreatedAt time.Time `gorm:"index"`
|
|
}
|
|
|
|
type APIKey struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
UserID string `gorm:"size:32;index;not null"`
|
|
Name string `gorm:"size:100;not null"`
|
|
KeyPreview string `gorm:"size:32;not null"`
|
|
KeyHash string `gorm:"size:255;uniqueIndex;not null"`
|
|
CreatedAt time.Time
|
|
LastUsedAt *time.Time
|
|
}
|
|
|
|
type ShowcaseItem struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
Kind string `gorm:"size:32;index;not null"`
|
|
Title string `gorm:"size:255"`
|
|
Subtitle string `gorm:"size:255"`
|
|
Prompt string `gorm:"type:text"`
|
|
Gradient string `gorm:"type:text"`
|
|
Span string `gorm:"size:100"`
|
|
Image string `gorm:"size:500;index"`
|
|
Weight int `gorm:"not null;default:0"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
type EventLog struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
TS time.Time `gorm:"index;not null"`
|
|
Kind string `gorm:"size:32;index;not null"`
|
|
Status string `gorm:"size:32;index;not null"`
|
|
Model string `gorm:"size:255;index"`
|
|
Provider string `gorm:"size:100;index"`
|
|
Prompt string `gorm:"type:text"`
|
|
Ratio string `gorm:"size:32"`
|
|
Resolution string `gorm:"size:32"`
|
|
Duration string `gorm:"size:32"`
|
|
Refs int `gorm:"not null;default:0"`
|
|
DeAI bool `gorm:"not null;default:false"` // 去AI特征 was applied (image only)
|
|
RefFiles datatypes.JSON `gorm:"type:jsonb"` // relative paths of saved reference images, for回显 on reload
|
|
Source string `gorm:"size:32;index"`
|
|
// AccountID is the channel chosen to fulfil this generation, stamped when
|
|
// the upstream call begins. (Field name kept from the pre-channel schema.)
|
|
AccountID string `gorm:"size:64;index"`
|
|
// AccountEmail denormalizes the channel's display name at stamp time, so log
|
|
// rows keep showing which channel fulfilled the generation even after the
|
|
// channel is deleted or renamed.
|
|
AccountEmail string `gorm:"size:255"`
|
|
// UpstreamJob is the job id assigned by the upstream vivid instance (from
|
|
// the NDJSON stream's jobId field) — kept for cross-instance debugging.
|
|
UpstreamJob string `gorm:"size:64"`
|
|
UserID string `gorm:"size:32;index"`
|
|
Cost float64 `gorm:"not null;default:0"`
|
|
// Refunded marks that this event's up-front charge has already been credited
|
|
// back, so the normal failure path and the abandoned-purge sweep can never
|
|
// double-refund the same generation.
|
|
Refunded bool `gorm:"not null;default:false"`
|
|
ElapsedMS int `gorm:"not null;default:0"`
|
|
File string `gorm:"size:500;index"`
|
|
Error string `gorm:"type:text"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
type ModelConfig struct {
|
|
ID string `gorm:"primaryKey;size:255"`
|
|
Type string `gorm:"size:32;index;not null"`
|
|
Name string `gorm:"size:255;not null"`
|
|
Alias string `gorm:"column:alias;size:255;not null;default:''"`
|
|
Provider string `gorm:"size:100;index;not null"`
|
|
Enabled bool `gorm:"not null;default:true"`
|
|
FreeAllowed bool `gorm:"not null;default:false"` // 普号(free)能否调度此模型，默认不可用
|
|
Ratios datatypes.JSON `gorm:"type:jsonb"`
|
|
Prices datatypes.JSONMap `gorm:"type:jsonb"`
|
|
Resolutions datatypes.JSON `gorm:"type:jsonb"`
|
|
ImageToImage bool `gorm:"not null;default:false"`
|
|
DurationPrices datatypes.JSONMap `gorm:"type:jsonb"`
|
|
// Agent (代理) pricing — optional overlay over Prices/DurationPrices. A tier
|
|
// left unset here means agent users pay the normal price for that tier; the
|
|
// set of *supported* tiers is always driven by Prices, not these.
|
|
PricesAgent datatypes.JSONMap `gorm:"type:jsonb;column:prices_agent"`
|
|
DurationPricesAgent datatypes.JSONMap `gorm:"type:jsonb;column:duration_prices_agent"`
|
|
Durations datatypes.JSON `gorm:"type:jsonb"`
|
|
// DurationMin/DurationMax bound a free-form duration (seconds) for models
|
|
// whose duration spec is a range rather than discrete options. 0 = unset.
|
|
DurationMin int `gorm:"not null;default:0"`
|
|
DurationMax int `gorm:"not null;default:0"`
|
|
MaxReferenceImages int `gorm:"not null;default:0"`
|
|
// Per-kind reference caps for upstreams that accept video/audio refs.
|
|
// 0 = that kind is not accepted.
|
|
MaxReferenceVideos int `gorm:"not null;default:0"`
|
|
MaxReferenceAudios int `gorm:"not null;default:0"`
|
|
// MaxRefVideoSeconds/MaxRefAudioSeconds cap the combined duration of
|
|
// reference videos/audios. 0 = unlimited.
|
|
MaxRefVideoSeconds int `gorm:"not null;default:0"`
|
|
MaxRefAudioSeconds int `gorm:"not null;default:0"`
|
|
ReferenceMode string `gorm:"size:32;not null;default:'none'"`
|
|
// Weight controls display order in the model dropdown / admin list: higher
|
|
// weight floats to the top (matches ShowcaseItem.Weight semantics). Ties fall
|
|
// back to created_at desc. Default 0.
|
|
Weight int `gorm:"not null;default:0;index"`
|
|
// GenerationCount is a persistent success counter, incremented once per
|
|
// successful generation. Independent of the event_log (which is subject to
|
|
// retention / manual clearing), so the admin "次数" is a true running total.
|
|
GenerationCount int64 `gorm:"not null;default:0"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
func (m ModelConfig) EffectiveName() string {
|
|
if strings.TrimSpace(m.Alias) != "" {
|
|
return strings.TrimSpace(m.Alias)
|
|
}
|
|
return m.ID
|
|
}
|
|
|
|
type CDKCode struct {
|
|
Code string `gorm:"primaryKey;size:32"`
|
|
Amount int `gorm:"not null"`
|
|
Status string `gorm:"size:32;index;not null"`
|
|
Type string `gorm:"size:16;not null;default:normal;index"` // normal | marketing
|
|
BatchID string `gorm:"size:32;index"` // groups one generate call
|
|
Note string `gorm:"type:text"`
|
|
RedeemedBy *string `gorm:"size:32;index"`
|
|
RedeemedAt *time.Time
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
// Channel is one upstream vivid-protocol instance this gateway routes
|
|
// generation to. Replaces the old per-provider token account pool: a channel
|
|
// is just a base_url + api_key plus routing knobs (enabled/weight/concurrency)
|
|
// and local usage counters. ApiKey is stored AES-GCM encrypted.
|
|
type Channel struct {
|
|
ID string `gorm:"primaryKey;size:40"`
|
|
Name string `gorm:"size:255;not null"`
|
|
BaseURL string `gorm:"size:500;not null"`
|
|
APIKeyEnc string `gorm:"type:text;not null"` // AES-GCM encrypted upstream API key
|
|
Enabled bool `gorm:"not null;default:true;index"`
|
|
// Concurrency caps simultaneous in-flight jobs on this channel. 0 = default.
|
|
Concurrency int `gorm:"not null;default:0"`
|
|
// Weight biases weighted-random channel order: weight 2 is picked first at
|
|
// roughly 2x the rate of weight 1.
|
|
Weight int `gorm:"not null;default:1"`
|
|
SuccessTotal int64 `gorm:"not null;default:0"` // locally-tracked successful generations
|
|
UsedCredits float64 `gorm:"not null;default:0"` // upstream credits spent (reported by the upstream)
|
|
LastUsedAt *time.Time
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
// ChannelModel binds a local model_configs row to a channel and records the
|
|
// upstream model name to send on the wire (the local row's alias stays the
|
|
// user-facing name).
|
|
type ChannelModel struct {
|
|
ChannelID string `gorm:"primaryKey;size:40"`
|
|
ModelID string `gorm:"primaryKey;size:255;index"`
|
|
UpstreamModel string `gorm:"size:255;not null"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
type SiteSetting struct {
|
|
Key string `gorm:"primaryKey;size:100"`
|
|
Value string `gorm:"type:text"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
func AutoMigrateModels() []any {
|
|
return []any{
|
|
&User{},
|
|
&BannedWord{},
|
|
&BannedWordHit{},
|
|
&APIKey{},
|
|
&ShowcaseItem{},
|
|
&EventLog{},
|
|
&ModelConfig{},
|
|
&CDKCode{},
|
|
&Channel{},
|
|
&ChannelModel{},
|
|
&SiteSetting{},
|
|
&StatCounter{},
|
|
&ConcurrencyGroup{},
|
|
&Order{},
|
|
&Canvas{},
|
|
}
|
|
}
|
|
|
|
// Order is a points-recharge order paid via 易支付 (epay). ID is our merchant
|
|
// order number (out_trade_no). Status: pending | paid | cancelled. Unpaid orders
|
|
// auto-cancel 30 min after creation (ExpiresAt). Besides epay recharges, the
|
|
// table also records credit grants from admin manual adjustments (source=admin)
|
|
// and CDK redemptions (source=cdk) as already-paid rows, so 订单管理 shows the
|
|
// full credit history in one place.
|
|
type Order struct {
|
|
ID string `gorm:"primaryKey;size:40"`
|
|
UserID string `gorm:"size:32;index;not null"`
|
|
Amount float64 `gorm:"not null"` // 充值金额(元)
|
|
Points int `gorm:"not null"` // 到账积分
|
|
PayType string `gorm:"size:16"` // wxpay | alipay | admin | cdk
|
|
Status string `gorm:"size:16;index;not null"` // pending | paid | cancelled
|
|
Source string `gorm:"size:16;index;not null;default:'epay'"` // epay | admin | cdk
|
|
Remark string `gorm:"type:text"` // e.g. 兑换码 code / 管理员操作说明
|
|
TradeNo string `gorm:"size:64;index"` // 易支付平台订单号
|
|
PayInfo string `gorm:"type:text"` // 二维码 url / 跳转 url
|
|
PayInfoType string `gorm:"size:16"` // qrcode | jump | html | ...
|
|
ExpiresAt time.Time `gorm:"index"`
|
|
PaidAt *time.Time
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
// ConcurrencyGroup caps how many generations a member user may run AT ONCE
|
|
// (across their API key + 画图台). MaxConcurrency 0 = unlimited. Exactly one
|
|
// group is IsDefault — new users are bound to it and it can't be deleted.
|
|
type ConcurrencyGroup struct {
|
|
ID string `gorm:"primaryKey;size:32"`
|
|
Name string `gorm:"size:100;not null"`
|
|
MaxConcurrency int `gorm:"not null;default:10"` // 0 = 不限制
|
|
IsDefault bool `gorm:"not null;default:false;index"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
// Canvas is one user's infinite-canvas document. Data stores the whole
|
|
// node/edge graph as a versioned JSON blob; saves are whole-document
|
|
// replacements (last write wins), which matches the single-editor frontend.
|
|
type Canvas struct {
|
|
ID string `gorm:"primaryKey;size:40"`
|
|
UserID string `gorm:"size:32;index;not null"`
|
|
Title string `gorm:"size:120;not null;default:'未命名画布'"`
|
|
Data string `gorm:"type:text"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time `gorm:"index"`
|
|
}
|
|
|
|
// StatCounter is a persistent monotonic counter (key → value), independent of the
|
|
// event_log (which is retention-pruned / clearable). Used for the dashboard
|
|
// cumulative cards (total/success/failed/image/video/api) so they never reset.
|
|
type StatCounter struct {
|
|
Key string `gorm:"primaryKey;size:64"`
|
|
Value int64 `gorm:"not null;default:0"`
|
|
UpdatedAt time.Time
|
|
}
|