Blizzard
693a8f09e9
按 SAAS_DESIGN.md P1 第一刀（真·多租户 + 桌面端为主）：先立租户身份，不动查询。 - store.Tenant / TenantMember 表 + AutoMigrate - store: CreateTenant / AddMember(幂等) / DefaultTenantForUser / EnsureDefaultTenant(幂等) / BackfillDefaultTenants / GetTenant / MemberRole - 注册即建单人默认租户(owner)；启动回填给存量用户补建(幂等) - middleware.TenantContext(挂 Auth 后)解析当前租户→注入 tenant_id；handler.tenantID(c) 助手 - GET /api/v1/tenants/current 验证端点(租户上下文 + 角色) - live：存量用户(回填 7 租户)/tenants/current 返回默认租户+owner；新注册自动建租户 下一步(增量2)：核心表加 tenant_id + 统一 gorm scope 强制隔离 + 存量行回填 + 重写查询。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
97 lines
4.6 KiB
Go
97 lines
4.6 KiB
Go
package store
|
||
|
||
// 数据库映射模型。表名经 GORM NamingStrategy 统一加 sundynix_ 前缀 + 单数：
|
||
// User → sundynix_user，Task → sundynix_task。
|
||
// 约定：均嵌入 BaseModel（雪花字符串 id + created_at/updated_at + 软删 deleted_at）；
|
||
// 建表/改表一律走 AutoMigrate，不手写 DDL。
|
||
|
||
// User 是平台用户（Users）。
|
||
type User struct {
|
||
BaseModel
|
||
Email string `gorm:"uniqueIndex;size:255"`
|
||
Name string `gorm:"size:64"`
|
||
PasswordHash string `gorm:"size:255" json:"-"` // bcrypt；绝不出 JSON
|
||
}
|
||
|
||
// Task 是一次提交的 Agent 编排任务（DSL）。
|
||
// 业务 id（task_xxx，用于 NATS subject/stream）单列 TaskID，主键统一雪花。
|
||
type Task struct {
|
||
BaseModel
|
||
TaskID string `gorm:"uniqueIndex;size:64"` // task_xxx
|
||
Graph string `gorm:"type:jsonb"` // React Flow 导出的 DSL 原文
|
||
Status string `gorm:"size:32"` // submitted / running / done / failed / timeout
|
||
Detail string `gorm:"type:text"` // 失败/超时原因等（状态机回写）
|
||
// 收尾持久化：供「运行历史复盘」永久回放（Redis 流仅 10min TTL，过期后历史任务靠这两列）。
|
||
Output string `gorm:"type:text"` // 最终模型输出（收尾时由网关从流快照落库）
|
||
Trace string `gorm:"type:text"` // 执行轨迹事件 JSON 数组（存为文本，容忍空串；不在库内查它）
|
||
}
|
||
|
||
// Eval 是一次任务的自动化评测结果（dispatcher 评完经 NATS 回写，每任务一条，按 task_id upsert）。
|
||
type Eval struct {
|
||
BaseModel
|
||
TaskID string `gorm:"uniqueIndex;size:64"`
|
||
Overall float64 // 综合分 [0,1]
|
||
Rule float64 // 规则分
|
||
LLM float64 // LLM 质量分
|
||
Faithful float64 // RAG 忠实度分（0=无来源未评）
|
||
Level string `gorm:"size:16"` // ok / warn / poor
|
||
Flags string `gorm:"type:text"` // 命中问题（JSON 数组字符串）
|
||
Reason string `gorm:"type:text"` // 评语
|
||
Sources int // 检索来源数
|
||
Corrected bool // 是否经低分自动纠偏重生成后采纳（恒温器闭环）
|
||
}
|
||
|
||
func (Eval) TableName() string { return "sundynix_eval" }
|
||
|
||
// AuditLog 是一条敏感操作留痕（改模型/密钥/激活 prompt/审批 等）。
|
||
// 由 Audit 中间件在请求收尾时 best-effort 写入；只增不改，供运维溯源。
|
||
type AuditLog struct {
|
||
BaseModel
|
||
Actor string `gorm:"size:64;index"` // 操作者 uid（未登录/系统留空）
|
||
Action string `gorm:"size:8"` // HTTP 方法：POST / PUT / DELETE
|
||
Route string `gorm:"size:128"` // 路由模式，如 /api/v1/admin/models/:id
|
||
Path string `gorm:"size:256"` // 实际请求路径
|
||
Status int // HTTP 响应状态码
|
||
IP string `gorm:"size:64"`
|
||
Detail string `gorm:"type:text"` // 备注（可选，如目标名/关键参数）
|
||
}
|
||
|
||
func (AuditLog) TableName() string { return "sundynix_audit_log" }
|
||
|
||
// GuardrailEvent 是一次输入护栏命中（硬拦截 blocked / 灰区 suspect）。
|
||
// 由 Guardrail 中间件命中时 best-effort 写入；供安全溯源"谁触发多少次护栏"。
|
||
type GuardrailEvent struct {
|
||
BaseModel
|
||
Actor string `gorm:"size:64;index"` // 操作者 uid（未登录留空）
|
||
Kind string `gorm:"size:16;index"` // blocked（硬拦）/ suspect（灰区放行）
|
||
Reason string `gorm:"size:256"` // 拦截原因（blocked）
|
||
Signals string `gorm:"type:text"` // 命中软信号 JSON 数组（suspect）
|
||
Method string `gorm:"size:8"`
|
||
Path string `gorm:"size:256"`
|
||
IP string `gorm:"size:64"`
|
||
}
|
||
|
||
func (GuardrailEvent) TableName() string { return "sundynix_guardrail_event" }
|
||
|
||
// Tenant 是多租户的计费/隔离单位（组织/账户）。个人用户 = 一个单人默认租户；团队/企业 = 多成员。
|
||
type Tenant struct {
|
||
BaseModel
|
||
Name string `gorm:"size:128"`
|
||
Slug string `gorm:"size:64;uniqueIndex"` // 唯一短标识（默认租户用 default-<uid>）
|
||
Plan string `gorm:"size:32;default:free"` // free / pro / enterprise
|
||
Status string `gorm:"size:16;default:active"` // active / suspended
|
||
}
|
||
|
||
func (Tenant) TableName() string { return "sundynix_tenant" }
|
||
|
||
// TenantMember 是用户与租户的成员关系（带角色）。一个用户可属多个租户。
|
||
type TenantMember struct {
|
||
BaseModel
|
||
TenantID string `gorm:"size:64;uniqueIndex:idx_tenant_user;index"`
|
||
UserID string `gorm:"size:64;uniqueIndex:idx_tenant_user;index"`
|
||
Role string `gorm:"size:16"` // owner / admin / member / viewer / billing_admin
|
||
Status string `gorm:"size:16;default:active"` // active / invited / removed
|
||
}
|
||
|
||
func (TenantMember) TableName() string { return "sundynix_tenant_member" }
|