package index
|
|
import (
|
"fmt"
|
"os"
|
)
|
|
const currentSchemaVersion = 2
|
|
// migrate 幂等的 schema 升级：v1 → v2。
|
// v2 变更：nodes 加 aliases/status/size/mtime/content_hash 列；
|
// edges 加 provenance 列；新建 unresolved_links、schema_versions 表；
|
// nodes_fts 改 external-content + 触发器。迁移前自动备份 .db 文件。
|
func (s *Store) migrate() error {
|
// 读当前版本
|
var version int
|
err := s.db.QueryRow(`SELECT value FROM schema_versions WHERE key='version'`).Scan(&version)
|
if err != nil && version == 0 {
|
// 首次：schema_versions 表可能不存在
|
version = 1
|
}
|
if version >= currentSchemaVersion {
|
return nil
|
}
|
|
// 备份
|
if err := s.backupDB(); err != nil {
|
return fmt.Errorf("备份失败: %w", err)
|
}
|
|
// v1 → v2
|
if err := s.migrateV1toV2(); err != nil {
|
return fmt.Errorf("v1→v2 迁移失败: %w", err)
|
}
|
return nil
|
}
|
|
// backupDB 迁移前备份 .db 文件（已有备份则跳过）。
|
// 用 VACUUM INTO 而非 os.ReadFile：WAL 模式下未 checkpoint 的数据在
|
// -wal 文件里，直读主文件会漏掉这部分，导致备份不一致；VACUUM INTO
|
// 会合并 WAL 生成一份完整的独立库文件（SQLite 3.27+）。
|
func (s *Store) backupDB() error {
|
if s.dbPath == "" {
|
return nil // 无路径信息时跳过备份（测试场景）
|
}
|
dst := s.dbPath + ".bak"
|
if _, err := os.Stat(dst); err == nil {
|
return nil // 已有备份
|
}
|
_, err := s.db.Exec(`VACUUM INTO ?`, dst)
|
return err
|
}
|
|
// migrateV1toV2 执行 v1 → v2 的全部变更。
|
// 注意：版本号（schema_versions）必须在最后一步（FTS 重建）成功后才写入，
|
// 否则中途失败时后续 Open 会读到 version=2 直接早退，半迁移状态无法重试。
|
func (s *Store) migrateV1toV2() error {
|
// 1. schema_versions 表（只建表，版本号推迟到最后写入）
|
if _, err := s.db.Exec(`CREATE TABLE IF NOT EXISTS schema_versions (
|
key TEXT PRIMARY KEY, value TEXT NOT NULL)`); err != nil {
|
return err
|
}
|
|
// 2. nodes 加列（逐列检查，幂等）
|
for _, col := range []struct{ name, def string }{
|
{"aliases", "ALTER TABLE nodes ADD COLUMN aliases TEXT"},
|
{"status", "ALTER TABLE nodes ADD COLUMN status TEXT"},
|
{"size", "ALTER TABLE nodes ADD COLUMN size INTEGER"},
|
{"mtime", "ALTER TABLE nodes ADD COLUMN mtime INTEGER"},
|
{"content_hash", "ALTER TABLE nodes ADD COLUMN content_hash TEXT"},
|
} {
|
if !s.nodeColumnExists(col.name) {
|
if _, err := s.db.Exec(col.def); err != nil {
|
return fmt.Errorf("加列 %s 失败: %w", col.name, err)
|
}
|
}
|
}
|
|
// 3. edges 加 provenance 列
|
if !s.edgeColumnExists("provenance") {
|
if _, err := s.db.Exec(`ALTER TABLE edges ADD COLUMN provenance TEXT`); err != nil {
|
return fmt.Errorf("edges 加 provenance 失败: %w", err)
|
}
|
}
|
|
// 4. unresolved_links 表
|
if _, err := s.db.Exec(`CREATE TABLE IF NOT EXISTS unresolved_links (
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
from_node INTEGER NOT NULL,
|
link_text TEXT NOT NULL,
|
name_tail TEXT NOT NULL,
|
status TEXT NOT NULL DEFAULT 'pending',
|
created_at TEXT DEFAULT (datetime('now')),
|
last_attempt TEXT)`); err != nil {
|
return err
|
}
|
if _, err := s.db.Exec(`CREATE INDEX IF NOT EXISTS idx_unresolved_tail
|
ON unresolved_links(name_tail) WHERE status='failed'`); err != nil {
|
return err
|
}
|
|
// 5. FTS 改 external-content：删旧表重建 + 触发器 + 重灌
|
if _, err := s.db.Exec(`DROP TABLE IF EXISTS nodes_fts`); err != nil {
|
return err
|
}
|
if err := s.CreateFTS(); err != nil {
|
return err
|
}
|
if err := s.PopulateFTS(); err != nil {
|
return err
|
}
|
|
// 6. 最后一步：全部成功后才写入版本号（参数化，引用常量 currentSchemaVersion）。
|
// 提前写入会导致中途失败时后续 Open 读到 version=2 直接早退，半迁移状态无法重试。
|
if _, err := s.db.Exec(`INSERT OR REPLACE INTO schema_versions VALUES ('version', ?)`,
|
currentSchemaVersion); err != nil {
|
return err
|
}
|
return nil
|
}
|
|
// nodeColumnExists 检查 nodes 表是否已有指定列
|
func (s *Store) nodeColumnExists(name string) bool {
|
var n int
|
s.db.QueryRow(`SELECT COUNT(*) FROM pragma_table_info('nodes') WHERE name=?`, name).Scan(&n)
|
return n > 0
|
}
|
|
// edgeColumnExists 检查 edges 表是否已有指定列
|
func (s *Store) edgeColumnExists(name string) bool {
|
var n int
|
s.db.QueryRow(`SELECT COUNT(*) FROM pragma_table_info('edges') WHERE name=?`, name).Scan(&n)
|
return n > 0
|
}
|
