Merge pull request #170 from attestantio/allow-http-client

Allow custom HTTP client.

http/parameters.go

@@ -15,6 +15,7 @@ package http
 
 import (
 	"errors"
+	"net/http"
 	"time"
 
 	"github.com/attestantio/go-eth2-client/metrics"
@@ -34,6 +35,7 @@ type parameters struct {
 	hooks              *Hooks
 	reducedMemoryUsage bool
 	customSpecSupport  bool
+	client             *http.Client
 }
 
 // Parameter is the interface for service parameters.
@@ -134,6 +136,14 @@ func WithCustomSpecSupport(customSpecSupport bool) Parameter {
 	})
 }
 
+// WithHTTPClient provides a custom HTTP client for communication with the HTTP server.
+// If not supplied then a standard HTTP client is used.
+func WithHTTPClient(client *http.Client) Parameter {
+	return parameterFunc(func(p *parameters) {
+		p.client = client
+	})
+}
+
 // parseAndCheckParameters parses and checks parameters to ensure that mandatory parameters are present and correct.
 func parseAndCheckParameters(params ...Parameter) (*parameters, error) {
 	parameters := parameters{

http/service.go

@@ -95,18 +95,21 @@ func New(ctx context.Context, params ...Parameter) (client.Service, error) {
 		}
 	}
 
-	httpClient := &http.Client{
-		Transport: &http.Transport{
-			DialContext: (&net.Dialer{
-				Timeout:   parameters.timeout,
-				KeepAlive: 30 * time.Second,
-				DualStack: true,
-			}).DialContext,
-			MaxIdleConns:        64,
-			MaxConnsPerHost:     64,
-			MaxIdleConnsPerHost: 64,
-			IdleConnTimeout:     600 * time.Second,
-		},
+	httpClient := parameters.client
+	if httpClient == nil {
+		httpClient = &http.Client{
+			Transport: &http.Transport{
+				DialContext: (&net.Dialer{
+					Timeout:   parameters.timeout,
+					KeepAlive: 30 * time.Second,
+					DualStack: true,
+				}).DialContext,
+				MaxIdleConns:        64,
+				MaxConnsPerHost:     64,
+				MaxIdleConnsPerHost: 64,
+				IdleConnTimeout:     600 * time.Second,
+			},
+		}
 	}
 
 	base, address, err := parseAddress(parameters.address)