Enhanced configuration of CLI ARG

sibalma · Sep 5, 2023 · 8055fd0 · 8055fd0
1 parent 0ebb4f2
commit 8055fd0
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/relay.go b/relay.go
@@ -307,14 +307,10 @@ func Copy(src io.Reader, dst io.Writer) {
 }
 
 func main() {
-	var ip string
-	var port string
-	flag.StringVar(&ip, "b", "0.0.0.0", "Server IP address")
-	flag.StringVar(&port, "p", "6666", "Server Port number")
+	var config Config
+	flag.StringVar(&config.Host, "b", "0.0.0.0", "Server IP address")
+	flag.StringVar(&config.Port, "p", "6666", "Server Port number")
 	flag.Parse()
-	server := New(&Config{
-		Host: ip,
-		Port: port,
-	})
+	server := New(&config)
 	server.Run()
 }