Added possibility to set http transport (#21)

Co-authored-by: Igor Suvorov <[email protected]>
banzaicloud · Mar 23, 2023 · 39c3e85 · 39c3e85
1 parent a3f54dd
commit 39c3e85
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pkg/client/client.go b/pkg/client/client.go
@@ -137,8 +137,13 @@ func NewClient(opts *Config) (*Client, error) {
 	// 	Jar:       jar,
 	// 	Timeout:   0,
 	// }
+	transport := http.DefaultTransport
+	if opts.Transport != nil {
+		transport = opts.Transport
+	}
+
 	client.httpClient = &http.Client{
-		Transport: http.DefaultTransport,
+		Transport: transport,
 	}
 
 	serverURL := DefaultServerURL

diff --git a/pkg/client/config.go b/pkg/client/config.go
@@ -17,6 +17,7 @@ limitations under the License.
 package client
 
 import (
+	"net/http"
 	"os"
 )
 
@@ -38,6 +39,7 @@ type Config struct {
 	Password    string
 	AccessToken string
 	UserAgent   string
+	Transport   http.RoundTripper
 }
 
 func (c *Config) ReadFromEnvironment() {