refactor(kb): 文件主表 + 文档关联改用雪花ID(弃用按名关联)

把 sundynix_doc 明确为"文件主表",补齐文件基础信息字段;
文档间 [[双链]] 改为以 Doc.ID 关联,查询/渲染一律按文件 ID。

store:
- Doc 增 Ext(后缀)/MD5(内容指纹) 字段;ObjectKey 即"存放链接"
- DocLink 由 (FromName,ToName) 改为 (FromID,ToID,ToName)
  · FromID/ToID 关联 Doc.ID;ToName 保留用于悬空链接展示与回填
- SaveDoc 返回新建/更新文件的雪花 ID(供建链)
- 新增 GetDocByID(按 ID + owner 取正文,防越权)
- ReplaceDocLinks 以 fromID 重建出链,按 [[名称]] 解析目标 ID
- 新增 ResolveInboundLinks:目标入库后回填指向它的悬空链接
- ListLinks 只返回已解析(to_id 非空)的 ID→ID 边
- migrateDocLinkToID:旧按名双链表无 from_id 列则重建为按 ID 关联

gateway/handler:
- runIngest 计算 ext/md5,SaveDoc 取回 ID 后建链 + 回填悬空
- KbDoc 改为 GET ?id=(按文件 ID 取全文)
- KbVault 返回 id+ext;KbLinks 返回 from/to 为 ID

desktop:
- VaultDoc 增 id/ext;getDoc(docId) 按 ID 取正文
- VaultPanel 选中态/正文/反链/关系图改用 ID,名↔ID 双向映射
  渲染;保存笔记后按名定位回其新 ID

验证(gateway+PG+MinIO 实测):vault 带 id+ext;双链 ID→ID 且
A→B 悬空链接在 B 入库后成功回填;按 ID 取大文档(15006字)从
MinIO 完整取回;跨 owner 按 ID 取文档 404(隔离生效)。桌面端
文库 Tab 按 ID 选中/载入/反链渲染正常,无控制台报错。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Blizzard
2026-06-15 09:38:02 +08:00
parent f610d8d2da
commit 5d76652bff
5 changed files with 142 additions and 70 deletions
+7 -5
View File
@@ -122,13 +122,15 @@ export async function createKb(id: Identity, name: string, kind: string): Promis
} }
export interface VaultDoc { export interface VaultDoc {
id: string; // 文件主表雪花 ID —— 选中/取正文/关联一律用它
name: string; name: string;
ext?: string; // 文件后缀(.md/.pdf…;笔记为空)
size?: number; size?: number;
preview?: string; preview?: string;
} }
export interface DocLink { export interface DocLink {
from: string; from: string; // 源文件 ID
to: string; to: string; // 目标文件 ID
} }
// ---- 我的 Agent 编排(服务端保存,owner 隔离)---- // ---- 我的 Agent 编排(服务端保存,owner 隔离)----
@@ -182,9 +184,9 @@ export async function listVault(id: Identity, kb: string): Promise<VaultDoc[]> {
return data.docs ?? []; return data.docs ?? [];
} }
// getDoc: GET /api/v1/kb/doc —— 取单篇文档全文(按需加载,避免列表拉全量)。 // getDoc: GET /api/v1/kb/doc?id= —— 按文件 ID 取单篇全文(按需加载,避免列表拉全量)。
export async function getDoc(id: Identity, kb: string, name: string): Promise<string> { export async function getDoc(id: Identity, docId: string): Promise<string> {
const res = await fetch(`${GATEWAY}/api/v1/kb/doc?kb=${encodeURIComponent(kb)}&name=${encodeURIComponent(name)}`, { headers: idHeaders(id) }); const res = await fetch(`${GATEWAY}/api/v1/kb/doc?id=${encodeURIComponent(docId)}`, { headers: idHeaders(id) });
const data = (await res.json()) as { content?: string; error?: string }; const data = (await res.json()) as { content?: string; error?: string };
if (!res.ok) throw new Error(data.error ?? `doc failed: ${res.status}`); if (!res.ok) throw new Error(data.error ?? `doc failed: ${res.status}`);
return data.content ?? ""; return data.content ?? "";
+38 -18
View File
@@ -451,13 +451,21 @@ function VaultPanel({ identity, kb }: { identity: Identity; kb: string }) {
const [draftName, setDraftName] = useState(""); const [draftName, setDraftName] = useState("");
const [saving, setSaving] = useState(false); const [saving, setSaving] = useState(false);
const load = useCallback(async () => { // sel 为文件主表雪花 ID(不再是文件名)—— 取正文/反链/关系图一律用 ID 关联。
// preferName 给定时(如保存笔记后)按名定位到其新 ID。
const load = useCallback(async (preferName?: string) => {
setLoading(true); setLoading(true);
try { try {
const [d, lk] = await Promise.all([listVault(identity, kb), listLinks(identity, kb)]); const [d, lk] = await Promise.all([listVault(identity, kb), listLinks(identity, kb)]);
setDocs(d); setDocs(d);
setLinks(lk); setLinks(lk);
setSel((s) => (d.some((x) => x.name === s) ? s : d[0]?.name ?? null)); setSel((s) => {
if (preferName) {
const hit = d.find((x) => x.name === preferName);
if (hit) return hit.id;
}
return d.some((x) => x.id === s) ? s : d[0]?.id ?? null;
});
} catch { } catch {
setDocs([]); setDocs([]);
setLinks([]); setLinks([]);
@@ -471,32 +479,40 @@ function VaultPanel({ identity, kb }: { identity: Identity; kb: string }) {
setMode("read"); setMode("read");
}, [load]); }, [load]);
// 选中文档时按取全文(不随列表一起拉)。 // 选中文档时按 ID 取全文(不随列表一起拉)。
useEffect(() => { useEffect(() => {
if (!sel || editing) return; if (!sel || editing) return;
let alive = true; let alive = true;
setLoadingDoc(true); setLoadingDoc(true);
getDoc(identity, kb, sel) getDoc(identity, sel)
.then((c) => alive && setContent(c)) .then((c) => alive && setContent(c))
.catch(() => alive && setContent("")) .catch(() => alive && setContent(""))
.finally(() => alive && setLoadingDoc(false)); .finally(() => alive && setLoadingDoc(false));
return () => { return () => {
alive = false; alive = false;
}; };
}, [sel, kb, identity, editing]); }, [sel, identity, editing]);
const names = new Set(docs.map((d) => d.name)); // 名称↔ID 映射:[[双链]]按名书写、关系图按名展示,但选中/关联走 ID。
const current = docs.find((d) => d.name === sel); const idToName = new Map(docs.map((d) => [d.id, d.name] as const));
const nameToId = new Map(docs.map((d) => [d.name, d.id] as const));
const current = docs.find((d) => d.id === sel);
const open = (name: string) => { const open = (name: string) => {
if (names.has(name)) { const id = nameToId.get(name);
setSel(name); if (id) {
setSel(id);
setMode("read"); setMode("read");
setEditing(false); setEditing(false);
} }
}; };
// 反链/笔记关系图来自服务端 [[双链]]索引,不扫全文。 // 反链/笔记关系图来自服务端 [[双链]]索引FromID→ToID),ID 关联、按名渲染,不扫全文。
const backlinks = sel ? [...new Set(links.filter((l) => l.to === sel).map((l) => l.from))] : []; const backlinks = sel
const noteTriples: Triple[] = links.filter((l) => names.has(l.from) && names.has(l.to)).map((l) => ({ s: l.from, p: "链接", o: l.to })); ? [...new Set(links.filter((l) => l.to === sel).map((l) => idToName.get(l.from)).filter((n): n is string => !!n))]
: [];
const noteTriples: Triple[] = links
.map((l) => ({ s: idToName.get(l.from), o: idToName.get(l.to) }))
.filter((e): e is { s: string; o: string } => !!e.s && !!e.o)
.map((e) => ({ s: e.s, p: "链接", o: e.o }));
const startNew = () => { const startNew = () => {
setCreatingNew(true); setCreatingNew(true);
@@ -524,8 +540,7 @@ function VaultPanel({ identity, kb }: { identity: Identity; kb: string }) {
setEditing(false); setEditing(false);
setCreatingNew(false); setCreatingNew(false);
setContent(draft); setContent(draft);
setSel(name); setTimeout(() => void load(name), 400);
setTimeout(() => void load(), 400);
} catch (e) { } catch (e) {
toast.push("error", (e as Error).message); toast.push("error", (e as Error).message);
} finally { } finally {
@@ -543,19 +558,24 @@ function VaultPanel({ identity, kb }: { identity: Identity; kb: string }) {
<button onClick={startNew} className="ml-auto text-slate-500 hover:text-brand-400" title="新建笔记"> <button onClick={startNew} className="ml-auto text-slate-500 hover:text-brand-400" title="新建笔记">
<Plus className="h-3.5 w-3.5" /> <Plus className="h-3.5 w-3.5" />
</button> </button>
<button onClick={load} className="text-slate-600 hover:text-slate-300" title="刷新"> <button onClick={() => void load()} className="text-slate-600 hover:text-slate-300" title="刷新">
<RefreshCw className={cn("h-3.5 w-3.5", loading && "animate-spin")} /> <RefreshCw className={cn("h-3.5 w-3.5", loading && "animate-spin")} />
</button> </button>
</div> </div>
<ul className="space-y-0.5"> <ul className="space-y-0.5">
{docs.map((d) => ( {docs.map((d) => (
<li key={d.name}> <li key={d.id}>
<button <button
onClick={() => open(d.name)} onClick={() => {
className={cn("flex w-full items-center gap-1.5 rounded px-2 py-1.5 text-left text-xs", d.name === sel ? "bg-brand/15 text-brand-400" : "text-slate-300 hover:bg-ink-800")} setSel(d.id);
setMode("read");
setEditing(false);
}}
className={cn("flex w-full items-center gap-1.5 rounded px-2 py-1.5 text-left text-xs", d.id === sel ? "bg-brand/15 text-brand-400" : "text-slate-300 hover:bg-ink-800")}
> >
<FileText className="h-3.5 w-3.5 shrink-0" /> <FileText className="h-3.5 w-3.5 shrink-0" />
<span className="truncate" title={d.name}>{d.name}</span> <span className="truncate" title={d.name}>{d.name}</span>
{d.ext && <span className="ml-auto shrink-0 text-[9px] uppercase text-slate-600">{d.ext.replace(/^\./, "")}</span>}
</button> </button>
</li> </li>
))} ))}
+17 -8
View File
@@ -2,6 +2,7 @@ package handler
import ( import (
"context" "context"
"crypto/md5"
"crypto/rand" "crypto/rand"
"encoding/base64" "encoding/base64"
"encoding/hex" "encoding/hex"
@@ -131,14 +132,14 @@ func (h *Handler) KbVault(c *gin.Context) {
} }
docs := make([]gin.H, 0, len(rows)) docs := make([]gin.H, 0, len(rows))
for _, r := range rows { for _, r := range rows {
docs = append(docs, gin.H{"name": r.Name, "size": r.Size, "preview": r.Preview}) docs = append(docs, gin.H{"id": r.ID, "name": r.Name, "ext": r.Ext, "size": r.Size, "preview": r.Preview})
} }
c.JSON(http.StatusOK, gin.H{"docs": docs}) c.JSON(http.StatusOK, gin.H{"docs": docs})
} }
// KbDoc: GET /api/v1/kb/doc?kb=&name= —— 取单篇文档全文(按需加载,不在列表里拉全量)。 // KbDoc: GET /api/v1/kb/doc?id= —— 按文件 ID 取单篇全文(按需加载,不在列表里拉全量)。
func (h *Handler) KbDoc(c *gin.Context) { func (h *Handler) KbDoc(c *gin.Context) {
d, err := h.db.GetDoc(c.Request.Context(), userID(c), rawKB(c.Query("kb")), c.Query("name")) d, err := h.db.GetDocByID(c.Request.Context(), userID(c), c.Query("id"))
if err != nil || d == nil { if err != nil || d == nil {
c.JSON(http.StatusNotFound, gin.H{"error": "文档不存在"}) c.JSON(http.StatusNotFound, gin.H{"error": "文档不存在"})
return return
@@ -149,10 +150,10 @@ func (h *Handler) KbDoc(c *gin.Context) {
content = obj content = obj
} }
} }
c.JSON(http.StatusOK, gin.H{"name": d.Name, "content": content, "size": d.Size}) c.JSON(http.StatusOK, gin.H{"id": d.ID, "name": d.Name, "ext": d.Ext, "content": content, "size": d.Size})
} }
// KbLinks: GET /api/v1/kb/links?kb= —— 某库全部 [[双链]]from→to),供反链/笔记关系图。 // KbLinks: GET /api/v1/kb/links?kb= —— 某库已解析的 [[双链]]FromID→ToID),供反链/笔记关系图按 ID 渲染
func (h *Handler) KbLinks(c *gin.Context) { func (h *Handler) KbLinks(c *gin.Context) {
rows, err := h.db.ListLinks(c.Request.Context(), userID(c), rawKB(c.Query("kb"))) rows, err := h.db.ListLinks(c.Request.Context(), userID(c), rawKB(c.Query("kb")))
if err != nil { if err != nil {
@@ -161,7 +162,7 @@ func (h *Handler) KbLinks(c *gin.Context) {
} }
links := make([]gin.H, 0, len(rows)) links := make([]gin.H, 0, len(rows))
for _, l := range rows { for _, l := range rows {
links = append(links, gin.H{"from": l.FromName, "to": l.ToName}) links = append(links, gin.H{"from": l.FromID, "to": l.ToID})
} }
c.JSON(http.StatusOK, gin.H{"links": links}) c.JSON(http.StatusOK, gin.H{"links": links})
} }
@@ -243,6 +244,9 @@ func (h *Handler) runIngest(job, owner, kbName, scoped, forceDoc, filename strin
} }
if text != "" { if text != "" {
size := len([]rune(text)) size := len([]rune(text))
ext := strings.ToLower(filepath.Ext(filename)) // 笔记/文本入库时 filename 为空 → ext 为空
sum := md5.Sum([]byte(text))
md5hex := hex.EncodeToString(sum[:])
inline, objectKey := text, "" inline, objectKey := text, ""
// 大文档正文落对象存储,PG 只留元数据+预览+对象键(避免把十几万字塞进 PG)。 // 大文档正文落对象存储,PG 只留元数据+预览+对象键(避免把十几万字塞进 PG)。
if size > docInlineMax && h.blob.Ready() { if size > docInlineMax && h.blob.Ready() {
@@ -253,8 +257,13 @@ func (h *Handler) runIngest(job, owner, kbName, scoped, forceDoc, filename strin
log.Printf("[gateway] 大文档转 MinIO 失败,回退内联: %v", err) log.Printf("[gateway] 大文档转 MinIO 失败,回退内联: %v", err)
} }
} }
_ = h.db.SaveDoc(ctx, owner, kbName, docName, inline, objectKey, size, head(text, 500)) docID, err := h.db.SaveDoc(ctx, owner, kbName, docName, ext, md5hex, inline, objectKey, size, head(text, 500))
_ = h.db.ReplaceDocLinks(ctx, owner, kbName, docName, wikiLinks(text)) // 维护 [[双链]] 索引 if err != nil {
log.Printf("[gateway] 文件入库失败: %v", err)
} else if docID != "" {
_ = h.db.ReplaceDocLinks(ctx, owner, kbName, docID, wikiLinks(text)) // 以本文件 ID 维护出链
_ = h.db.ResolveInboundLinks(ctx, owner, kbName, docName, docID) // 回填指向本文件的悬空链接
}
} }
// 调 mcp-go kb_ingest(带 job_id):它会发 切块/向量化/写入/完成 事件 + CompleteStream。 // 调 mcp-go kb_ingest(带 job_id):它会发 切块/向量化/写入/完成 事件 + CompleteStream。
+66 -39
View File
@@ -2,6 +2,7 @@ package store
import ( import (
"context" "context"
"errors"
"time" "time"
"gorm.io/gorm" "gorm.io/gorm"
@@ -84,52 +85,59 @@ func (p *Postgres) DeleteAgent(ctx context.Context, owner, name string) error {
return p.db.WithContext(ctx).Where("owner = ? AND name = ?", owner, name).Delete(&Agent{}).Error return p.db.WithContext(ctx).Where("owner = ? AND name = ?", owner, name).Delete(&Agent{}).Error
} }
// Doc 是入库的一份原始文档/笔记(供 Obsidian 式"文库"浏览:列表 + Markdown 阅读 + 双链)。 // Doc 是入库的一份文件/笔记 —— 文件主表(供 Obsidian 式"文库"浏览:列表 + Markdown 阅读 + 双链)。
// 表名 sundynix_doc。(owner,kb,name) 唯一;按 owner 隔离。 // 表名 sundynix_doc。(owner,kb,name) 唯一;按 owner 隔离。文档间关联一律用本表的雪花 ID 关联。
type Doc struct { type Doc struct {
BaseModel BaseModel
Owner string `gorm:"size:64;uniqueIndex:idx_doc_okn"` Owner string `gorm:"size:64;uniqueIndex:idx_doc_okn"`
KB string `gorm:"size:64;uniqueIndex:idx_doc_okn"` KB string `gorm:"size:64;uniqueIndex:idx_doc_okn"`
Name string `gorm:"size:160;uniqueIndex:idx_doc_okn"` Name string `gorm:"size:160;uniqueIndex:idx_doc_okn"` // 文件名称(不含扩展名 / 笔记名)
Ext string `gorm:"size:16"` // 文件后缀(.md/.pdf/.docx…;笔记/文本为空)
MD5 string `gorm:"size:32;index"` // 正文内容指纹(去重 / 校验)
Size int // 原文字数(rune Size int // 原文字数(rune
Preview string `gorm:"size:600"` // 前若干字预览(列表/反链用,不拉全文) Preview string `gorm:"size:600"` // 前若干字预览(列表/反链用,不拉全文)
Content string `gorm:"type:text"` // 小文档内联正文;大文档转 MinIO 后置空 Content string `gorm:"type:text"` // 小文档内联正文;大文档转 MinIO 后置空
ObjectKey string `gorm:"size:160"` // 大文档在 MinIO 的对象键(空=内联 Content ObjectKey string `gorm:"size:160"` // 存放链接:大文档在 MinIO 的对象键(空=内联 Content
} }
func (Doc) TableName() string { return "sundynix_doc" } func (Doc) TableName() string { return "sundynix_doc" }
// DocLink 是文档间 [[双链]] 的索引(owner+kb 内 from→to),供反链/笔记关系图按 SQL 查询, // DocLink 是文档间 [[双链]] 的索引(owner+kb 内 from→to),以 Doc.ID 关联,供反链/笔记关系图按 SQL 查询,
// 避免在前端扫全部正文。入库/编辑时按 from 文档重建其出链。 // 避免在前端扫全部正文。入库/编辑时按 from 文档重建其出链;目标尚未入库时 ToID 为空(悬空,记 ToName 待回填)
type DocLink struct { type DocLink struct {
BaseModel BaseModel
Owner string `gorm:"size:64;index:idx_link_okf"` Owner string `gorm:"size:64;index:idx_link_of"`
KB string `gorm:"size:64;index:idx_link_okf"` KB string `gorm:"size:64;index:idx_link_of"`
FromName string `gorm:"size:160;index:idx_link_okf"` FromID string `gorm:"size:24;index:idx_link_of"` // 源文档 Doc.ID
ToName string `gorm:"size:160;index"` ToID string `gorm:"size:24;index"` // 目标文档 Doc.ID(空=悬空:目标尚未入库)
ToName string `gorm:"size:160"` // [[原始名]],供悬空链接展示 / 目标入库后回填 ToID
} }
func (DocLink) TableName() string { return "sundynix_doc_link" } func (DocLink) TableName() string { return "sundynix_doc_link" }
// docHead 取前 n 个 rune 作预览 // SaveDoc 写入/更新一份文件(owner+kb+name 唯一,重名覆盖),返回该文件的雪花 ID(供关联用)
func docHead(s string, n int) string { // content 为内联正文(大文档转 MinIO 时传空 + objectKey);ext/md5/preview/size 由调用方按全文给出。
r := []rune(s) func (p *Postgres) SaveDoc(ctx context.Context, owner, kb, name, ext, md5, content, objectKey string, size int, preview string) (string, error) {
if len(r) <= n {
return s
}
return string(r[:n])
}
// SaveDoc 写入/更新一份文档(owner+kb+name 唯一,重名覆盖)。
// content 为内联正文(大文档转 MinIO 时传空 + objectKey);preview/size 由调用方按全文给出。
func (p *Postgres) SaveDoc(ctx context.Context, owner, kb, name, content, objectKey string, size int, preview string) error {
if p.db == nil { if p.db == nil {
return nil return "", nil
} }
return p.db.WithContext(ctx).Clauses(clause.OnConflict{ var d Doc
Columns: []clause.Column{{Name: "owner"}, {Name: "kb"}, {Name: "name"}}, err := p.db.WithContext(ctx).Where("owner = ? AND kb = ? AND name = ?", owner, kb, name).First(&d).Error
DoUpdates: clause.Assignments(map[string]any{"content": content, "object_key": objectKey, "size": size, "preview": preview, "updated_at": time.Now()}), if errors.Is(err, gorm.ErrRecordNotFound) {
}).Create(&Doc{Owner: owner, KB: kb, Name: name, Content: content, ObjectKey: objectKey, Size: size, Preview: preview}).Error d = Doc{Owner: owner, KB: kb, Name: name, Ext: ext, MD5: md5, Content: content, ObjectKey: objectKey, Size: size, Preview: preview}
if err := p.db.WithContext(ctx).Create(&d).Error; err != nil {
return "", err
}
return d.ID, nil
}
if err != nil {
return "", err
}
d.Ext, d.MD5, d.Content, d.ObjectKey, d.Size, d.Preview = ext, md5, content, objectKey, size, preview
if err := p.db.WithContext(ctx).Save(&d).Error; err != nil {
return "", err
}
return d.ID, nil
} }
// ListVault 返回文库列表(仅元数据 + 预览,不含全文),避免一次拉回整库正文。 // ListVault 返回文库列表(仅元数据 + 预览,不含全文),避免一次拉回整库正文。
@@ -139,37 +147,46 @@ func (p *Postgres) ListVault(ctx context.Context, owner, kb string) ([]Doc, erro
} }
var rows []Doc var rows []Doc
err := p.db.WithContext(ctx). err := p.db.WithContext(ctx).
Select("id", "name", "size", "preview", "object_key", "updated_at"). Select("id", "name", "ext", "size", "preview", "object_key", "updated_at").
Where("owner = ? AND kb = ?", owner, kb).Order("updated_at desc").Find(&rows).Error Where("owner = ? AND kb = ?", owner, kb).Order("updated_at desc").Find(&rows).Error
return rows, err return rows, err
} }
// GetDoc 取单篇文档(含全文 Content 与 ObjectKey,供按需阅读) // GetDocByID 按文件 ID 取单篇文档(含全文 Content 与 ObjectKey),owner 作用域防越权
func (p *Postgres) GetDoc(ctx context.Context, owner, kb, name string) (*Doc, error) { func (p *Postgres) GetDocByID(ctx context.Context, owner, id string) (*Doc, error) {
if p.db == nil { if p.db == nil {
return nil, nil return nil, nil
} }
var d Doc var d Doc
if err := p.db.WithContext(ctx).Where("owner = ? AND kb = ? AND name = ?", owner, kb, name).First(&d).Error; err != nil { if err := p.db.WithContext(ctx).Where("owner = ? AND id = ?", owner, id).First(&d).Error; err != nil {
return nil, err return nil, err
} }
return &d, nil return &d, nil
} }
// ReplaceDocLinks 重建某文档的出链(先删旧,再插新)—— 入库/编辑时调用。 // ReplaceDocLinks 以源文件 ID 重建其出链(先删旧,再按 [[名称]] 解析目标 ID 后插新)—— 入库/编辑时调用。
func (p *Postgres) ReplaceDocLinks(ctx context.Context, owner, kb, from string, tos []string) error { // 目标文档尚未入库时 ToID 留空(悬空),待其入库时由 ResolveInboundLinks 回填。
func (p *Postgres) ReplaceDocLinks(ctx context.Context, owner, kb, fromID string, toNames []string) error {
if p.db == nil { if p.db == nil {
return nil return nil
} }
return p.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error { return p.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
if err := tx.Where("owner = ? AND kb = ? AND from_name = ?", owner, kb, from).Delete(&DocLink{}).Error; err != nil { if err := tx.Where("owner = ? AND kb = ? AND from_id = ?", owner, kb, fromID).Delete(&DocLink{}).Error; err != nil {
return err return err
} }
for _, to := range tos { for _, name := range toNames {
if to == "" || to == from { if name == "" {
continue continue
} }
if err := tx.Create(&DocLink{Owner: owner, KB: kb, FromName: from, ToName: to}).Error; err != nil { var t Doc
toID := ""
if e := tx.Select("id").Where("owner = ? AND kb = ? AND name = ?", owner, kb, name).First(&t).Error; e == nil {
toID = t.ID
}
if toID == fromID { // 自链跳过
continue
}
if err := tx.Create(&DocLink{Owner: owner, KB: kb, FromID: fromID, ToID: toID, ToName: name}).Error; err != nil {
return err return err
} }
} }
@@ -177,13 +194,23 @@ func (p *Postgres) ReplaceDocLinks(ctx context.Context, owner, kb, from string,
}) })
} }
// ListLinks 返回某 kb 的全部 [[双链]]from→to),供反链/笔记关系图按需查询(数据小) // ResolveInboundLinks 把指向 name 的悬空链接(ToID 空)回填为 id —— 目标文档入库后调用,使反链/关系图即时成形
func (p *Postgres) ResolveInboundLinks(ctx context.Context, owner, kb, name, id string) error {
if p.db == nil {
return nil
}
return p.db.WithContext(ctx).Model(&DocLink{}).
Where("owner = ? AND kb = ? AND to_name = ? AND (to_id = '' OR to_id IS NULL)", owner, kb, name).
Update("to_id", id).Error
}
// ListLinks 返回某 kb 已解析(两端均为本库文件)的 [[双链]](FromID→ToID),供反链/笔记关系图按 ID 渲染。
func (p *Postgres) ListLinks(ctx context.Context, owner, kb string) ([]DocLink, error) { func (p *Postgres) ListLinks(ctx context.Context, owner, kb string) ([]DocLink, error) {
if p.db == nil { if p.db == nil {
return nil, nil return nil, nil
} }
var rows []DocLink var rows []DocLink
err := p.db.WithContext(ctx).Where("owner = ? AND kb = ?", owner, kb).Find(&rows).Error err := p.db.WithContext(ctx).Where("owner = ? AND kb = ? AND to_id <> ''", owner, kb).Find(&rows).Error
return rows, err return rows, err
} }
+14
View File
@@ -37,6 +37,7 @@ func OpenPostgres(dsn string) *Postgres {
// 一次性迁移:旧表用整型自增 id,与新雪花字符串 id 不兼容(AutoMigrate 不改主键类型)。 // 一次性迁移:旧表用整型自增 id,与新雪花字符串 id 不兼容(AutoMigrate 不改主键类型)。
// 备份模型密钥(唯一不可再生的数据) → 重建全部表 → 回灌模型。其余为可重建的测试数据。 // 备份模型密钥(唯一不可再生的数据) → 重建全部表 → 回灌模型。其余为可重建的测试数据。
migrateLegacyIntIDs(db) migrateLegacyIntIDs(db)
migrateDocLinkToID(db)
if err := db.AutoMigrate(&User{}, &Task{}, &LLMModel{}, &KB{}, &Doc{}, &Agent{}, &DocLink{}); err != nil { if err := db.AutoMigrate(&User{}, &Task{}, &LLMModel{}, &KB{}, &Doc{}, &Agent{}, &DocLink{}); err != nil {
log.Printf("[store] postgres AutoMigrate 失败,降级运行: %v", err) log.Printf("[store] postgres AutoMigrate 失败,降级运行: %v", err)
@@ -71,6 +72,19 @@ func migrateLegacyIntIDs(db *gorm.DB) {
log.Printf("[store] 已回灌 %d 条模型配置(新雪花 id)", len(saved)) log.Printf("[store] 已回灌 %d 条模型配置(新雪花 id)", len(saved))
} }
// migrateDocLinkToID 把旧的按名双链表(from_name/to_name)迁到按 Doc.ID 关联的新表。
// 旧表无 from_id 列即判定为旧 schema:直接删表,由 AutoMigrate 重建;链接随文档再入库/编辑重建。
func migrateDocLinkToID(db *gorm.DB) {
if !db.Migrator().HasTable("sundynix_doc_link") {
return
}
if db.Migrator().HasColumn(&DocLink{}, "from_id") {
return // 已是按 ID 关联的新 schema
}
log.Println("[store] 双链表升级为按文件 ID 关联,重建 sundynix_doc_link(链接随文档再入库重建)")
db.Exec("DROP TABLE IF EXISTS sundynix_doc_link CASCADE")
}
// Enabled 报告是否处于真实持久化模式。 // Enabled 报告是否处于真实持久化模式。
func (p *Postgres) Enabled() bool { return p.db != nil } func (p *Postgres) Enabled() bool { return p.db != nil }