feat: allow changing the listening port

Allow changing the listening port via command line or configuration file. Signed-off-by: cubercsl <[email protected]>
stulzq · Oct 7, 2023 · 3208f42 · 3208f42
1 parent c40e27c
commit 3208f42
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cmd/main.go b/cmd/main.go
@@ -36,7 +36,7 @@ func main() {
 	registerRoute(r)
 
 	srv := &http.Server{
-		Addr:    ":8080",
+		Addr:    viper.GetString("listen"),
 		Handler: r,
 	}
 
@@ -64,6 +64,7 @@ func runServer(srv *http.Server) {
 
 func parseFlag() {
 	pflag.StringP("configFile", "c", "config.yaml", "config file")
+	pflag.StringP("listen", "l", ":8080", "listen address")
 	pflag.BoolP("version", "v", false, "version information")
 	pflag.Parse()
 	if err := viper.BindPFlags(pflag.CommandLine); err != nil {