Using the user-agent informed for the user

goadapp · Oct 1, 2016 · 302d3c6 · 302d3c6
1 parent cd2fa1f
commit 302d3c6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lambda/lambda.go b/lambda/lambda.go
@@ -228,7 +228,6 @@ func fetch(loadTestStartTime time.Time, client *http.Client, address string, req
 			fmt.Println("Error creating the HTTP request:", err)
 			return
 		}
-		req.Header.Add("User-Agent", "Mozilla/5.0 (compatible; Goad/1.0; +https://goad.io)")
 		req.Header.Add("Accept-Encoding", "gzip")
 		for _, v := range requestHeaders {
 			header := strings.Split(v, ":")
@@ -239,6 +238,10 @@ func fetch(loadTestStartTime time.Time, client *http.Client, address string, req
 			}
 		}
 
+		if req.Header.Get("User-Agent") == "" {
+			req.Header.Add("User-Agent", "Mozilla/5.0 (compatible; Goad/1.0; +https://goad.io)")
+		}
+
 		response, err := client.Do(req)
 		var status string
 		var elapsedFirstByte time.Duration