实现代理节点功能，包括主程序入口、配置加载、心跳检测和状态管理，添加相关服务和API响应结构

atsf_agent/internal/sync/service.go

@@ -0,0 +1,73 @@
+package sync
+
+import (
+	"context"
+
+	"atsflare-agent/internal/nginx"
+	"atsflare-agent/internal/protocol"
+	"atsflare-agent/internal/state"
+)
+
+const (
+	ApplyResultSuccess = "success"
+	ApplyResultFailed  = "failed"
+)
+
+type ConfigClient interface {
+	GetActiveConfig(ctx context.Context) (*protocol.ActiveConfigResponse, error)
+	ReportApplyLog(ctx context.Context, payload protocol.ApplyLogPayload) error
+}
+
+type Service struct {
+	client       ConfigClient
+	nginxManager *nginx.Manager
+	stateStore   *state.Store
+}
+
+func New(client ConfigClient, nginxManager *nginx.Manager, stateStore *state.Store) *Service {
+	return &Service{
+		client:       client,
+		nginxManager: nginxManager,
+		stateStore:   stateStore,
+	}
+}
+
+func (s *Service) SyncOnce(ctx context.Context) error {
+	snapshot, err := s.stateStore.Load()
+	if err != nil {
+		return err
+	}
+	config, err := s.client.GetActiveConfig(ctx)
+	if err != nil {
+		return err
+	}
+	if snapshot.CurrentVersion == config.Version && snapshot.CurrentChecksum == config.Checksum {
+		return nil
+	}
+	if err = s.nginxManager.Apply(ctx, config.RenderedConfig); err != nil {
+		snapshot.LastError = err.Error()
+		_ = s.stateStore.Save(snapshot)
+		reportErr := s.client.ReportApplyLog(ctx, protocol.ApplyLogPayload{
+			NodeID:  snapshot.NodeID,
+			Version: config.Version,
+			Result:  ApplyResultFailed,
+			Message: err.Error(),
+		})
+		if reportErr != nil {
+			return reportErr
+		}
+		return err
+	}
+	snapshot.CurrentVersion = config.Version
+	snapshot.CurrentChecksum = config.Checksum
+	snapshot.LastError = ""
+	if err = s.stateStore.Save(snapshot); err != nil {
+		return err
+	}
+	return s.client.ReportApplyLog(ctx, protocol.ApplyLogPayload{
+		NodeID:  snapshot.NodeID,
+		Version: config.Version,
+		Result:  ApplyResultSuccess,
+		Message: "apply success",
+	})
+}

atsf_agent/internal/sync/service_test.go

@@ -0,0 +1,150 @@
+package sync
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"atsflare-agent/internal/nginx"
+	"atsflare-agent/internal/protocol"
+	"atsflare-agent/internal/state"
+)
+
+type fakeExecutor struct {
+	testErr   error
+	reloadErr error
+}
+
+type fakeClient struct {
+	config  protocol.ActiveConfigResponse
+	reports []protocol.ApplyLogPayload
+}
+
+func (f *fakeExecutor) Test(ctx context.Context) error {
+	return f.testErr
+}
+
+func (f *fakeExecutor) Reload(ctx context.Context) error {
+	return f.reloadErr
+}
+
+func (f *fakeClient) GetActiveConfig(ctx context.Context) (*protocol.ActiveConfigResponse, error) {
+	return &f.config, nil
+}
+
+func (f *fakeClient) ReportApplyLog(ctx context.Context, payload protocol.ApplyLogPayload) error {
+	f.reports = append(f.reports, payload)
+	return nil
+}
+
+func TestSyncOnceSuccess(t *testing.T) {
+	client := &fakeClient{
+		config: protocol.ActiveConfigResponse{
+			Version:        "20260309-001",
+			Checksum:       "checksum-1",
+			RenderedConfig: "server { listen 80; }",
+			CreatedAt:      time.Now().Format(time.RFC3339),
+		},
+	}
+
+	stateStore := state.NewStore(filepath.Join(t.TempDir(), "state.json"))
+	nodeID, err := stateStore.EnsureNodeID()
+	if err != nil {
+		t.Fatalf("EnsureNodeID failed: %v", err)
+	}
+	snapshot, _ := stateStore.Load()
+	snapshot.NodeID = nodeID
+	if err = stateStore.Save(snapshot); err != nil {
+		t.Fatalf("failed to save initial state: %v", err)
+	}
+
+	routePath := filepath.Join(t.TempDir(), "routes.conf")
+	service := New(client, &nginx.Manager{
+		RouteConfigPath: routePath,
+		Executor:        &fakeExecutor{},
+	}, stateStore)
+
+	if err = service.SyncOnce(context.Background()); err != nil {
+		t.Fatalf("SyncOnce failed: %v", err)
+	}
+
+	data, err := os.ReadFile(routePath)
+	if err != nil {
+		t.Fatalf("failed to read route config: %v", err)
+	}
+	if string(data) != "server { listen 80; }" {
+		t.Fatal("expected rendered config to be written to route file")
+	}
+	snapshot, err = stateStore.Load()
+	if err != nil {
+		t.Fatalf("failed to load state: %v", err)
+	}
+	if snapshot.CurrentVersion != "20260309-001" || snapshot.CurrentChecksum != "checksum-1" {
+		t.Fatal("expected state store to persist current version and checksum")
+	}
+	if len(client.reports) != 1 || client.reports[0].Result != ApplyResultSuccess {
+		t.Fatal("expected successful apply report to be sent")
+	}
+}
+
+func TestSyncOnceRollbackOnNginxFailure(t *testing.T) {
+	client := &fakeClient{
+		config: protocol.ActiveConfigResponse{
+			Version:        "20260309-002",
+			Checksum:       "checksum-2",
+			RenderedConfig: "server { listen 81; }",
+			CreatedAt:      time.Now().Format(time.RFC3339),
+		},
+	}
+
+	tempDir := t.TempDir()
+	routePath := filepath.Join(tempDir, "routes.conf")
+	if err := os.WriteFile(routePath, []byte("server { listen 80; }"), 0o644); err != nil {
+		t.Fatalf("failed to seed route file: %v", err)
+	}
+
+	stateStore := state.NewStore(filepath.Join(tempDir, "state.json"))
+	nodeID, err := stateStore.EnsureNodeID()
+	if err != nil {
+		t.Fatalf("EnsureNodeID failed: %v", err)
+	}
+	if err = stateStore.Save(&state.Snapshot{
+		NodeID:          nodeID,
+		CurrentVersion:  "20260309-001",
+		CurrentChecksum: "checksum-1",
+	}); err != nil {
+		t.Fatalf("failed to seed state: %v", err)
+	}
+
+	service := New(client, &nginx.Manager{
+		RouteConfigPath: routePath,
+		Executor: &fakeExecutor{
+			testErr: context.DeadlineExceeded,
+		},
+	}, stateStore)
+
+	err = service.SyncOnce(context.Background())
+	if err == nil {
+		t.Fatal("expected SyncOnce to fail when nginx test fails")
+	}
+
+	data, readErr := os.ReadFile(routePath)
+	if readErr != nil {
+		t.Fatalf("failed to read route file after rollback: %v", readErr)
+	}
+	if string(data) != "server { listen 80; }" {
+		t.Fatal("expected original route config to be restored after rollback")
+	}
+	snapshot, loadErr := stateStore.Load()
+	if loadErr != nil {
+		t.Fatalf("failed to load state: %v", loadErr)
+	}
+	if snapshot.CurrentVersion != "20260309-001" {
+		t.Fatal("expected failed sync not to overwrite current version")
+	}
+	if len(client.reports) != 1 || client.reports[0].Result != ApplyResultFailed {
+		t.Fatal("expected failed apply report to be sent")
+	}
+}