sundynix/sundynix-agentix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(gateway): 多租户增量2 —— gorm 租户插件按上下文自动隔离查询（SaaS P1）

Browse Source 
统一强制、别靠人肉：受租户模型标记 isTenantScoped() 后，store/tenant_scope.go
的 gorm 回调按请求 ctx 自动给查询加 WHERE tenant_id、创建自动填 tenant_id。
- KB / Agent 加 TenantID 字段 + isTenantScoped() 标记
- middleware.TenantContext 把 tenant 注入 request context 供 store 插件读取
- ctx 无租户（系统/回填/未登录）不过滤，保留跨租户操作能力
- 启动 BackfillRowTenants 回填存量行 tenant_id=owner 默认租户（幂等）

live 验证：创建自动写 tenant_id ✓；同 owner 不同 tenant 的行被查询过滤 ✓。
Doc/DocLink（异步入库）、Task/Eval（无 owner）留待增量2b。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Blizzard
2026-07-06 17:58:58 +08:00
parent f7d12cbf66
commit 30d667954b
6 changed files with 116 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sundynix-gateway/internal/store/model.go
+11 -7
View File
@@ -16,12 +16,14 @@ import (
// 向量/全文/图谱实际以 "owner/name" 作分区键，保证只有 owner 能查到自己的库。
type KB struct {
BaseModel
Owner string `gorm:"size:64;uniqueIndex:idx_kb_owner_name"`
Name string `gorm:"size:64;uniqueIndex:idx_kb_owner_name"`
Kind string `gorm:"size:16"` // folder / project / case / general
TenantID string `gorm:"size:64;index"` // 多租户作用域（gorm tenant 插件按上下文自动填/过滤）
Owner string `gorm:"size:64;uniqueIndex:idx_kb_owner_name"`
Name string `gorm:"size:64;uniqueIndex:idx_kb_owner_name"`
Kind string `gorm:"size:16"` // folder / project / case / general
}
func (KB) TableName() string { return "sundynix_kb" }
func (KB) TableName() string { return "sundynix_kb" }
func (KB) isTenantScoped() {}
// ListKB 列出某 owner 的全部知识库（按创建时间）。
func (p *Postgres) ListKB(ctx context.Context, owner string) ([]KB, error) {
@@ -51,12 +53,14 @@ func (p *Postgres) EnsureKB(ctx context.Context, owner, name, kind string) error
// 表名 sundynix_agent。(owner,name) 唯一 —— 同一用户下编排名不重复。
type Agent struct {
BaseModel
Owner string `gorm:"size:64;uniqueIndex:idx_agent_on"`
Name string `gorm:"size:128;uniqueIndex:idx_agent_on"`
Graph string `gorm:"type:text"` // {nodes,edges} 的 JSON（含布局）
TenantID string `gorm:"size:64;index"` // 多租户作用域（tenant 插件自动填/过滤）
Owner string `gorm:"size:64;uniqueIndex:idx_agent_on"`
Name string `gorm:"size:128;uniqueIndex:idx_agent_on"`
Graph string `gorm:"type:text"` // {nodes,edges} 的 JSON（含布局）
}
func (Agent) TableName() string { return "sundynix_agent" }
func (Agent) isTenantScoped() {}
// ListAgents 返回某 owner 的全部编排（最近更新在前）。
func (p *Postgres) ListAgents(ctx context.Context, owner string) ([]Agent, error) {
sundynix-gateway/internal/store/pgsql.go
+1
View File
@@ -70,6 +70,7 @@ func OpenPostgres(dsn string) *Postgres {
log.Printf("[store] postgres AutoMigrate 失败，降级运行: %v", err)
return &Postgres{}
}
registerTenantScope(db) // 多租户：受租户模型的查询/创建自动按上下文注入 tenant_id（统一强制隔离）
log.Println("[store] postgres connected & migrated (雪花 id + 软删 规约)")
return &Postgres{db: db}
}
sundynix-gateway/internal/store/tenant_scope.go
+91
View File
@@ -0,0 +1,91 @@
package store
import (
"context"
"reflect"
"gorm.io/gorm"
"gorm.io/gorm/clause"
)
// tenantScopedMarker：模型实现此(空)接口即"受租户隔离"——gorm tenant 插件据此对其
// 查询/更新/删除自动加 `tenant_id = <ctx租户>`、创建时自动填 tenant_id。
// 未实现的模型（User/Tenant/TenantMember/Prompt/Model/Pricing…）不受影响。
// 未实现方法用小写→只有本包模型能标记（KB/Agent 已标；Doc/DocLink/Task 待增量铺开）。
type tenantScopedMarker interface{ isTenantScoped() }
// ---- 请求上下文携带 tenant_id（中间件注入 → 传到 store → 插件读取）----
type ctxKeyTenant struct{}
// WithTenant 把 tenant_id 放进 context（空则原样返回，避免污染系统/回填查询）。
func WithTenant(ctx context.Context, tenantID string) context.Context {
if tenantID == "" {
return ctx
}
return context.WithValue(ctx, ctxKeyTenant{}, tenantID)
}
func tenantFromCtx(ctx context.Context) string {
if ctx == nil {
return ""
}
s, _ := ctx.Value(ctxKeyTenant{}).(string)
return s
}
// registerTenantScope 挂 gorm 回调，实现"统一强制、别靠人肉"的租户隔离。
func registerTenantScope(db *gorm.DB) {
q := func(d *gorm.DB) { addTenantWhere(d) }
_ = db.Callback().Query().Before("gorm:query").Register("tenant:query", q)
_ = db.Callback().Update().Before("gorm:update").Register("tenant:update", q)
_ = db.Callback().Delete().Before("gorm:delete").Register("tenant:delete", q)
_ = db.Callback().Create().Before("gorm:create").Register("tenant:create", setTenantOnCreate)
}
func isTenantScopedStmt(db *gorm.DB) bool {
if db.Statement.Schema == nil {
return false
}
_, ok := reflect.New(db.Statement.Schema.ModelType).Interface().(tenantScopedMarker)
return ok
}
// addTenantWhere：受租户模型 + ctx 有 tenant → 追加 tenant_id 过滤。
// ctx 无 tenant（系统/回填/未登录）→ 不过滤（这些路径本就需跨租户；用户面由中间件保证有 tenant）。
func addTenantWhere(db *gorm.DB) {
if !isTenantScopedStmt(db) {
return
}
if tid := tenantFromCtx(db.Statement.Context); tid != "" {
db.Statement.AddClause(clause.Where{Exprs: []clause.Expression{
clause.Eq{Column: clause.Column{Table: db.Statement.Table, Name: "tenant_id"}, Value: tid},
}})
}
}
// setTenantOnCreate：受租户模型 + ctx 有 tenant → 强制把 tenant_id 设为 ctx 租户（防越权写他租）。
func setTenantOnCreate(db *gorm.DB) {
if !isTenantScopedStmt(db) {
return
}
if tid := tenantFromCtx(db.Statement.Context); tid != "" {
db.Statement.SetColumn("TenantID", tid)
}
}
// BackfillRowTenants 给受租户表的存量行回填 tenant_id = owner 的默认租户（幂等：只填空的）。
// 系统操作、无 ctx 租户 → 不受插件过滤影响。当前覆盖 KB / Agent（同步路径已接插件）。
func (p *Postgres) BackfillRowTenants(ctx context.Context) error {
if p.db == nil {
return errStoreDisabled
}
// 子查询：owner(user) → 其默认(最早 active)租户。
sub := "SELECT tenant_id FROM sundynix_tenant_member m WHERE m.user_id = t.owner AND m.status = 'active' ORDER BY m.created_at ASC LIMIT 1"
for _, tbl := range []string{"sundynix_kb", "sundynix_agent"} {
q := "UPDATE " + tbl + " t SET tenant_id = (" + sub + ") WHERE (t.tenant_id IS NULL OR t.tenant_id = '')"
if err := p.db.WithContext(ctx).Exec(q).Error; err != nil {
return err
}
}
return nil
}