35 lines
1.2 KiB
Go
35 lines
1.2 KiB
Go
package config
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
)
|
|
|
|
// Config 保存程序配置
|
|
type Config struct {
|
|
Port int
|
|
Debug bool
|
|
LogFile string
|
|
ShutdownGrace int // 优雅停机等待秒数
|
|
RequestTimeout int // 上游整体请求超时时间（秒）
|
|
LogLevel string // 日志等级: debug|info|warn|error
|
|
}
|
|
|
|
// Parse 解析命令行参数返回配置
|
|
func Parse() *Config {
|
|
cfg := &Config{}
|
|
flag.IntVar(&cfg.Port, "port", 8080, "代理服务器监听端口")
|
|
flag.BoolVar(&cfg.Debug, "debug", false, "调试模式 (debug level log)")
|
|
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
|
|
}
|
|
|
|
func (c *Config) Addr() string { return fmt.Sprintf(":%d", c.Port) }
|