feat(space): 共享工作区增量3a —— Agent 编排按 Space 共享(纯 PG 打样)
引入 Space 中间容器(租户>Space>成员),资源作用域从 owner 改为 space_id，
让"个人私有/项目临时组队/整租户共享"出自同一模型(设计见 SPACE_DESIGN.md)。
先在纯 PG 的 Agent 上打样，零存储风险，验证协作+RBAC+切换 UX。
后端:
- 新表 Space{tenant_id,name,kind,creator,archived} + SpaceMember{space_id,user_id,role}
(如 Tenant 般不 isTenantScoped)；User.ActiveSpaceID；Agent 作用域 owner→space_id，
owner 降级为创建人(供 UI 显示 / 删他人鉴权)
- store/space.go:个人空间幂等/活跃空间解析/切换/列表/建/成员CRUD/归档
- 迁移顺序坑:结构体只放非唯一 index，MigrateAgentSpaces 回填 space_id 后再建唯一
索引 idx_agent_sn + DROP 旧 idx_agent_on(否则存量空 space_id 撞车)；启动序4步幂等
- 中间件 SpaceContext(注入 space_id) + RequireSpaceRole(照 RequireTenantRole)
- handler/space.go 空间端点 + 路由；agent.go 改空间作用域(删/覆盖他人需 admin)
- 计费零改动(Space 与 ResolveBillingTenantID 正交)
桌面端:
- api.ts space 接口；顶栏 SpaceSwitcher(含新建项目空间)；StudioView 随空间切换
重拉编排 + viewer 禁保存；Agent 列表显示创建人 + 按 mine 控删除
验证:中间件6门控单测 + DB迁移(13个人空间/9 Agent全re-key/索引换新) + 后端HTTP全
场景(member见他人编排/删他人403、viewer存403、非成员切空间400+隔离、owner删他人200)
+ 浏览器实机(切换器3空间/Studio空间编排随切换隔离刷新/创建人显示/console无错)
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Blizzard
@@ -49,46 +49,80 @@ func (p *Postgres) EnsureKB(ctx context.Context, owner, name, kind string) error
|
||||
}).Create(&KB{Owner: owner, Name: name, Kind: kind}).Error
|
||||
}
|
||||
|
||||
// Agent 是一份保存的 Agent 编排（React Flow 图 JSON，按 owner 隔离）。
|
||||
// 表名 sundynix_agent。(owner,name) 唯一 —— 同一用户下编排名不重复。
|
||||
// Agent 是一份保存的 Agent 编排（React Flow 图 JSON）。
|
||||
// 表名 sundynix_agent。增量3：作用域从 owner 改为 space_id——(space_id,name) 唯一，
|
||||
// 同一工作区内编排名不重复；owner 降级为"创建人"归属（谁建的，供 UI 显示 / 删他人鉴权）。
|
||||
// 唯一索引 idx_agent_sn(space_id,name) 不在结构体标记里建（否则 AutoMigrate 早于回填、存量 space_id 空会撞车），
|
||||
// 而由启动迁移 MigrateAgentSpaces 在回填 space_id 后显式建；旧 idx_agent_on(owner,name) 同步 DROP。
|
||||
type Agent struct {
|
||||
BaseModel
|
||||
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"`
|
||||
TenantID string `gorm:"size:64;index"` // 多租户作用域（tenant 插件自动填/过滤，防越权写他租）
|
||||
SpaceID string `gorm:"size:64;index"` // 所属工作区（共享单位；唯一约束见 MigrateAgentSpaces）
|
||||
Owner string `gorm:"size:64;index"` // 创建人 user.id（归属，非隔离键）
|
||||
Name string `gorm:"size:128"`
|
||||
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) {
|
||||
// AgentInfo 是编排列表视图（含创建人邮箱/名字，供共享工作区显示"谁建的"）。
|
||||
type AgentInfo struct {
|
||||
Name string `json:"name"`
|
||||
Graph string `json:"graph"`
|
||||
Owner string `json:"owner"` // 创建人 user.id
|
||||
CreatorName string `json:"creator_name"`
|
||||
CreatorMail string `json:"creator_mail"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// ListAgents 返回某工作区(space)的全部编排（最近更新在前，join 创建人名字）。
|
||||
// tenant 插件仍会按 ctx 租户过滤（防越权）；空间是主作用域。
|
||||
func (p *Postgres) ListAgents(ctx context.Context, spaceID string) ([]AgentInfo, error) {
|
||||
if p.db == nil {
|
||||
return nil, nil
|
||||
}
|
||||
var rows []Agent
|
||||
err := p.db.WithContext(ctx).Where("owner = ?", owner).Order("updated_at desc").Find(&rows).Error
|
||||
return rows, err
|
||||
var out []AgentInfo
|
||||
err := p.db.WithContext(ctx).Table("sundynix_agent a").
|
||||
Select("a.name, a.graph, a.owner, u.name as creator_name, u.email as creator_mail, a.updated_at").
|
||||
Joins("LEFT JOIN sundynix_user u ON u.id = a.owner").
|
||||
Where("a.space_id = ? AND a.deleted_at IS NULL", spaceID).
|
||||
Order("a.updated_at desc").Scan(&out).Error
|
||||
return out, err
|
||||
}
|
||||
|
||||
// SaveAgent 新建/更新一份编排（owner+name 唯一，重名覆盖图与更新时间）。
|
||||
func (p *Postgres) SaveAgent(ctx context.Context, owner, name, graph string) error {
|
||||
// SaveAgent 新建/更新一份编排（space_id+name 唯一，重名覆盖图与更新时间；owner 首建时记创建人，覆盖不改）。
|
||||
func (p *Postgres) SaveAgent(ctx context.Context, spaceID, creator, name, graph string) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
return p.db.WithContext(ctx).Clauses(clause.OnConflict{
|
||||
Columns: []clause.Column{{Name: "owner"}, {Name: "name"}},
|
||||
Columns: []clause.Column{{Name: "space_id"}, {Name: "name"}},
|
||||
DoUpdates: clause.Assignments(map[string]any{"graph": graph, "updated_at": time.Now()}),
|
||||
}).Create(&Agent{Owner: owner, Name: name, Graph: graph}).Error
|
||||
}).Create(&Agent{SpaceID: spaceID, Owner: creator, Name: name, Graph: graph}).Error
|
||||
}
|
||||
|
||||
// DeleteAgent 删除某 owner 的一份编排。
|
||||
func (p *Postgres) DeleteAgent(ctx context.Context, owner, name string) error {
|
||||
// GetAgent 取某工作区内一份编排（含 owner，供删他人鉴权判定）。
|
||||
func (p *Postgres) GetAgent(ctx context.Context, spaceID, name string) (*Agent, error) {
|
||||
if p.db == nil {
|
||||
return nil, nil
|
||||
}
|
||||
var a Agent
|
||||
if err := p.db.WithContext(ctx).Where("space_id = ? AND name = ?", spaceID, name).First(&a).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &a, nil
|
||||
}
|
||||
|
||||
// DeleteAgent 删除某工作区的一份编排。
|
||||
func (p *Postgres) DeleteAgent(ctx context.Context, spaceID, name string) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
return p.db.WithContext(ctx).Where("owner = ? AND name = ?", owner, name).Delete(&Agent{}).Error
|
||||
return p.db.WithContext(ctx).Where("space_id = ? AND name = ?", spaceID, name).Delete(&Agent{}).Error
|
||||
}
|
||||
|
||||
// Doc 是入库的一份文件/笔记 —— 文件主表（供 Obsidian 式"文库"浏览：列表 + Markdown 阅读 + 双链）。
|
||||
|
||||
@@ -12,6 +12,7 @@ type User struct {
|
||||
Name string `gorm:"size:64"`
|
||||
PasswordHash string `gorm:"size:255" json:"-"` // bcrypt；绝不出 JSON
|
||||
ActiveTenantID string `gorm:"size:64" json:"-"` // 当前活跃租户（多租户切换；空=用默认）
|
||||
ActiveSpaceID string `gorm:"size:64" json:"-"` // 当前活跃工作区(Space)（增量3；空/失效=用活跃租户的个人空间）
|
||||
}
|
||||
|
||||
// Task 是一次提交的 Agent 编排任务（DSL）。
|
||||
|
||||
@@ -66,7 +66,7 @@ func OpenPostgres(dsn string) *Postgres {
|
||||
migrateLegacyIntIDs(db)
|
||||
migrateDocLinkToID(db)
|
||||
|
||||
if err := db.AutoMigrate(&User{}, &Task{}, &Eval{}, &LLMModel{}, &KB{}, &Doc{}, &Agent{}, &DocLink{}, &Pricing{}, &Prompt{}, &AuditLog{}, &GuardrailEvent{}, &Tenant{}, &TenantMember{}, &UsageEvent{}, &CreditLedger{}, &UsageRollup{}, &Setting{}); err != nil {
|
||||
if err := db.AutoMigrate(&User{}, &Task{}, &Eval{}, &LLMModel{}, &KB{}, &Doc{}, &Agent{}, &DocLink{}, &Pricing{}, &Prompt{}, &AuditLog{}, &GuardrailEvent{}, &Tenant{}, &TenantMember{}, &Space{}, &SpaceMember{}, &UsageEvent{}, &CreditLedger{}, &UsageRollup{}, &Setting{}); err != nil {
|
||||
log.Printf("[store] postgres AutoMigrate 失败，降级运行: %v", err)
|
||||
return &Postgres{}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,337 @@
|
||||
package store
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/clause"
|
||||
)
|
||||
|
||||
// Space 是一个「工作区/协作空间」——共享的资源容器（增量3，见 SPACE_DESIGN.md）。
|
||||
// 资源(Agent/KB…)挂在 Space 上而非 user 上，让"个人私有 / 项目临时组队 / 整租户共享"三种场景
|
||||
// 出自同一模型。Space 属单一租户（隔离/计费边界仍在租户）。
|
||||
// 与 Tenant/TenantMember 一样是"基础设施表"，不标 isTenantScoped（tenant 过滤在查询里显式做）。
|
||||
type Space struct {
|
||||
BaseModel
|
||||
TenantID string `gorm:"size:64;index"`
|
||||
Name string `gorm:"size:128"`
|
||||
Kind string `gorm:"size:16;index"` // personal / project / tenant
|
||||
Creator string `gorm:"size:64;index"` // 建者 user.id
|
||||
Archived bool // 软归档（临时空间完事即归档，从切换器隐藏）
|
||||
}
|
||||
|
||||
func (Space) TableName() string { return "sundynix_space" }
|
||||
|
||||
// SpaceMember 是空间成员关系（谁在这个空间 + 空间内角色）。复用租户角色语义（RoleRank 阶梯）。
|
||||
type SpaceMember struct {
|
||||
BaseModel
|
||||
SpaceID string `gorm:"size:64;uniqueIndex:idx_space_user;index"`
|
||||
UserID string `gorm:"size:64;uniqueIndex:idx_space_user;index"`
|
||||
Role string `gorm:"size:16"`
|
||||
Status string `gorm:"size:16"` // active / removed
|
||||
}
|
||||
|
||||
func (SpaceMember) TableName() string { return "sundynix_space_member" }
|
||||
|
||||
// 空间类型常量。
|
||||
const (
|
||||
SpaceKindPersonal = "personal"
|
||||
SpaceKindProject = "project"
|
||||
SpaceKindTenant = "tenant"
|
||||
)
|
||||
|
||||
// ---- 基础查询 ----
|
||||
|
||||
// GetSpace 按 id 取空间（不存在返回 nil）。
|
||||
func (p *Postgres) GetSpace(ctx context.Context, id string) (*Space, error) {
|
||||
if p.db == nil {
|
||||
return nil, errStoreDisabled
|
||||
}
|
||||
var s Space
|
||||
if err := p.db.WithContext(ctx).First(&s, "id = ?", id).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &s, nil
|
||||
}
|
||||
|
||||
// SpaceMemberRole 返回用户在某空间的角色（无 active 成员关系返回空）。
|
||||
func (p *Postgres) SpaceMemberRole(ctx context.Context, spaceID, userID string) string {
|
||||
if p.db == nil {
|
||||
return ""
|
||||
}
|
||||
var m SpaceMember
|
||||
if err := p.db.WithContext(ctx).
|
||||
Where("space_id = ? AND user_id = ? AND status = ?", spaceID, userID, "active").
|
||||
First(&m).Error; err != nil {
|
||||
return ""
|
||||
}
|
||||
return m.Role
|
||||
}
|
||||
|
||||
// addSpaceMember 幂等加成员（命中唯一 (space_id,user_id) 则改角色并置 active，复活已移除者）。
|
||||
func (p *Postgres) addSpaceMember(ctx context.Context, spaceID, userID, role string) error {
|
||||
return p.db.WithContext(ctx).Clauses(clause.OnConflict{
|
||||
Columns: []clause.Column{{Name: "space_id"}, {Name: "user_id"}},
|
||||
DoUpdates: clause.Assignments(map[string]any{"role": role, "status": "active", "updated_at": time.Now()}),
|
||||
}).Create(&SpaceMember{SpaceID: spaceID, UserID: userID, Role: role, Status: "active"}).Error
|
||||
}
|
||||
|
||||
// ---- 个人空间（迁移地基）----
|
||||
|
||||
// personalSpaceID 返回用户在某租户的个人空间 id（无则空）。
|
||||
func (p *Postgres) personalSpaceID(ctx context.Context, tenantID, userID string) string {
|
||||
var s Space
|
||||
if err := p.db.WithContext(ctx).Select("id").
|
||||
Where("tenant_id = ? AND creator = ? AND kind = ?", tenantID, userID, SpaceKindPersonal).
|
||||
Order("created_at asc").First(&s).Error; err != nil {
|
||||
return ""
|
||||
}
|
||||
return s.ID
|
||||
}
|
||||
|
||||
// EnsurePersonalSpace 幂等保证用户在某租户有个人空间（无则建 + 建者为 owner 成员）。
|
||||
func (p *Postgres) EnsurePersonalSpace(ctx context.Context, tenantID, userID string) (*Space, error) {
|
||||
if p.db == nil {
|
||||
return nil, errStoreDisabled
|
||||
}
|
||||
if id := p.personalSpaceID(ctx, tenantID, userID); id != "" {
|
||||
return p.GetSpace(ctx, id)
|
||||
}
|
||||
s := &Space{TenantID: tenantID, Name: "个人空间", Kind: SpaceKindPersonal, Creator: userID}
|
||||
if err := p.db.WithContext(ctx).Create(s).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := p.addSpaceMember(ctx, s.ID, userID, RoleOwner); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
|
||||
// ---- 活跃空间解析 / 切换 ----
|
||||
|
||||
// ActiveSpaceForUser 解析用户在某租户的活跃空间：user.active_space_id 若有效（存在、属该租户、且用户是
|
||||
// active 成员）→ 用之；否则回退该租户的个人空间（并保证其存在）。SpaceContext 中间件每请求调用。
|
||||
func (p *Postgres) ActiveSpaceForUser(ctx context.Context, tenantID, userID string) (*Space, error) {
|
||||
def, err := p.EnsurePersonalSpace(ctx, tenantID, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var u User
|
||||
if e := p.db.WithContext(ctx).Select("active_space_id").First(&u, "id = ?", userID).Error; e == nil {
|
||||
if u.ActiveSpaceID != "" && u.ActiveSpaceID != def.ID {
|
||||
if s, _ := p.GetSpace(ctx, u.ActiveSpaceID); s != nil && s.TenantID == tenantID && !s.Archived &&
|
||||
p.SpaceMemberRole(ctx, s.ID, userID) != "" {
|
||||
return s, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
return def, nil
|
||||
}
|
||||
|
||||
// SetActiveSpace 切换用户活跃空间（须为其 active 成员）。
|
||||
func (p *Postgres) SetActiveSpace(ctx context.Context, userID, spaceID string) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
if p.SpaceMemberRole(ctx, spaceID, userID) == "" {
|
||||
return errors.New("你不是该空间的成员")
|
||||
}
|
||||
return p.db.WithContext(ctx).Model(&User{}).Where("id = ?", userID).Update("active_space_id", spaceID).Error
|
||||
}
|
||||
|
||||
// ---- 空间列表 / 创建 ----
|
||||
|
||||
// SpaceInfo 是空间列表一行（含当前用户角色 + 成员数）。
|
||||
type SpaceInfo struct {
|
||||
ID string `json:"id"`
|
||||
TenantID string `json:"tenant_id"`
|
||||
Name string `json:"name"`
|
||||
Kind string `json:"kind"`
|
||||
Creator string `json:"creator"`
|
||||
Archived bool `json:"archived"`
|
||||
Role string `json:"role"` // 当前用户在此空间的角色
|
||||
Members int64 `json:"members"` // active 成员数
|
||||
}
|
||||
|
||||
// MySpaces 返回某用户在某租户内所属（active 成员）的空间（供切换器；含角色 + 成员数）。
|
||||
func (p *Postgres) MySpaces(ctx context.Context, tenantID, userID string) ([]SpaceInfo, error) {
|
||||
if p.db == nil {
|
||||
return nil, nil
|
||||
}
|
||||
var out []SpaceInfo
|
||||
err := p.db.WithContext(ctx).Table("sundynix_space s").
|
||||
Select("s.id, s.tenant_id, s.name, s.kind, s.creator, s.archived, m.role, "+
|
||||
"(SELECT count(*) FROM sundynix_space_member m2 WHERE m2.space_id = s.id AND m2.status = 'active') as members").
|
||||
Joins("JOIN sundynix_space_member m ON m.space_id = s.id").
|
||||
Where("m.user_id = ? AND m.status = 'active' AND s.tenant_id = ? AND s.deleted_at IS NULL", userID, tenantID).
|
||||
Order("s.kind = 'personal' desc, s.created_at asc").Scan(&out).Error
|
||||
return out, err
|
||||
}
|
||||
|
||||
// CreateSpace 在某租户建一个空间（建者自动成为 owner 成员）。kind 空则 project。
|
||||
func (p *Postgres) CreateSpace(ctx context.Context, tenantID, name, kind, creator string) (*Space, error) {
|
||||
if p.db == nil {
|
||||
return nil, errStoreDisabled
|
||||
}
|
||||
if kind == "" {
|
||||
kind = SpaceKindProject
|
||||
}
|
||||
if strings.TrimSpace(name) == "" {
|
||||
return nil, errors.New("空间名必填")
|
||||
}
|
||||
s := &Space{TenantID: tenantID, Name: strings.TrimSpace(name), Kind: kind, Creator: creator}
|
||||
if err := p.db.WithContext(ctx).Create(s).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := p.addSpaceMember(ctx, s.ID, creator, RoleOwner); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s, nil
|
||||
}
|
||||
|
||||
// ArchiveSpace 归档空间（软；个人空间不可归档）。
|
||||
func (p *Postgres) ArchiveSpace(ctx context.Context, spaceID string) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
s, err := p.GetSpace(ctx, spaceID)
|
||||
if err != nil || s == nil {
|
||||
return errors.New("空间不存在")
|
||||
}
|
||||
if s.Kind == SpaceKindPersonal {
|
||||
return errors.New("个人空间不可归档")
|
||||
}
|
||||
return p.db.WithContext(ctx).Model(&Space{}).Where("id = ?", spaceID).Update("archived", true).Error
|
||||
}
|
||||
|
||||
// ---- 空间成员 CRUD（照 TenantMember 套路）----
|
||||
|
||||
// ListSpaceMembers 返回某空间成员（含邮箱/名字/角色），排除已移除。
|
||||
func (p *Postgres) ListSpaceMembers(ctx context.Context, spaceID string) ([]MemberInfo, error) {
|
||||
if p.db == nil {
|
||||
return nil, nil
|
||||
}
|
||||
var out []MemberInfo
|
||||
err := p.db.WithContext(ctx).Table("sundynix_space_member m").
|
||||
Select("m.user_id, u.email, u.name, m.role, m.status, m.created_at as joined_at").
|
||||
Joins("JOIN sundynix_user u ON u.id = m.user_id").
|
||||
Where("m.space_id = ? AND m.status <> ?", spaceID, "removed").
|
||||
Order("m.created_at asc").Scan(&out).Error
|
||||
return out, err
|
||||
}
|
||||
|
||||
// AddSpaceMemberByEmail 按邮箱把已注册用户加入空间（幂等；须对方先注册）。
|
||||
// 调用方应先确保该用户是租户成员（空间成员应是租户成员的子集）。
|
||||
func (p *Postgres) AddSpaceMemberByEmail(ctx context.Context, spaceID, email, role string) (*MemberInfo, error) {
|
||||
if p.db == nil {
|
||||
return nil, errStoreDisabled
|
||||
}
|
||||
if !ValidRole(role) {
|
||||
return nil, errors.New("非法角色")
|
||||
}
|
||||
u, err := p.GetUserByEmail(ctx, strings.ToLower(strings.TrimSpace(email)))
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) || u == nil {
|
||||
return nil, errors.New("该邮箱尚未注册，请对方先注册后再加入")
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
s, err := p.GetSpace(ctx, spaceID)
|
||||
if err != nil || s == nil {
|
||||
return nil, errors.New("空间不存在")
|
||||
}
|
||||
if p.MemberRole(ctx, s.TenantID, u.ID) == "" {
|
||||
return nil, errors.New("对方不是本租户成员，请先加入租户")
|
||||
}
|
||||
if err := p.addSpaceMember(ctx, spaceID, u.ID, role); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &MemberInfo{UserID: u.ID, Email: u.Email, Name: u.Name, Role: role, Status: "active"}, nil
|
||||
}
|
||||
|
||||
// SetSpaceMemberRole 改空间成员角色（不能改/授 owner）。
|
||||
func (p *Postgres) SetSpaceMemberRole(ctx context.Context, spaceID, userID, role string) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
if !ValidRole(role) {
|
||||
return errors.New("非法角色")
|
||||
}
|
||||
if p.SpaceMemberRole(ctx, spaceID, userID) == RoleOwner {
|
||||
return errors.New("不能改 owner 的角色")
|
||||
}
|
||||
if role == RoleOwner {
|
||||
return errors.New("不能直接授予 owner")
|
||||
}
|
||||
return p.db.WithContext(ctx).Model(&SpaceMember{}).
|
||||
Where("space_id = ? AND user_id = ?", spaceID, userID).Update("role", role).Error
|
||||
}
|
||||
|
||||
// RemoveSpaceMember 软移除空间成员（不能移除 owner）。
|
||||
func (p *Postgres) RemoveSpaceMember(ctx context.Context, spaceID, userID string) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
if p.SpaceMemberRole(ctx, spaceID, userID) == RoleOwner {
|
||||
return errors.New("不能移除 owner")
|
||||
}
|
||||
return p.db.WithContext(ctx).Model(&SpaceMember{}).
|
||||
Where("space_id = ? AND user_id = ?", spaceID, userID).Update("status", "removed").Error
|
||||
}
|
||||
|
||||
// ---- 迁移回填（启动时幂等调用）----
|
||||
|
||||
// BackfillPersonalSpaces 给每个 active 租户成员补建个人空间（幂等）。返回补建数。
|
||||
func (p *Postgres) BackfillPersonalSpaces(ctx context.Context) (int, error) {
|
||||
if p.db == nil {
|
||||
return 0, errStoreDisabled
|
||||
}
|
||||
var members []TenantMember
|
||||
if err := p.db.WithContext(ctx).Where("status = ?", "active").Find(&members).Error; err != nil {
|
||||
return 0, err
|
||||
}
|
||||
n := 0
|
||||
for _, m := range members {
|
||||
if p.personalSpaceID(ctx, m.TenantID, m.UserID) != "" {
|
||||
continue
|
||||
}
|
||||
if _, err := p.EnsurePersonalSpace(ctx, m.TenantID, m.UserID); err != nil {
|
||||
return n, err
|
||||
}
|
||||
n++
|
||||
}
|
||||
return n, nil
|
||||
}
|
||||
|
||||
// MigrateAgentSpaces 把存量 Agent 的 space_id 回填为其 (tenant_id, owner) 个人空间，
|
||||
// 然后 DROP 旧唯一索引 idx_agent_on(owner,name) 并建新唯一索引 idx_agent_sn(space_id,name)。
|
||||
// 顺序关键：先回填再建唯一索引，否则存量 space_id 空会撞车。全程幂等。
|
||||
func (p *Postgres) MigrateAgentSpaces(ctx context.Context) error {
|
||||
if p.db == nil {
|
||||
return errStoreDisabled
|
||||
}
|
||||
// 回填 space_id：按 agent.tenant_id + agent.owner 命中个人空间。
|
||||
sub := "SELECT id FROM sundynix_space s WHERE s.tenant_id = a.tenant_id AND s.creator = a.owner AND s.kind = 'personal' ORDER BY s.created_at ASC LIMIT 1"
|
||||
if err := p.db.WithContext(ctx).Exec(
|
||||
"UPDATE sundynix_agent a SET space_id = ("+sub+") WHERE (a.space_id IS NULL OR a.space_id = '') AND a.owner <> ''",
|
||||
).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
// 旧唯一索引 DROP（AutoMigrate 不删；留着会过约束"同人跨空间不能重名"）。
|
||||
if err := p.db.WithContext(ctx).Exec("DROP INDEX IF EXISTS idx_agent_on").Error; err != nil {
|
||||
return err
|
||||
}
|
||||
// 新唯一索引（回填后建，安全）。
|
||||
if err := p.db.WithContext(ctx).Exec(
|
||||
"CREATE UNIQUE INDEX IF NOT EXISTS idx_agent_sn ON sundynix_agent(space_id, name)",
|
||||
).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user