mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-10-15 02:16:56 +00:00
viettranx
c4d13ca4a7
- Convert importCron/importUserProfiles/importUserOverrides to multi-row UPSERT with chunking (200→2 queries per import) - Add GetTenantsByIDs batch method to TenantStore (PG+SQLite), refactor handleMine() to single-query tenant fetch (N→1) - Refactor buildAgentKeyMap() to use existing GetByKeys with tenant-scoped context instead of per-key SELECT (N→1) - Add GetDocumentsByIDs, GetDocumentByBasename, CreateLinks batch methods to VaultStore (PG+SQLite) - Batch-fetch docs in enrichment Phase 0, carry title through pipeline to avoid per-doc refetch in classify phase - Replace ListDocuments(limit=500) fallback in wikilink resolution with targeted GetDocumentByBasename DB query - Batch CreateLink calls in SyncDocLinks and classifyLinks - Refactor usage handleGet/handleSummary to use ListPagedRich instead of List+GetOrCreate N+1 loop (100→1 queries) - Batch INSERT for team import members/comments/events/links (250→5) - Add migration 47: UNIQUE constraint on cron_jobs(agent_id,tenant_id,name) with dedup, SQLite schema v15
85 lines
3.4 KiB
Go
85 lines
3.4 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// MasterTenantID is the fixed UUID v7 for the default/master tenant.
|
|
// All existing data defaults to this tenant during migration.
|
|
var MasterTenantID = uuid.MustParse("0193a5b0-7000-7000-8000-000000000001")
|
|
|
|
// Tenant status constants.
|
|
const (
|
|
TenantStatusActive = "active"
|
|
TenantStatusSuspended = "suspended"
|
|
TenantStatusArchived = "archived"
|
|
)
|
|
|
|
// Tenant role constants (hierarchy: owner > admin > operator > member > viewer).
|
|
const (
|
|
TenantRoleOwner = "owner"
|
|
TenantRoleAdmin = "admin"
|
|
TenantRoleOperator = "operator"
|
|
TenantRoleMember = "member"
|
|
TenantRoleViewer = "viewer"
|
|
)
|
|
|
|
// TenantData represents a tenant in the database.
|
|
type TenantData struct {
|
|
ID uuid.UUID `json:"id" db:"id"`
|
|
Name string `json:"name" db:"name"`
|
|
Slug string `json:"slug" db:"slug"`
|
|
Status string `json:"status" db:"status"`
|
|
Settings json.RawMessage `json:"settings,omitempty" db:"settings"`
|
|
CreatedAt time.Time `json:"created_at" db:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at" db:"updated_at"`
|
|
}
|
|
|
|
// TenantUserData represents a user's membership in a tenant.
|
|
type TenantUserData struct {
|
|
ID uuid.UUID `json:"id" db:"id"`
|
|
TenantID uuid.UUID `json:"tenant_id" db:"tenant_id"`
|
|
UserID string `json:"user_id" db:"user_id"`
|
|
DisplayName *string `json:"display_name,omitempty" db:"display_name"`
|
|
Role string `json:"role" db:"role"`
|
|
Metadata json.RawMessage `json:"metadata,omitempty" db:"metadata"`
|
|
CreatedAt time.Time `json:"created_at" db:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at" db:"updated_at"`
|
|
}
|
|
|
|
// TenantStore manages tenants and tenant-user membership.
|
|
type TenantStore interface {
|
|
// Tenant CRUD
|
|
CreateTenant(ctx context.Context, tenant *TenantData) error
|
|
GetTenant(ctx context.Context, id uuid.UUID) (*TenantData, error)
|
|
GetTenantBySlug(ctx context.Context, slug string) (*TenantData, error)
|
|
ListTenants(ctx context.Context) ([]TenantData, error)
|
|
UpdateTenant(ctx context.Context, id uuid.UUID, updates map[string]any) error
|
|
|
|
// Tenant-user membership
|
|
AddUser(ctx context.Context, tenantID uuid.UUID, userID, role string) error
|
|
RemoveUser(ctx context.Context, tenantID uuid.UUID, userID string) error
|
|
GetUserRole(ctx context.Context, tenantID uuid.UUID, userID string) (string, error)
|
|
ListUsers(ctx context.Context, tenantID uuid.UUID) ([]TenantUserData, error)
|
|
ListUserTenants(ctx context.Context, userID string) ([]TenantUserData, error)
|
|
|
|
// GetTenantsByIDs returns tenants matching the given UUIDs in a single query.
|
|
GetTenantsByIDs(ctx context.Context, ids []uuid.UUID) ([]TenantData, error)
|
|
|
|
// ResolveUserTenant returns the tenant_id for a user.
|
|
// If user belongs to multiple tenants, returns the first (by created_at).
|
|
// If no membership, returns MasterTenantID (backward compat).
|
|
ResolveUserTenant(ctx context.Context, userID string) (uuid.UUID, error)
|
|
|
|
// GetTenantUser returns a single tenant_user by primary key.
|
|
GetTenantUser(ctx context.Context, id uuid.UUID) (*TenantUserData, error)
|
|
|
|
// CreateTenantUserReturning creates a tenant_user and returns the row.
|
|
// On conflict (tenant_id, user_id), updates role/display_name and returns existing row.
|
|
CreateTenantUserReturning(ctx context.Context, tenantID uuid.UUID, userID, displayName, role string) (*TenantUserData, error)
|
|
}
|