Pass request context to proxyDialer

bdwyertech · Oct 29, 2022 · bbf4078 · bbf4078
1 parent d02fd35
commit bbf4078
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/pkg/gontlm-proxy.go b/pkg/gontlm-proxy.go
@@ -174,11 +174,8 @@ func Run() {
 	}
 
 	// HTTPS
-	proxy.ConnectDial = func(network, addr string) (net.Conn, error) {
-		ctx, cancel := context.WithCancel(context.Background())
-		defer cancel()
-
-		return proxyDialer("https", addr, proxyUrl)(ctx, network, addr)
+	proxy.ConnectDialWithReq = func(req *http.Request, network, addr string) (net.Conn, error) {
+		return proxyDialer("https", addr, proxyUrl)(req.Context(), network, addr)
 	}
 
 	//