From 075cd7ecf920de2a35741741f9c902af94bb330f Mon Sep 17 00:00:00 2001 From: "haibo.chen" <495810242@qq.com> Date: Wed, 30 Apr 2025 11:13:09 +0800 Subject: [PATCH] Add command-line flag for configuration file path in main.go --- main/main.go | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/main/main.go b/main/main.go index 9c40527..2648524 100644 --- a/main/main.go +++ b/main/main.go @@ -2,6 +2,7 @@ package main import ( "context" + "flag" "net/http" "os" "os/signal" @@ -28,9 +29,18 @@ func WaitTerminationSignal(cancel context.CancelFunc) { } func main() { + // 定义配置文件路径参数 + configPath := flag.String("c", "", "配置文件路径") + flag.Parse() + + if *configPath == "" { + logger.E(nil, "错误: 通过 -c 参数指定配置文件路径,比如:./srs-sip -c conf/config.yaml") + return + } + ctx, cancel := context.WithCancel(context.Background()) - conf, err := config.LoadConfig("conf/config.yaml") + conf, err := config.LoadConfig(*configPath) if err != nil { logger.E(nil, "load config failed: %v", err) return