Add ability to set a custom http client on the client

fullcontact · May 23, 2023 · bce5080 · bce5080
1 parent 250914c
commit bce5080
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/fc/client.go b/fc/client.go
@@ -39,8 +39,11 @@ func NewFullContactClient(options ...ClientOption) (*fullContactClient, error) {
 	if c.retryHandler == nil {
 		c.retryHandler = &DefaultRetryHandler{}
 	}
-	c.httpClient = &http.Client{
-		Timeout: time.Duration(c.connectTimeoutMillis) * time.Millisecond,
+
+	if c.httpClient == nil {
+		c.httpClient = &http.Client{
+			Timeout: time.Duration(c.connectTimeoutMillis) * time.Millisecond,
+		}
 	}
 
 	return c, nil
@@ -71,3 +74,9 @@ func WithRetryHandler(retryHandler RetryHandler) ClientOption {
 		fc.retryHandler = retryHandler
 	}
 }
+
+func WithHTTPClient(httpClient *http.Client) ClientOption {
+	return func(fc *fullContactClient) {
+		fc.httpClient = httpClient
+	}
+}