Merge pull request #11 from webconnex/custom-http-client

Add ability to use a custom http client
fullcontact · Jun 29, 2023 · c97bce3 · c97bce3
2 parents dad44cf + bce5080
commit c97bce3
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
+	}
+}