mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-03 15:53:56 +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:
143
internal/ipc/ipc.go
Normal file
143
internal/ipc/ipc.go
Normal file
@ -0,0 +1,143 @@
|
||||
// Package ipc 实现 guard↔worker 之间的 PING/ACK 心跳协议:worker 监听 unix socket,
|
||||
// guard 发 PING,worker 回 ACK(含 kernel 状态快照),实现带自诊断上报的存活判定,
|
||||
// 取代单纯的文件心跳 + 退出码。
|
||||
package ipc
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
)
|
||||
|
||||
// sinkPath 心跳 socket 文件路径。
|
||||
func sinkPath(dataDir string) string {
|
||||
return filepath.Join(dataDir, "worker.ipc")
|
||||
}
|
||||
|
||||
// Status worker 上报给 guard 的自诊断快照。
|
||||
type Status struct {
|
||||
PID int `json:"pid"`
|
||||
Boot string `json:"boot"` // normal | failback
|
||||
UptimeSec int64 `json:"uptime_sec"` // 进程存活秒数
|
||||
LLMOK *bool `json:"llm_ok,omitempty"` // rescue 源可达性(nil=未探)
|
||||
Tools int `json:"tools"` // 已注册工具数
|
||||
LastDiag string `json:"last_diag,omitempty"` // 最近一次自诊断结论(如 diag_loc.cause)
|
||||
}
|
||||
|
||||
// StatusFunc 组装 worker 当前状态。
|
||||
type StatusFunc func() *Status
|
||||
|
||||
// Server worker 侧:监听 unix socket,处理 PING→ACK。
|
||||
type Server struct {
|
||||
path string
|
||||
ln net.Listener
|
||||
status StatusFunc
|
||||
stop chan struct{}
|
||||
done chan struct{}
|
||||
}
|
||||
|
||||
// NewServer 创建心跳服务端(尚未 Listen,见 Start)。
|
||||
func NewServer(dataDir string, status StatusFunc) *Server {
|
||||
return &Server{path: sinkPath(dataDir), status: status, stop: make(chan struct{}), done: make(chan struct{})}
|
||||
}
|
||||
|
||||
// Start 启动监听。若 socket 已存在则先清理(无心跳残留)。
|
||||
func (s *Server) Start() error {
|
||||
if s.status == nil {
|
||||
s.status = func() *Status { return nil }
|
||||
}
|
||||
if err := os.RemoveAll(s.path); err != nil {
|
||||
return err
|
||||
}
|
||||
ln, err := net.Listen("unix", s.path)
|
||||
if err != nil {
|
||||
return fmt.Errorf("ipc listen %s: %w", s.path, err)
|
||||
}
|
||||
s.ln = ln
|
||||
go s.acceptLoop()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Server) acceptLoop() {
|
||||
defer close(s.done)
|
||||
for {
|
||||
conn, err := s.ln.Accept()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
go s.handle(conn)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) handle(conn net.Conn) {
|
||||
defer conn.Close()
|
||||
conn.SetDeadline(time.Now().Add(5 * time.Second))
|
||||
sc := bufio.NewScanner(conn)
|
||||
if !sc.Scan() {
|
||||
return
|
||||
}
|
||||
line := sc.Text()
|
||||
if line != "PING" {
|
||||
return
|
||||
}
|
||||
resp := map[string]interface{}{"type": "ACK"}
|
||||
if st := s.status(); st != nil {
|
||||
resp["status"] = st
|
||||
}
|
||||
data, _ := json.Marshal(resp)
|
||||
conn.Write(append(data, '\n'))
|
||||
}
|
||||
|
||||
// Stop 关闭监听。
|
||||
func (s *Server) Stop() {
|
||||
if s.ln != nil {
|
||||
s.ln.Close()
|
||||
}
|
||||
<-s.done
|
||||
}
|
||||
|
||||
// Client guard 侧:向 worker 心跳 socket 发 PING 并等 ACK。
|
||||
type Client struct {
|
||||
path string
|
||||
}
|
||||
|
||||
// NewClient 创建客户端。
|
||||
func NewClient(dataDir string) *Client {
|
||||
return &Client{path: sinkPath(dataDir)}
|
||||
}
|
||||
|
||||
// Ping 发一次 PING、收 ACK。timeout 内未收到返回错误。返回 (status, error);
|
||||
// status 可能为 nil(ACK 无状态体)。
|
||||
func (c *Client) Ping(timeout time.Duration) (*Status, error) {
|
||||
conn, err := net.DialTimeout("unix", c.path, timeout)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer conn.Close()
|
||||
conn.SetDeadline(time.Now().Add(timeout))
|
||||
if _, err := conn.Write([]byte("PING\n")); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sc := bufio.NewScanner(conn)
|
||||
if !sc.Scan() {
|
||||
return nil, fmt.Errorf("ipc: empty ACK")
|
||||
}
|
||||
var resp struct {
|
||||
Type string `json:"type"`
|
||||
Status *Status `json:"status"`
|
||||
}
|
||||
if err := json.Unmarshal(sc.Bytes(), &resp); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.Type != "ACK" {
|
||||
return nil, fmt.Errorf("ipc: unexpected reply %q", resp.Type)
|
||||
}
|
||||
return resp.Status, nil
|
||||
}
|
||||
|
||||
// Close 无状态(客户端用完即释放连接)。
|
||||
func (c *Client) Close() {}
|
||||
67
internal/ipc/ipc_test.go
Normal file
67
internal/ipc/ipc_test.go
Normal file
@ -0,0 +1,67 @@
|
||||
package ipc
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestPingAck(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
ok := true
|
||||
srv := NewServer(dir, func() *Status {
|
||||
return &Status{PID: 123, Boot: "normal", UptimeSec: 42, LLMOK: &ok, Tools: 5, LastDiag: "diag:network"}
|
||||
})
|
||||
if err := srv.Start(); err != nil {
|
||||
t.Fatalf("Start: %v", err)
|
||||
}
|
||||
defer srv.Stop()
|
||||
|
||||
cli := NewClient(dir)
|
||||
st, err := cli.Ping(2 * time.Second)
|
||||
if err != nil {
|
||||
t.Fatalf("Ping: %v", err)
|
||||
}
|
||||
if st == nil {
|
||||
t.Fatal("nil status")
|
||||
}
|
||||
if st.PID != 123 || st.Boot != "normal" || st.Tools != 5 {
|
||||
t.Fatalf("status mismatch: %+v", st)
|
||||
}
|
||||
if st.LLMOK == nil || !*st.LLMOK {
|
||||
t.Fatal("LLMOK should be true")
|
||||
}
|
||||
if st.LastDiag != "diag:network" {
|
||||
t.Fatalf("LastDiag = %q", st.LastDiag)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPingTimeout(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
// 未启动 server → 立即超时
|
||||
cli := NewClient(dir)
|
||||
start := time.Now()
|
||||
_, err := cli.Ping(500 * time.Millisecond)
|
||||
if err == nil {
|
||||
t.Fatal("expected error when no server")
|
||||
}
|
||||
if time.Since(start) > 3*time.Second {
|
||||
t.Fatal("timeout took too long")
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerNoStatusFunc(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
srv := NewServer(dir, nil)
|
||||
if err := srv.Start(); err != nil {
|
||||
t.Fatalf("Start: %v", err)
|
||||
}
|
||||
defer srv.Stop()
|
||||
cli := NewClient(dir)
|
||||
st, err := cli.Ping(2 * time.Second)
|
||||
if err != nil {
|
||||
t.Fatalf("Ping: %v", err)
|
||||
}
|
||||
if st != nil {
|
||||
t.Fatalf("expected nil status when no status func, got %+v", st)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user