diff --git a/agent/main.go b/agent/main.go index 416cbed..984f355 100644 --- a/agent/main.go +++ b/agent/main.go @@ -49,15 +49,21 @@ type Config struct { } func main() { - cfg := Config{ - Server: *flag.String("server", "", "AIOps 后端上报地址(output/discover 模式不需要)"), - Token: *flag.String("token", "", "Bearer 鉴权令牌"), - Hostname: *flag.String("hostname", "", "上报主机名,默认使用系统主机名"), - Interval: *flag.Int("interval", 10, "采集与上报间隔,单位:秒"), - TopN: *flag.Int("top-n", 3, "进程 TOP N 数量"), - } + serverFlag := flag.String("server", "", "AIOps 后端上报地址(output/discover 模式不需要)") + tokenFlag := flag.String("token", "", "Bearer 鉴权令牌") + hostnameFlag := flag.String("hostname", "", "上报主机名,默认使用系统主机名") + intervalFlag := flag.Int("interval", 10, "采集与上报间隔,单位:秒") + topNFlag := flag.Int("top-n", 3, "进程 TOP N 数量") flag.Parse() + cfg := Config{ + Server: *serverFlag, + Token: *tokenFlag, + Hostname: *hostnameFlag, + Interval: *intervalFlag, + TopN: *topNFlag, + } + if cfg.Hostname == "" { if h, err := os.Hostname(); err == nil { cfg.Hostname = h