mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-15 12:15:16 +00:00
三层回退恢复机制(L0写前留档/L1恢复梯子/L2离线回滚)+ guard 父守护
- L0: files 插件写受保护系统路径(/etc 等)前自动留档,AbstractBeforeWrite 到 data/file_baseline - L1: failback 受限 worker 执行恢复梯子 probe→还原DNS/proxy→还原LLM配置+ReloadFromConfig→probe,N轮有界 - L2: tracker changeset 持久化原文 blob,guard 离线 RollbackFromDisk 回滚 agentfs;SystemSnapshot 支撑 - guard 父守护: 心跳 IPC(PING/ACK unix socket, 文件心跳回退)、失败计数、退出码协议(42/43/44)、最后手段 - 发行版路径适配: system.protected_paths/network_paths 可注入,默认面向主流 Linux - Windows 兼容: guard.go/failback.go 加 //go:build linux, guard_windows.go 提供 no-op 桩 - 修复: guard.yaml last_resort 键冲突、changeset Content 不落盘导致离线回滚丢原文 Build 全绿, vet 干净, system/recovery/ipc/tracker 单元测试全过
This commit is contained in:
92
internal/tracker/disk_test.go
Normal file
92
internal/tracker/disk_test.go
Normal file
@ -0,0 +1,92 @@
|
||||
package tracker
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// 构造一条写入磁盘的 changeset(模拟 worker 运行期间的变更),随后用
|
||||
// RollbackFromDisk 离线还原。
|
||||
func TestRollbackFromDisk(t *testing.T) {
|
||||
dataDir := t.TempDir()
|
||||
workDir := t.TempDir()
|
||||
|
||||
trk := NewTracker(dataDir, workDir)
|
||||
if err := trk.Init(); err != nil {
|
||||
t.Fatalf("Init: %v", err)
|
||||
}
|
||||
upper := trk.upperDir
|
||||
os.MkdirAll(upper, 0755)
|
||||
|
||||
// 1. 先创建一个文件(对应"新建")
|
||||
newFile := filepath.Join(upper, "new.txt")
|
||||
os.WriteFile(newFile, []byte("brand new"), 0644)
|
||||
|
||||
// 2. 修改一个文件(对应"修改",带原文)
|
||||
modFile := filepath.Join(upper, "mod.txt")
|
||||
os.WriteFile(modFile, []byte("after"), 0644)
|
||||
beforeContent := []byte("before")
|
||||
csMod := &ChangeSet{
|
||||
ID: "cs_mod",
|
||||
Files: []FileChange{{
|
||||
Path: "mod.txt",
|
||||
Type: ChangeFileModified,
|
||||
Content: beforeContent,
|
||||
}},
|
||||
}
|
||||
trk.saveChangeSet(csMod)
|
||||
csNew := &ChangeSet{
|
||||
ID: "cs_new",
|
||||
Files: []FileChange{{
|
||||
Path: "new.txt",
|
||||
Type: ChangeFileCreated,
|
||||
}},
|
||||
}
|
||||
trk.saveChangeSet(csNew)
|
||||
|
||||
if trk.ChangesetsOnDisk() != 2 {
|
||||
t.Fatalf("ChangesetsOnDisk = %d, want 2", trk.ChangesetsOnDisk())
|
||||
}
|
||||
|
||||
// 离线回滚(模拟 guard 在 worker 崩溃后调用)
|
||||
n, err := trk.RollbackFromDisk()
|
||||
if err != nil {
|
||||
t.Fatalf("RollbackFromDisk: %v", err)
|
||||
}
|
||||
if n != 2 {
|
||||
t.Fatalf("rolled back %d, want 2", n)
|
||||
}
|
||||
if trk.ChangesetsOnDisk() != 0 {
|
||||
t.Fatalf("ChangesetsOnDisk after rollback = %d, want 0", trk.ChangesetsOnDisk())
|
||||
}
|
||||
|
||||
// 新建文件被删除
|
||||
if _, err := os.Stat(newFile); !os.IsNotExist(err) {
|
||||
t.Fatal("created file should be removed after rollback")
|
||||
}
|
||||
// 修改文件还原原文
|
||||
data, err := os.ReadFile(modFile)
|
||||
if err != nil {
|
||||
t.Fatalf("read modFile: %v", err)
|
||||
}
|
||||
if string(data) != "before" {
|
||||
t.Fatalf("modFile restored to %q, want %q", data, beforeContent)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRollbackFromDiskEmpty(t *testing.T) {
|
||||
dataDir := t.TempDir()
|
||||
workDir := t.TempDir()
|
||||
trk := NewTracker(dataDir, workDir)
|
||||
if err := trk.Init(); err != nil {
|
||||
t.Fatalf("Init: %v", err)
|
||||
}
|
||||
n, err := trk.RollbackFromDisk()
|
||||
if err != nil {
|
||||
t.Fatalf("RollbackFromDisk: %v", err)
|
||||
}
|
||||
if n != 0 {
|
||||
t.Fatalf("rolled back %d, want 0", n)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user