From 20f89da11ad06cfb1b0511b645b10a0f8093623e Mon Sep 17 00:00:00 2001
From: ax_rd <ax_rd@aisim.cn>
Date: Thu, 03 Sep 2026 08:14:51 +0800
Subject: [PATCH] docs: 对标CodeGraph增强设计（RWR排序/增量同步/悬空链接补全/explore命令）
---
cmd/rebuild.go | 31 ++++++++++++++++++++++++++-----
1 files changed, 26 insertions(+), 5 deletions(-)
diff --git a/cmd/rebuild.go b/cmd/rebuild.go
index 4c488cd..dfff961 100644
--- a/cmd/rebuild.go
+++ b/cmd/rebuild.go
@@ -28,18 +28,39 @@
// 构建图
g := graph.BuildGraph(files)
- // 写入节点
+ // 写入节点，并记录 BuildGraph ID -> SQLite ID 的映射
+ idMap := make(map[int64]int64) // BuildGraph ID -> SQLite ID
for _, n := range g.Nodes {
- if _, err := store.InsertNode(n); err != nil {
+ sqliteID, err := store.InsertNode(n)
+ if err != nil {
return fmt.Errorf("插入节点失败 [%s]: %w", n.Path, err)
}
+ idMap[n.ID] = sqliteID
}
- // 写入边
+ // 写入边，将 BuildGraph ID 转换为 SQLite ID
+ actualEdgeCount := 0
for _, e := range g.Edges {
- if err := store.InsertEdge(e); err != nil {
+ fromID, ok := idMap[e.FromNode]
+ if !ok {
+ // 可能是虚拟节点（tag/entity），跳过
+ continue
+ }
+ toID, ok := idMap[e.ToNode]
+ if !ok {
+ // 目标节点不存在，跳过
+ continue
+ }
+ edge := &graph.Edge{
+ FromNode: fromID,
+ ToNode: toID,
+ Relation: e.Relation,
+ Label: e.Label,
+ }
+ if err := store.InsertEdge(edge); err != nil {
return fmt.Errorf("插入边失败: %w", err)
}
+ actualEdgeCount++
}
// 创建并填充 FTS5 索引
@@ -60,6 +81,6 @@
return fmt.Errorf("记录构建时间失败: %w", err)
}
- fmt.Fprintf(os.Stderr, "已索引 %d 个节点, %d 条边\n", len(g.Nodes), len(g.Edges))
+ fmt.Fprintf(os.Stderr, "已索引 %d 个节点, %d 条边\n", len(g.Nodes), actualEdgeCount)
return nil
}
--
Gitblit v1.10.0