custom HTTPClient for DigestTransport

Author: nitper

digest_auth_client.go

@@ -22,9 +22,9 @@ type DigestRequest struct {
 }
 
 type DigestTransport struct {
-	Password string
-	Username string
-	Timeout  time.Duration
+	Password   string
+	Username   string
+	HTTPClient *http.Client
 }
 
 // NewRequest creates a new DigestRequest object
@@ -40,7 +40,6 @@ func NewTransport(username, password string) DigestTransport {
 	dt := DigestTransport{}
 	dt.Password = password
 	dt.Username = username
-	dt.Timeout = 30 * time.Second
 	return dt
 }
 
@@ -88,6 +87,10 @@ func (dt *DigestTransport) RoundTrip(req *http.Request) (resp *http.Response, er
 	}
 
 	dr := NewRequest(username, password, method, uri, body)
+	if dt.HTTPClient != nil {
+		dr.HTTPClient = dt.HTTPClient
+	}
+
 	return dr.Execute()
 }