feat(memory): 图数据库自报真实读写事件,取代「按工具名猜节点」

## 问题(生产实证)

星图的「跟随 agent 活动」此前靠**工具名词元猜节点**:把 `memory_recall`
拆成 [memory, recall] 去匹配实体名。而图谱里的实体名是**中文概念**
(小宅/对话/待命),工具名是英文 —— 两套命名空间永不交集。

实测(生产库)每个工具名命中 **0** 个节点:

    hint=memory_recall  → memory/recall  → 0
    hint=files_read     → files/read     → 0
    hint=knowledge_search → knowledge/search → 0

匹配失败后走「补齐」分支,按 mention_count 取前 8 个 ⇒ **无论 agent 在干什么,
点亮的都是同一批无关节点**(小宅/对话/待命/投递/图片/文档/插件/老大)。
这就是「星图没有读写反馈」的真身。

## 改法:让图数据库在真正读写处上报

新增 `events.EventMemoryAccess` + `GraphDB.AccessHook`,在读写入口埋点上报
**真实块 ID**(不再猜):

| 操作 | 埋点 | 上报 |
|---|---|---|
| 读 | RecallSorted | blocks |
| 写 | commit | blocks + **created**(真正新建的) |
| 删 | DeleteEntity | removed |
| 合 | MergeBlocks | merged(from→to) + removed |

三处关键设计:

1. **钩子用独立锁** `hookMu`。用 `g.mu` 会**自死锁** —— 埋点方法已持该锁,
   而 Go 的 RWMutex 不可重入。
2. **上报在锁外**。拆 `commit`→`commitLocked` 两层解决 `defer Unlock` 的时序:
   钩子内部可能要回调内存系统(如 SSE 订阅者补数据),持锁调用必死锁。
3. **区分「新建」与「更新」**(`putBlockTx` 返回 isNew)。用 SELECT 判存在性
   而非 `RETURNING (xmax = 0)` —— 后者在**事务 + ON CONFLICT** 下报
   `no such column: xmax`(实测:autocommit 可用、包进 tx 就不可用),
   而本函数只在事务里被调用。重复提交同一事实不会重复播「生长」动画。

接线:`bootstrap.newMainAgent` 里 `memDB.SetAccessHook(...)` → `events.Bus`。
只在 `evBus != nil` 时接,把「内核进程」与「独立 CLI 工具」
(homed-graph-migrate / memgc 也用 NewGraphDB)区分开 —— 后者不该为一次
批量迁移向不存在的订阅者发几万条事件。

## 验证

- `access_hook_test.go`:钉住 commit/recall/delete 三类埋点真的触发、
  且重复提交不误报新建
- `access_hook_race_test.go`:4 写 + 4 读 + 钩子热替换并发,`-race` 下无竞态、
  无死锁;**断言事件数 > 0**(否则「钩子从未触发」也会假通过)
- 该测试起初是 flaky 的(读者可能在写者落库前查询 ⇒ 空结果不报事件 ⇒
  断言随机失败)。修法是先播种确定数据再并发读,已连跑 3 次稳定
- 生产实例端到端:CLI 注入一句话 → SSE 收到 6 条 memory_access,
  带真实块 ID 与 op/created
This commit is contained in:
JianFeeeee
2026-10-08 10:41:07 +08:00
parent e6c1bfa6bf
commit 9036db9321
10 changed files with 568 additions and 81 deletions

View File

@ -124,10 +124,10 @@ func (g *GraphDB) MigrateLegacyTextEntities(embed EntityEmbedder) (MigrateResult
blockID := legacyEntityBlockID(e.id, e.name)
src := NewSentenceBlock(e.name, e.createdAt, e.updatedAt)
srcBlockID := src.ID
if err := putBlockTx(tx, src); err != nil {
if _, err := putBlockTx(tx, src); err != nil {
return res, fmt.Errorf("put sentence block for entity %d: %w", e.id, err)
}
if err := putBlockTx(tx, MemoryBlock{
if _, err := putBlockTx(tx, MemoryBlock{
ID: blockID,
Modality: BlockText,
Text: e.name,
@ -244,7 +244,7 @@ func (g *GraphDB) MigrateLegacyTextEntities(embed EntityEmbedder) (MigrateResult
return res, err
}
sb := NewSentenceBlock(text, time.Time{}, time.Time{})
if err := putBlockTx(tx, sb); err != nil {
if _, err := putBlockTx(tx, sb); err != nil {
_ = sentRows.Close()
return res, fmt.Errorf("put sentence block: %w", err)
}
@ -406,12 +406,23 @@ func ensureSentenceTx(tx *sql.Tx, text string) (int64, error) {
return id, nil
}
func putBlockTx(tx *sql.Tx, b MemoryBlock) error {
// ★ 返回值:第一个是**是否真的新建了行**(供星图「生长」动画区分新建/更新),
// 第二个是 error。
//
// 为何要区分:星图的「生长」动画只该在**真·新块**上播。若把「写过了」
// 也算新建,重复提交同一事实(LLM 高频重试)会让旧块反复长出来 ——
// 看着像新记忆,实际是同一个东西。
//
// 判据走「先 SELECT 存在性」而不是 `RETURNING (xmax = 0)`:后者在
// **事务 + ON CONFLICT** 组合下报 `no such column: xmax`
// (实测:autocommit 下可用,套上 tx.Begin() 后不可用),而本函数只在事务里被调用。
// 代价是多一次主键点查,开销可忽略。
func putBlockTx(tx *sql.Tx, b MemoryBlock) (bool, error) {
vectorJSON := ""
if len(b.Vector) > 0 {
raw, err := json.Marshal(b.Vector)
if err != nil {
return err
return false, err
}
vectorJSON = string(raw)
}
@ -455,6 +466,26 @@ func putBlockTx(tx *sql.Tx, b MemoryBlock) error {
if !explicitTime {
tsUpdate = "updated_at = memory_blocks.updated_at"
}
// ★ 先用 SELECT 判定「是否已存在」,而不是靠 RETURNING (xmax = 0)。
//
// 为什么不用 xmax:它在**事务 + ON CONFLICT** 组合下直接报
// `no such column: xmax`(实测:同样的语句在 autocommit 下可用,
// 一句 tx.Begin() 包起来后就不可用)。而本函数**只在事务里**被调用
// (调用方全部持 tx)⇒ xmax 在这里永远不可用。
//
// 代价:多一次主键查找。这是索引点查,开销可忽略;
// 而它换来的是可靠的「新建 vs 更新」判据 —— 星图的生长动画依赖它,
// 报错就等于整个记忆写入失败(比多一次点查贵得多)。
var existed int
switch err := tx.QueryRow(`SELECT 1 FROM memory_blocks WHERE id = ?`, b.ID).Scan(&existed); err {
case nil:
// 已存在 ⇒ 本次是更新,不算新建。
case sql.ErrNoRows:
// 不存在 ⇒ 本次会新建。
default:
return false, err
}
_, err := tx.Exec(`INSERT INTO memory_blocks
(id, modality, text_content, payload_digest, mime, size, width, height,
vector, fingerprint, source, tool, scene, semantic_type, created_at, updated_at)
@ -471,7 +502,10 @@ func putBlockTx(tx *sql.Tx, b MemoryBlock) error {
b.ID, b.Modality, b.Text, b.PayloadDigest, b.MIME, b.Size, b.Width, b.Height,
vectorJSON, b.Fingerprint, b.Source, b.Tool, b.Scene, b.SemanticType,
created, updated)
return err
if err != nil {
return false, err
}
return existed == 0, nil
}
// ★ 返回 (error, 是否真的新增了一行)。第二个返回值是给报告口径用的 ——