From 09c2a03cefa687832257d3b816da50db3cfcc0f2 Mon Sep 17 00:00:00 2001
From: ai_xiaopei <xiaopei@aisim.cn>
Date: Sat, 25 Jul 2026 23:48:11 +0800
Subject: [PATCH] feat: 实现输出格式化
---
internal/index/fts_test.go | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 54 insertions(+), 0 deletions(-)
diff --git a/internal/index/fts_test.go b/internal/index/fts_test.go
new file mode 100644
index 0000000..1a42430
--- /dev/null
+++ b/internal/index/fts_test.go
@@ -0,0 +1,54 @@
+package index
+
+import (
+ "path/filepath"
+ "testing"
+
+ "github.com/aisim/kb-cli/internal/graph"
+)
+
+func TestFTSSearch(t *testing.T) {
+ tmpDir := t.TempDir()
+ dbPath := filepath.Join(tmpDir, "test.db")
+
+ store, err := Open(dbPath)
+ if err != nil {
+ t.Fatalf("Open failed: %v", err)
+ }
+ defer store.Close()
+
+ // 插入测试数据
+ store.InsertNode(&graph.Node{
+ ID: 1,
+ Path: "FAQ/充装类/001-test.md",
+ Title: "充装问题排查",
+ Section: "FAQ",
+ Tags: []string{"充装"},
+ Content: "关于智能枪充装问题的排查方法",
+ })
+ store.InsertNode(&graph.Node{
+ ID: 2,
+ Path: "知识/002-config.md",
+ Title: "充装规格配置",
+ Section: "知识",
+ Tags: []string{"配置"},
+ Content: "充装规格配置说明",
+ })
+
+ // 创建 FTS
+ if err := store.CreateFTS(); err != nil {
+ t.Fatalf("CreateFTS failed: %v", err)
+ }
+ if err := store.PopulateFTS(); err != nil {
+ t.Fatalf("PopulateFTS failed: %v", err)
+ }
+
+ // 搜索 - FTS5 对中文按字符 tokenize,"充装" 两个节点都包含
+ results, err := store.FTSSearch([]string{"充装"}, 10)
+ if err != nil {
+ t.Fatalf("FTSSearch failed: %v", err)
+ }
+ if len(results) < 1 {
+ t.Errorf("result count = %d, want >= 1", len(results))
+ }
+}
--
Gitblit v1.9.1