[优化] 改名

2026-10-04 23:16:37 +08:00 · 2026-03-15 16:04:54 +08:00
parent d68773c554
commit 32d90ba641
304 changed files with 629 additions and 969 deletions
@@ -0,0 +1,113 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"log/slog"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"openflare-agent/internal/agent"
+	"openflare-agent/internal/config"
+	"openflare-agent/internal/heartbeat"
+	"openflare-agent/internal/httpclient"
+	"openflare-agent/internal/logging"
+	"openflare-agent/internal/nginx"
+	"openflare-agent/internal/state"
+	syncservice "openflare-agent/internal/sync"
+	"openflare-agent/internal/updater"
+)
+
+func main() {
+	logging.Setup()
+
+	configPath := flag.String("config", "./agent.json", "agent config path")
+	flag.Parse()
+
+	cfg, err := config.Load(*configPath)
+	if err != nil {
+		slog.Error("load agent config failed", "error", err)
+		os.Exit(1)
+	}
+	cfg.NginxVersion = nginx.DetectVersion(
+		context.Background(),
+		nginx.ExecutorOptions{
+			NginxPath:                  cfg.OpenrestyPath,
+			DockerBinary:               cfg.DockerBinary,
+			ContainerName:              cfg.OpenrestyContainerName,
+			Image:                      cfg.OpenrestyDockerImage,
+			MainConfigPath:             cfg.MainConfigPath,
+			RouteConfigPath:            cfg.RouteConfigPath,
+			CertDir:                    cfg.CertDir,
+			NginxCertDir:               cfg.OpenrestyCertDir,
+			LuaDir:                     cfg.LuaDir,
+			NginxLuaDir:                cfg.OpenrestyLuaDir,
+			OpenrestyObservabilityPort: cfg.OpenrestyObservabilityPort,
+		},
+	)
+	slog.Info("agent config loaded",
+		"server", cfg.ServerURL,
+		"node", cfg.NodeName,
+		"ip", cfg.NodeIP,
+		"heartbeat_interval", cfg.HeartbeatInterval,
+		"route_config", cfg.RouteConfigPath,
+		"cert_dir", cfg.CertDir,
+		"lua_dir", cfg.LuaDir,
+	)
+
+	client := httpclient.New(cfg.ServerURL, cfg.InitialAuthToken(), cfg.RequestTimeout.Duration())
+	stateStore := state.NewStore(cfg.StatePath)
+	observabilityBuffer := state.NewObservabilityBufferStore(cfg.ObservabilityBufferPath)
+	runtimeRouteConfigPath := cfg.RouteConfigPath
+	if cfg.OpenrestyPath == "" {
+		runtimeRouteConfigPath = nginx.DockerRouteConfigPath
+	}
+	runtimeManager := &nginx.Manager{
+		MainConfigPath:               cfg.MainConfigPath,
+		RouteConfigPath:              cfg.RouteConfigPath,
+		RuntimeRouteConfigPath:       runtimeRouteConfigPath,
+		CertDir:                      cfg.CertDir,
+		NginxCertDir:                 cfg.OpenrestyCertDir,
+		LuaDir:                       cfg.LuaDir,
+		NginxLuaDir:                  cfg.OpenrestyLuaDir,
+		OpenrestyObservabilityListen: nginx.ObservabilityListenAddress(cfg.OpenrestyPath, cfg.OpenrestyObservabilityPort),
+		OpenrestyObservabilityPort:   cfg.OpenrestyObservabilityPort,
+		Executor: nginx.NewExecutor(nginx.ExecutorOptions{
+			NginxPath:                  cfg.OpenrestyPath,
+			DockerBinary:               cfg.DockerBinary,
+			ContainerName:              cfg.OpenrestyContainerName,
+			Image:                      cfg.OpenrestyDockerImage,
+			MainConfigPath:             cfg.MainConfigPath,
+			RouteConfigPath:            cfg.RouteConfigPath,
+			CertDir:                    cfg.CertDir,
+			NginxCertDir:               cfg.OpenrestyCertDir,
+			LuaDir:                     cfg.LuaDir,
+			NginxLuaDir:                cfg.OpenrestyLuaDir,
+			OpenrestyObservabilityPort: cfg.OpenrestyObservabilityPort,
+		}),
+	}
+	if err = runtimeManager.EnsureLuaAssets(); err != nil {
+		slog.Error("ensure managed lua assets failed", "error", err)
+		os.Exit(1)
+	}
+	runner := &agent.Runner{
+		Config:              cfg,
+		StateStore:          stateStore,
+		ObservabilityBuffer: observabilityBuffer,
+		HeartbeatService:    heartbeat.New(client),
+		SyncService:         syncservice.New(client, runtimeManager, stateStore),
+		Updater:             updater.New(),
+		RuntimeManager:      runtimeManager,
+	}
+
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+	slog.Info("agent process started")
+
+	if err = runner.Run(ctx); err != nil && err != context.Canceled {
+		slog.Error("agent process exited with error", "error", err)
+		os.Exit(1)
+	}
+	slog.Info("agent process stopped")
+}