feat(http/driver): added ConfigureTransport

bundleman · Oct 7, 2023 · b7af4a9 · b7af4a9
1 parent 36d1210
commit b7af4a9
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/pkg/drivers/http/driver.go b/pkg/drivers/http/driver.go
@@ -6,6 +6,7 @@ import (
 	"compress/flate"
 	"compress/gzip"
 	"context"
+	"golang.org/x/net/http2"
 	"io"
 	"io/ioutil"
 	"net/http"
@@ -81,9 +82,14 @@ func addProxy(httpClient *pester.Client, proxyStr string) error {
 	proxy := http.ProxyURL(proxyURL)
 
 	if httpClient.Transport != nil {
-		transport, isHTTPTransport := httpClient.Transport.(*http.Transport)
-		if isHTTPTransport {
-			transport.Proxy = proxy
+		if _, isHTTPTransport := httpClient.Transport.(*http.Transport); isHTTPTransport {
+			httpClient.Transport.(*http.Transport).Proxy = proxy
+		}
+
+		if _, isHTTPTransport := httpClient.Transport.(*http2.Transport); isHTTPTransport {
+			httpTr1 := &http.Transport{}
+			httpTr1.Proxy = proxy
+			http2.ConfigureTransport(httpTr1)
 		}
 
 		return nil