xfy
f3e5aad21e
- 添加 SIGTERM/SIGINT/SIGQUIT 信号处理和优雅停止逻辑 - 完善安全配置：访问控制、速率限制、认证参数 - 完善性能配置：Goroutine 池、连接限制 - 改进配置生成 YAML 格式和注释说明 - 添加编译产物 lolly 到 gitignore Co-Authored-By: Claude <noreply@anthropic.com>
142 lines
3.4 KiB
Go
142 lines
3.4 KiB
Go
// Package app 提供应用程序的启动和运行逻辑。
|
||
package app
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
"time"
|
||
|
||
"rua.plus/lolly/internal/config"
|
||
)
|
||
|
||
// 版本信息，通过 -ldflags 注入。
|
||
var (
|
||
Version = "dev"
|
||
GitCommit = "unknown"
|
||
GitBranch = "unknown"
|
||
BuildTime = "unknown"
|
||
GoVersion = "unknown"
|
||
BuildPlatform = "unknown"
|
||
)
|
||
|
||
// 应用状态。
|
||
var (
|
||
shutdownTimeout = 30 * time.Second // 优雅停止超时时间
|
||
)
|
||
|
||
// Run 应用程序入口。
|
||
func Run(cfgPath string, genConfig bool, outputPath string, showVersion bool) int {
|
||
if genConfig {
|
||
return generateConfig(outputPath)
|
||
}
|
||
|
||
if showVersion {
|
||
printVersion()
|
||
return 0
|
||
}
|
||
|
||
return startServer(cfgPath)
|
||
}
|
||
|
||
// generateConfig 生成默认配置文件。
|
||
func generateConfig(outputPath string) int {
|
||
cfg := config.DefaultConfig()
|
||
yamlData, err := config.GenerateConfigYAML(cfg)
|
||
if err != nil {
|
||
fmt.Fprintf(os.Stderr, "生成配置失败: %v\n", err)
|
||
return 1
|
||
}
|
||
|
||
if outputPath == "" {
|
||
fmt.Print(string(yamlData))
|
||
} else {
|
||
if err := os.WriteFile(outputPath, yamlData, 0644); err != nil {
|
||
fmt.Fprintf(os.Stderr, "写入文件失败: %v\n", err)
|
||
return 1
|
||
}
|
||
fmt.Printf("配置已写入: %s\n", outputPath)
|
||
}
|
||
return 0
|
||
}
|
||
|
||
// printVersion 打印版本信息。
|
||
func printVersion() {
|
||
fmt.Printf("lolly version %s\n", Version)
|
||
fmt.Printf(" Git: %s (%s)\n", GitCommit, GitBranch)
|
||
fmt.Printf(" Built: %s\n", BuildTime)
|
||
fmt.Printf(" Go: %s\n", GoVersion)
|
||
fmt.Printf(" Platform: %s\n", BuildPlatform)
|
||
}
|
||
|
||
// startServer 启动服务器。
|
||
func startServer(cfgPath string) int {
|
||
cfg, err := config.Load(cfgPath)
|
||
if err != nil {
|
||
fmt.Fprintf(os.Stderr, "加载配置失败: %v\n", err)
|
||
return 1
|
||
}
|
||
|
||
fmt.Printf("配置加载成功: %s\n", cfgPath)
|
||
fmt.Printf("监听地址: %s\n", cfg.Server.Listen)
|
||
|
||
// 创建取消上下文，用于优雅停止（Phase 2 使用）。
|
||
_, cancel := context.WithCancel(context.Background())
|
||
defer cancel()
|
||
|
||
// 启动信号监听。
|
||
sigChan := make(chan os.Signal, 1)
|
||
signal.Notify(sigChan,
|
||
syscall.SIGTERM, // 快速停止（kill 或 systemd stop）
|
||
syscall.SIGINT, // 快速停止（Ctrl+C）
|
||
syscall.SIGQUIT, // 优雅停止
|
||
)
|
||
|
||
// TODO: 启动 HTTP 服务器（Phase 2 实现）
|
||
// server := server.New(cfg)
|
||
// go server.Start()
|
||
|
||
fmt.Println("服务器启动中...（HTTP 服务器待 Phase 2 实现）")
|
||
|
||
// 等待信号。
|
||
sig := waitSignal(sigChan)
|
||
|
||
// 根据信号类型决定停止方式。
|
||
switch sig {
|
||
case syscall.SIGQUIT:
|
||
// 优雅停止：等待请求完成。
|
||
fmt.Printf("\n收到 SIGQUIT，优雅停止（等待 %v）...\n", shutdownTimeout)
|
||
// TODO: server.GracefulStop(shutdownTimeout)
|
||
time.Sleep(100 * time.Millisecond) // 模拟等待（Phase 2 实现真实逻辑）
|
||
case syscall.SIGTERM, syscall.SIGINT:
|
||
// 快速停止。
|
||
fmt.Printf("\n收到 %v，停止服务器...\n", sigName(sig))
|
||
// TODO: server.Stop()
|
||
}
|
||
|
||
cancel()
|
||
fmt.Println("服务器已停止")
|
||
return 0
|
||
}
|
||
|
||
// waitSignal 等待信号，返回接收到的信号。
|
||
func waitSignal(ch <-chan os.Signal) syscall.Signal {
|
||
sig := <-ch
|
||
return sig.(syscall.Signal)
|
||
}
|
||
|
||
// sigName 返回信号名称（用于日志输出）。
|
||
func sigName(sig syscall.Signal) string {
|
||
switch sig {
|
||
case syscall.SIGTERM:
|
||
return "SIGTERM"
|
||
case syscall.SIGINT:
|
||
return "SIGINT"
|
||
case syscall.SIGQUIT:
|
||
return "SIGQUIT"
|
||
default:
|
||
return fmt.Sprintf("Signal(%d)", sig)
|
||
}
|
||
} |