first commit

cmd/agent/main.go

@@ -0,0 +1,75 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"linkmaster-node/internal/config"
+	"linkmaster-node/internal/heartbeat"
+	"linkmaster-node/internal/recovery"
+	"linkmaster-node/internal/server"
+
+	"go.uber.org/zap"
+)
+
+func main() {
+	// 加载配置
+	cfg, err := config.Load()
+	if err != nil {
+		fmt.Printf("加载配置失败: %v\n", err)
+		os.Exit(1)
+	}
+
+	// 初始化日志
+	logger, err := initLogger(cfg)
+	if err != nil {
+		fmt.Printf("初始化日志失败: %v\n", err)
+		os.Exit(1)
+	}
+	defer logger.Sync()
+
+	logger.Info("节点服务启动", zap.String("version", "1.0.0"))
+
+	// 初始化错误恢复
+	recovery.Init()
+
+	// 启动心跳上报
+	heartbeatReporter := heartbeat.NewReporter(cfg)
+	go heartbeatReporter.Start(context.Background())
+
+	// 启动HTTP服务器
+	httpServer := server.NewHTTPServer(cfg)
+	go func() {
+		if err := httpServer.Start(); err != nil {
+			logger.Fatal("HTTP服务器启动失败", zap.Error(err))
+		}
+	}()
+
+	// 等待中断信号
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+	<-sigChan
+
+	logger.Info("收到停止信号，正在关闭服务...")
+
+	// 优雅关闭
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	httpServer.Shutdown(ctx)
+	heartbeatReporter.Stop()
+
+	logger.Info("服务已关闭")
+}
+
+func initLogger(cfg *config.Config) (*zap.Logger, error) {
+	if cfg.Debug {
+		return zap.NewDevelopment()
+	}
+	return zap.NewProduction()
+}
+