Blizzard
3a175e46f3
回应三点诉求：一次入一批文件、按文件夹/项目/案件组织、且只有我能查我的库。 隔离（核心）：知识库实际分区键 = "owner/name"，owner 由网关从 X-User-ID 注入， 客户端只发库名、发不了 owner —— 故任何人都只能查到自己 owner 前缀下的数据。 - gateway: scopedKB(owner/kb) 注入 ingest/search/graph；ingest/search/graph 全部带身份头。 - store: sundynix_kb 注册表（owner+name 唯一 + kind），ListKB/EnsureKB(OnConflict DoNothing)。 项目/案件组织： - gateway: GET /kb/list（owner 隔离列表）、POST /kb/create（folder/project/case/general）； 入库时 EnsureKB 自动登记。 - 前端: KbView 顶部知识库下拉 + 新建（项目/案件/文件夹/通用），检索/图谱/入库都绑定所选库。 批量文件： - 前端: 选择文件(multiple) + 选择文件夹(webkitdirectory) + 拖拽一批 → 每文件一个 job， 文件列表实时显示各自状态（排队/解析/向量化/写入/抽取/完成/失败）+ 完成/失败计数。 验证：curl 证隔离 —— wt 入 default→可检索；alice 查同名 default→[] 空；alice 列表不含 wt 案件库。 Preview 证 UI —— 知识库下拉含 案件-2024-001(案件)+default(通用)、owner 隔离徽标、批量/文件夹按钮。 tsc+vite+gateway build 通过；重建 .app 重启窗口。 注：身份目前来自 X-User-ID 头（可信前端），生产应换 JWT 鉴权中间件——隔离机制(owner 前缀)已就位。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
// Package store 封装 MainDB(PgSQL) 与 CacheDB(Redis) 的访问。
|
|
package store
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"log"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
// errStoreDisabled 表示 Postgres 处于降级（未连接）模式，写操作无法进行。
|
|
var errStoreDisabled = errors.New("postgres store disabled")
|
|
|
|
// Postgres 持有 MainDB 连接（Users / Billing / DSL）。
|
|
// db 为 nil 表示降级模式（连接失败时仍允许网关启动）。
|
|
type Postgres struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// OpenPostgres 用 GORM 连接 MainDB 并自动迁移表结构。
|
|
// 表名统一 sundynix_ 前缀 + 单数（User→sundynix_user, Task→sundynix_task）。
|
|
// 连接失败不 fatal：返回降级实例，网关仍可启动（无 Docker 跑 demo 时即此路径）。
|
|
func OpenPostgres(dsn string) *Postgres {
|
|
db, err := gorm.Open(postgres.New(postgres.Config{DSN: dsn}), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
TablePrefix: "sundynix_", // 所有表加前缀
|
|
SingularTable: true, // 单数表名
|
|
},
|
|
})
|
|
if err != nil {
|
|
log.Printf("[store] postgres 不可用，降级运行（不持久化）: %v", err)
|
|
return &Postgres{}
|
|
}
|
|
if err := db.AutoMigrate(&User{}, &Task{}, &LLMModel{}, &KB{}); err != nil {
|
|
log.Printf("[store] postgres AutoMigrate 失败，降级运行: %v", err)
|
|
return &Postgres{}
|
|
}
|
|
// 回填：kind 列新增前的旧模型行默认归为 chat（幂等）。
|
|
db.Model(&LLMModel{}).Where("kind = '' OR kind IS NULL").Update("kind", "chat")
|
|
log.Println("[store] postgres connected, migrated sundynix_user / sundynix_task")
|
|
return &Postgres{db: db}
|
|
}
|
|
|
|
// Enabled 报告是否处于真实持久化模式。
|
|
func (p *Postgres) Enabled() bool { return p.db != nil }
|
|
|
|
// SaveTask 持久化一次任务提交（best-effort：降级模式下静默跳过）。
|
|
func (p *Postgres) SaveTask(ctx context.Context, id, graph string) error {
|
|
if p.db == nil {
|
|
return nil
|
|
}
|
|
return p.db.WithContext(ctx).Create(&Task{ID: id, Graph: graph, Status: "submitted"}).Error
|
|
}
|
|
|
|
// CountTasks 返回已提交任务数（降级模式返回 0）。
|
|
func (p *Postgres) CountTasks(ctx context.Context) (int64, error) {
|
|
if p.db == nil {
|
|
return 0, nil
|
|
}
|
|
var n int64
|
|
err := p.db.WithContext(ctx).Model(&Task{}).Count(&n).Error
|
|
return n, err
|
|
}
|
|
|
|
// Close 释放底层连接。
|
|
func (p *Postgres) Close() {
|
|
if p.db == nil {
|
|
return
|
|
}
|
|
if sqlDB, err := p.db.DB(); err == nil {
|
|
_ = sqlDB.Close()
|
|
}
|
|
}
|