From deeb1c799e4ac87d8f4478c5b2110f636d7ea188 Mon Sep 17 00:00:00 2001
From: Mohammad Yosefpor <myusefpur@gmail.com>
Date: Tue, 28 Dec 2021 15:52:23 +0330
Subject: [PATCH] cleanup

---
 prober/http.go | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/prober/http.go b/prober/http.go
index d9d1192..76c5764 100644
--- a/prober/http.go
+++ b/prober/http.go
@@ -53,9 +53,7 @@ func NewHttp(name string, url string, rps float64, timeout time.Duration, tlsSki
 		Timeout: timeout,
 	}
 	customTransport := http.DefaultTransport.(*http.Transport).Clone()
-	if disableKeepAlives {
-		customTransport.DisableKeepAlives = true
-	}
+	customTransport.DisableKeepAlives = disableKeepAlives
 	if tlsSkipVerify {
 		customTransport.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
 	}
@@ -161,9 +159,8 @@ func (h *HTTP) sendRequest(ctx context.Context) HTTPResult {
 		},
 	}
 
-	req, _ := http.NewRequestWithContext(ctx, http.MethodGet, h.URL, nil)
-	clientTraceCtx := httptrace.WithClientTrace(req.Context(), httpTrace)
-	req = req.WithContext(clientTraceCtx)
+	clientTraceCtx := httptrace.WithClientTrace(ctx, httpTrace)
+	req, _ := http.NewRequestWithContext(clientTraceCtx, http.MethodGet, h.URL, nil)
 	if h.Host != "" {
 		req.Host = h.Host
 	}