Error out instead of crashing when server isn't specified

fungaren · Oct 5, 2020 · e0436f2 · e0436f2
1 parent d5eacb2
commit e0436f2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -128,6 +128,10 @@ func metricName(metric map[string]string) string {
 func main() {
 	plotutil.DefaultDashes = [][]vg.Length{{}}
 
+	if len(os.Args) < 2 {
+		fmt.Printf("Usage: %s server\n", os.Args[0])
+		os.Exit(1)
+	}
 	upstream = os.Args[1]
 	router := chi.NewRouter()
 	router.Use(middleware.RealIP)