From 39c3e85a647dc269540eeef29cc2b9f378be6d53 Mon Sep 17 00:00:00 2001 From: Igor Suvorov <34533360+suvorovis@users.noreply.github.com> Date: Thu, 23 Mar 2023 12:00:09 +0300 Subject: [PATCH] Added possibility to set http transport (#21) Co-authored-by: Igor Suvorov --- pkg/client/client.go | 7 ++++++- pkg/client/config.go | 2 ++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/pkg/client/client.go b/pkg/client/client.go index 45ddece..a6d3798 100644 --- 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 index fa3ed5e..dfe7f8f 100644 --- 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() {