feat(index): nodes 表增加 node_type 列，查询过滤虚拟节点

ai_xiaopei

2026-08-21 7bcbed8c3547e55287972bd0bccc7c9e9d8ec495

 internal/index/sqlite.go

@@ -55,6 +55,7 @@
   CREATE TABLE IF NOT EXISTS nodes (
      id          INTEGER PRIMARY KEY AUTOINCREMENT,
      path        TEXT NOT NULL UNIQUE,
      node_type   TEXT NOT NULL DEFAULT 'file',
      title       TEXT,
      section     TEXT,
      tags        TEXT,
@@ -83,6 +84,15 @@
   if err != nil {
      return fmt.Errorf("创建表失败: %w", err)
   }

   // 旧库迁移：补 node_type 列（幂等）
   var colCount int
   if err := s.db.QueryRow("SELECT COUNT(*) FROM pragma_table_info('nodes') WHERE name = 'node_type'").Scan(&colCount); err == nil && colCount == 0 {
      if _, err := s.db.Exec("ALTER TABLE nodes ADD COLUMN node_type TEXT NOT NULL DEFAULT 'file'"); err != nil {
         return fmt.Errorf("迁移 node_type 列失败: %w", err)
      }
   }

   return nil
}

@@ -99,9 +109,9 @@
   wikilinksJSON, _ := json.Marshal(n.Wikilinks)

   result, err := s.db.Exec(`
      INSERT INTO nodes (path, title, section, tags, entities, wikilinks, content_fts)
      VALUES (?, ?, ?, ?, ?, ?, ?)
   `, n.Path, n.Title, n.Section, string(tagsJSON), string(entitiesJSON),
      INSERT INTO nodes (path, node_type, title, section, tags, entities, wikilinks, content_fts)
      VALUES (?, ?, ?, ?, ?, ?, ?, ?)
   `, n.Path, nodeTypeOf(n), n.Title, n.Section, string(tagsJSON), string(entitiesJSON),
      string(wikilinksJSON), n.Content)
   if err != nil {
      return 0, err
@@ -109,6 +119,14 @@
   return result.LastInsertId()
}

// nodeTypeOf 节点类型（空值归一为 file）
func nodeTypeOf(n *graph.Node) string {
   if n.NodeType == "" {
      return "file"
   }
   return n.NodeType
}

// InsertEdge 插入边
func (s *Store) InsertEdge(e *graph.Edge) error {
   _, err := s.db.Exec(`
@@ -188,7 +206,7 @@

// GetAllNodes 获取所有节点（用于 GC 检查）
func (s *Store) GetAllNodes() ([]NodeInfo, error) {
   rows, err := s.db.Query("SELECT id, path FROM nodes")
   rows, err := s.db.Query("SELECT id, path FROM nodes WHERE node_type = 'file'")
   if err != nil {
      return nil, fmt.Errorf("查询节点失败: %w", err)
   }
@@ -296,7 +314,8 @@
   query := `
      SELECT id, path, title, section, tags, entities, wikilinks, content_fts
      FROM nodes
      WHERE title LIKE ? OR path LIKE ? OR tags LIKE ? OR entities LIKE ?
      WHERE (title LIKE ? OR path LIKE ? OR tags LIKE ? OR entities LIKE ?)
      AND node_type = 'file'
      LIMIT 20
   `
   rows, err := s.db.Query(query, keyword, keyword, keyword, keyword)
@@ -399,7 +418,8 @@
   query := `
      SELECT id, path, title, section, tags, entities
      FROM nodes
      WHERE title LIKE ? OR path LIKE ? OR tags LIKE ? OR entities LIKE ?
      WHERE (title LIKE ? OR path LIKE ? OR tags LIKE ? OR entities LIKE ?)
      AND node_type = 'file'
      LIMIT 5
   `
   rows, err := s.db.Query(query, keyword, keyword, keyword, keyword)
@@ -450,7 +470,7 @@
      rows, err := s.db.Query(`
         SELECT id, path, title, section, tags
         FROM nodes
         WHERE tags LIKE ?
         WHERE tags LIKE ? AND node_type = 'file'
      `, tagPattern)
      if err != nil {
         continue
@@ -480,7 +500,7 @@
      rows, err := s.db.Query(`
         SELECT id, path, title, section, tags
         FROM nodes
         WHERE entities LIKE ?
         WHERE entities LIKE ? AND node_type = 'file'
      `, entityPattern)
      if err != nil {
         continue