Load tool (#1421)

Simple tool to put a specific load on clio.

tools/requests_gun/internal/parse_args/parse_args.go

@@ -0,0 +1,40 @@
+package parse_args
+
+import (
+	"fmt"
+
+	flag "github.com/spf13/pflag"
+)
+
+type CliArgs struct {
+	Url         string
+	Port        uint
+	TargetLoad  uint
+	Ammo        string
+	PrintErrors bool
+	Help        bool
+}
+
+func Parse() (*CliArgs, error) {
+	flag.Usage = PrintUsage
+	url := flag.StringP("url", "u", "localhost", "URL to send the request to")
+	port := flag.UintP("port", "p", 51233, "Port to send the request to")
+	target_load := flag.UintP("load", "l", 100, "Target requests per second load")
+	print_errors := flag.BoolP("print-errors", "e", false, "Print errors")
+	help := flag.BoolP("help", "h", false, "Print help message")
+
+	flag.Parse()
+
+	if flag.NArg() == 0 {
+		return nil, fmt.Errorf("No ammo file provided")
+	}
+
+	return &CliArgs{*url, *port, *target_load, flag.Arg(0), *print_errors, *help}, nil
+}
+
+func PrintUsage() {
+	fmt.Println("Usage: requests_gun [options] <ammo_file>")
+	fmt.Println("  <ammo_file>        Path to file with requests to use")
+	fmt.Println("Options:")
+	flag.PrintDefaults()
+}