From a1aa05406c2857cbe0d61d9e9c149ec28fd04c11 Mon Sep 17 00:00:00 2001 From: xfy Date: Fri, 3 Apr 2026 15:02:34 +0800 Subject: [PATCH] =?UTF-8?q?refactor(app):=20=E9=9B=86=E6=88=90=20AppLogger?= =?UTF-8?q?=20=E7=BB=9F=E4=B8=80=E7=AE=A1=E7=90=86=E5=90=AF=E5=8A=A8/?= =?UTF-8?q?=E5=81=9C=E6=AD=A2/=E4=BF=A1=E5=8F=B7=E6=97=A5=E5=BF=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 使用 AppLogger 替代 fmt.Print/Fprintf 输出启动停止消息 - 支持 text/json 双格式输出，json 格式输出结构化日志 - 信号处理、配置重载、热升级等流程统一使用 AppLogger - 重载配置时重建 AppLogger 以应用新格式配置 Co-Authored-By: Claude Opus 4.6 --- internal/app/app.go | 83 +++++++++++++++++++++++---------------------- 1 file changed, 43 insertions(+), 40 deletions(-) diff --git a/internal/app/app.go b/internal/app/app.go index 1c1c613..29b3a5b 100644 --- a/internal/app/app.go +++ b/internal/app/app.go @@ -81,6 +81,9 @@ type App struct { // listeners 继承的监听器（热升级时使用） listeners []net.Listener + + // logger 应用日志管理器 + logger *logging.AppLogger } // NewApp 创建应用程序。 @@ -147,9 +150,18 @@ func printVersion() { // Run 启动应用程序。 func (a *App) Run() int { + // 加载配置 + cfg, err := config.Load(a.cfgPath) + if err != nil { + fmt.Fprintf(os.Stderr, "加载配置失败: %v\n", err) + return 1 + } + a.cfg = cfg + a.logger = logging.NewAppLogger(&cfg.Logging) + // 检查是否是子进程（热升级） if os.Getenv("GRACEFUL_UPGRADE") == "1" { - fmt.Println("检测到热升级模式，继承父进程监听器") + a.logger.LogStartup("检测到热升级模式，继承父进程监听器", nil) // 创建升级管理器以获取继承的监听器 a.upgradeMgr = server.NewUpgradeManager(nil) listeners, err := a.upgradeMgr.GetInheritedListeners() @@ -159,18 +171,11 @@ func (a *App) Run() int { } } - cfg, err := config.Load(a.cfgPath) - if err != nil { - fmt.Fprintf(os.Stderr, "加载配置失败: %v\n", err) - return 1 - } - a.cfg = cfg - - fmt.Printf("配置加载成功: %s\n", a.cfgPath) - fmt.Printf("监听地址: %s\n", cfg.Server.Listen) + a.logger.LogStartup("配置加载成功", map[string]string{"config_path": a.cfgPath}) + a.logger.LogStartup("监听地址", map[string]string{"listen": a.cfg.Server.Listen}) // 创建 HTTP 服务器 - a.srv = server.New(cfg) + a.srv = server.New(a.cfg) // 如果有继承的监听器，设置到服务器 if len(a.listeners) > 0 { @@ -178,9 +183,9 @@ func (a *App) Run() int { } // 创建 Stream 服务器（如果配置了） - if len(cfg.Stream) > 0 { + if len(a.cfg.Stream) > 0 { a.streamSrv = stream.NewServer() - for _, sc := range cfg.Stream { + for _, sc := range a.cfg.Stream { // 转换目标配置 targets := make([]stream.TargetSpec, len(sc.Upstream.Targets)) for i, t := range sc.Upstream.Targets { @@ -192,26 +197,26 @@ func (a *App) Run() int { // 添加上游配置 if err := a.streamSrv.AddUpstream(sc.Listen, targets, sc.Upstream.LoadBalance, stream.HealthCheckSpec{}); err != nil { - fmt.Fprintf(os.Stderr, "添加 Stream 上游失败: %v\n", err) + a.logger.Error().Err(err).Msg("添加 Stream 上游失败") } // 监听端口 if sc.Protocol == "udp" { if err := a.streamSrv.ListenUDP(sc.Listen, sc.Listen, 60*time.Second); err != nil { - fmt.Fprintf(os.Stderr, "监听 UDP %s 失败: %v\n", sc.Listen, err) + a.logger.Error().Err(err).Str("listen", sc.Listen).Msg("监听 UDP 失败") } } else { if err := a.streamSrv.ListenTCP(sc.Listen); err != nil { - fmt.Fprintf(os.Stderr, "监听 TCP %s 失败: %v\n", sc.Listen, err) + a.logger.Error().Err(err).Str("listen", sc.Listen).Msg("监听 TCP 失败") } } } // 启动 Stream 服务器 go func() { - fmt.Println("Stream 服务器启动中...") + a.logger.LogStartup("Stream 服务器启动中", nil) if err := a.streamSrv.Start(); err != nil { - fmt.Fprintf(os.Stderr, "Stream 服务器启动失败: %v\n", err) + a.logger.Error().Err(err).Msg("Stream 服务器启动失败") } }() } @@ -230,7 +235,7 @@ func (a *App) Run() int { // 启动 HTTP 服务器 errChan := make(chan error, 1) go func() { - fmt.Println("HTTP 服务器启动中...") + a.logger.LogStartup("HTTP 服务器启动中", nil) if err := a.srv.Start(); err != nil { errChan <- err } @@ -240,12 +245,12 @@ func (a *App) Run() int { for { select { case err := <-errChan: - fmt.Fprintf(os.Stderr, "服务器启动失败: %v\n", err) + a.logger.Error().Err(err).Msg("服务器启动失败") return 1 case sig := <-sigChan: if !a.handleSignal(sig) { // 返回 false 表示退出 - fmt.Println("服务器已停止") + a.logger.LogShutdown("服务器已停止") return 0 } // 返回 true 表示继续运行（如重载配置） @@ -270,36 +275,36 @@ func (a *App) handleSignal(sig os.Signal) bool { switch sig { case syscall.SIGQUIT: // 优雅停止：等待请求完成 - fmt.Printf("\n收到 SIGQUIT，优雅停止（等待 %v）...\n", shutdownTimeout) + a.logger.LogSignal("SIGQUIT", fmt.Sprintf("优雅停止（等待 %v）", shutdownTimeout)) a.srv.GracefulStop(shutdownTimeout) return false case syscall.SIGTERM, syscall.SIGINT: // 快速停止 - fmt.Printf("\n收到 %v，停止服务器...\n", sigName(sig.(syscall.Signal))) + a.logger.LogSignal(sigName(sig.(syscall.Signal)), "停止服务器") a.srv.Stop() return false case syscall.SIGHUP: // 重载配置 - fmt.Println("\n收到 SIGHUP，重载配置...") + a.logger.LogSignal("SIGHUP", "重载配置") a.reloadConfig() return true case syscall.SIGUSR1: // 重新打开日志 - fmt.Println("\n收到 SIGUSR1，重新打开日志...") + a.logger.LogSignal("SIGUSR1", "重新打开日志") a.reopenLogs() return true case syscall.SIGUSR2: // 热升级 - fmt.Println("\n收到 SIGUSR2，执行热升级...") + a.logger.LogSignal("SIGUSR2", "执行热升级") a.gracefulUpgrade() return true default: - fmt.Printf("\n收到未知信号: %v\n", sig) + a.logger.Info().Str("signal", sig.String()).Msg("收到未知信号") return true } } @@ -308,17 +313,14 @@ func (a *App) handleSignal(sig os.Signal) bool { func (a *App) reloadConfig() { newCfg, err := config.Load(a.cfgPath) if err != nil { - fmt.Fprintf(os.Stderr, "重载配置失败: %v\n", err) + a.logger.Error().Err(err).Msg("重载配置失败") return } // 更新配置 a.cfg = newCfg - fmt.Println("配置重载成功") - - // 注意：当前实现不重启服务器，仅更新配置 - // 如需应用新配置，需要重启服务器或实现热更新 - fmt.Println("配置已重新加载") + a.logger = logging.NewAppLogger(&newCfg.Logging) + a.logger.LogStartup("配置重载成功", nil) } // reopenLogs 重新打开日志文件。 @@ -326,8 +328,9 @@ func (a *App) reopenLogs() { // 重新初始化日志系统 if a.cfg != nil { logging.Init(a.cfg.Logging.Error.Level, false) + a.logger = logging.NewAppLogger(&a.cfg.Logging) } - fmt.Println("日志已重新打开") + a.logger.LogStartup("日志已重新打开", nil) } // gracefulUpgrade 执行热升级。 @@ -335,15 +338,15 @@ func (a *App) gracefulUpgrade() { // 获取当前可执行文件路径 execPath, err := os.Executable() if err != nil { - fmt.Fprintf(os.Stderr, "获取可执行文件路径失败: %v\n", err) + a.logger.Error().Err(err).Msg("获取可执行文件路径失败") return } // 尝试从服务器获取监听器 listeners := a.srv.GetListeners() if len(listeners) == 0 { - fmt.Fprintf(os.Stderr, "热升级失败: 服务器未保存监听器（热升级当前未完全实现）\n") - fmt.Fprintf(os.Stderr, "提示: 热升级需要服务器使用手动监听器管理模式\n") + a.logger.Error().Msg("热升级失败: 服务器未保存监听器（热升级当前未完全实现）") + a.logger.Info().Msg("提示: 热升级需要服务器使用手动监听器管理模式") return } @@ -352,11 +355,11 @@ func (a *App) gracefulUpgrade() { // 执行升级 if err := a.upgradeMgr.GracefulUpgrade(execPath); err != nil { - fmt.Fprintf(os.Stderr, "热升级失败: %v\n", err) + a.logger.Error().Err(err).Msg("热升级失败") return } - fmt.Println("热升级已启动，新进程正在接管") + a.logger.LogStartup("热升级已启动，新进程正在接管", nil) // 当前进程优雅停止 a.srv.GracefulStop(shutdownTimeout) @@ -380,4 +383,4 @@ func sigName(sig syscall.Signal) string { default: return fmt.Sprintf("Signal(%d)", sig) } -} +} \ No newline at end of file