Try to support multiple custom headers.

cmd/cassowary/cli.go

@@ -132,11 +132,14 @@ func validateCLI(c *cli.Context) error {
 		prometheusEnabled = true
 	}
 
-	if c.String("header") != "" {
-		length := 0
-		length, header = client.SplitHeader(c.String("header"))
-		if length != 2 {
-			return errNotValidHeader
+	if len(c.StringSlice("header")) != 0 {
+		allHeaders := c.StringSlice("header")
+		for _, hdr := range allHeaders {
+			thisLen, thisHdr := client.SplitHeader(hdr)
+			if thisLen != 2 {
+				return errNotValidHeader
+			}
+			header = append(header, thisHdr...)
 		}
 	}
 
@@ -279,7 +282,7 @@ func runCLI(args []string) {
 					Aliases: []string{"prompushgwurl"},
 					Usage:   "specify prometheus push gateway url to send metrics (optional)",
 				},
-				&cli.StringFlag{
+				&cli.StringSliceFlag{
 					Name:    "H",
 					Aliases: []string{"header"},
 					Usage:   "add arbitrary header, eg. 'Host: www.example.com'",

pkg/client/load.go

@@ -68,8 +68,13 @@ func (c *Cassowary) runLoadTest(outPutChan chan<- durationMetrics, workerChan ch
 			}
 		}
 
-		if len(c.RequestHeader) == 2 {
-			request.Header.Add(c.RequestHeader[0], c.RequestHeader[1])
+		if len(c.RequestHeader)%2 == 0 {
+			for idx := range c.RequestHeader {
+				if idx%2 == 1 {
+					continue
+				}
+				request.Header.Add(c.RequestHeader[idx], c.RequestHeader[idx+1])
+			}
 		}
 
 		var t0, t1, t2, t3, t4, t5, t6 time.Time