1 Commits
1.1.0 ... main

Author SHA1 Message Date
0f2fb51065 feat: 添加日志等级配置支持,优化日志初始化设置 2025-09-26 21:51:13 +08:00
3 changed files with 24 additions and 5 deletions

View File

@@ -50,7 +50,8 @@ WantedBy=multi-user.target
| 参数 | 是否可选 | 默认值 | 数据类型 | 解释 |
| -------- | -------: | :---------------: | -------- | --------------------------------- |
| -port | 是 | 8080 | int | 代理服务器监听端口 |
| -debug | 是 | false | bool | 调试模式(debug 级别日志) |
| -debug | 是 | false | bool | 调试模式(等价于未指定 -log-level 时将日志等级提升为 debug |
| -log-level | 是 | warn | string | 日志等级: debug / info / warn / error |
| -log | 是 | (输出到 stderr | string | 日志文件路径(默认输出到 stderr |
| -grace | 是 | 10 | int | 优雅停机等待秒数 |
| -timeout | 是 | 0 | int | 单次上游请求超时秒0 = 不设置) |

View File

@@ -12,6 +12,7 @@ type Config struct {
LogFile string
ShutdownGrace int // 优雅停机等待秒数
RequestTimeout int // 上游整体请求超时时间(秒)
LogLevel string // 日志等级: debug|info|warn|error
}
// Parse 解析命令行参数返回配置
@@ -22,7 +23,11 @@ func Parse() *Config {
flag.StringVar(&cfg.LogFile, "log", "", "日志文件路径 (默认输出到 stderr)")
flag.IntVar(&cfg.ShutdownGrace, "grace", 10, "优雅停机等待秒数")
flag.IntVar(&cfg.RequestTimeout, "timeout", 0, "单次上游请求超时秒(0=不设置)")
flag.StringVar(&cfg.LogLevel, "log-level", "warn", "日志等级: debug|info|warn|error (默认 warn)")
flag.Parse()
// 兼容旧的 -debug 参数: 当 -debug 为 true 且未显式指定其它日志等级(仍为默认 warn) 时,提升为 debug
if cfg.Debug && cfg.LogLevel == "warn" { cfg.LogLevel = "debug" }
return cfg
}

21
main.go
View File

@@ -7,6 +7,7 @@ import (
"net/http"
"os"
"os/signal"
"strings"
"syscall"
"time"
@@ -22,9 +23,21 @@ import (
func main() {
cfg := config.Parse()
// 日志初始化设置
// 日志初始化设置 (支持显式日志等级)
levelVar := new(slog.LevelVar)
if cfg.Debug { levelVar.Set(slog.LevelDebug) } else { levelVar.Set(slog.LevelInfo) }
lvlStr := strings.ToLower(cfg.LogLevel)
switch lvlStr {
case "debug":
levelVar.Set(slog.LevelDebug)
case "info":
levelVar.Set(slog.LevelInfo)
case "warn", "warning":
levelVar.Set(slog.LevelWarn)
case "error", "err":
levelVar.Set(slog.LevelError)
default:
levelVar.Set(slog.LevelWarn) // 回退到默认 warn
}
var writer io.Writer = os.Stderr
if cfg.LogFile != "" {
f, err := os.OpenFile(cfg.LogFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
@@ -35,7 +48,7 @@ func main() {
logger := slog.New(h)
slog.SetDefault(logger)
if cfg.Debug { gin.SetMode(gin.DebugMode) } else { gin.SetMode(gin.ReleaseMode) }
if cfg.Debug || lvlStr == "debug" { gin.SetMode(gin.DebugMode) } else { gin.SetMode(gin.ReleaseMode) }
// 可复用的 HTTP 客户端(保持连接复用)
transport := &http.Transport{Proxy: http.ProxyFromEnvironment, DisableCompression: true}
@@ -51,7 +64,7 @@ func main() {
r.Any("/proxy/*proxyPath", p.HandleProxyPath) // 处理 /proxy/*path 形式的请求
r.Any(":protocol/*remainder", p.HandleProtocol) // 处理 /:protocol/*remainder 形式的请求
logger.Info("服务器启动", "addr", cfg.Addr(), "debug", cfg.Debug, "version", version.Version, "commit", version.GitCommit)
logger.Info("服务器启动", "addr", cfg.Addr(), "debug", cfg.Debug, "log_level", lvlStr, "version", version.Version, "commit", version.GitCommit)
// 优雅停机设置:监听系统信号,执行平滑关闭
srv := &http.Server{Addr: cfg.Addr(), Handler: r}