Make connections per host configurable

1 files changed, 8 insertions, 5 deletions

diff --git a/client/go/internal/vespa/document/http.go b/client/go/internal/vespa/document/http.go
index 2f0d085085c..ad6765aecc8 100644
--- a/client/go/internal/vespa/document/http.go
+++ b/client/go/internal/vespa/document/http.go
@@ -26,10 +26,11 @@ type Client struct {
 
 // ClientOptions specifices the configuration options of a feed client.
 type ClientOptions struct {
-	BaseURL    string
-	Timeout    time.Duration
-	Route      string
-	TraceLevel *int
+	MaxConnsPerHost int
+	BaseURL         string
+	Timeout         time.Duration
+	Route           string
+	TraceLevel      *int
 }
 
 type countingReader struct {
@@ -44,12 +45,14 @@ func (r *countingReader) Read(p []byte) (int, error) {
 }
 
 func NewClient(options ClientOptions, httpClient util.HTTPClient) *Client {
-	return &Client{
+	c := &Client{
 		options:    options,
 		httpClient: httpClient,
 		stats:      NewStats(),
 		now:        time.Now,
 	}
+	httpClient.Transport().MaxConnsPerHost = options.MaxConnsPerHost
+	return c
 }
 
 func (c *Client) queryParams() url.Values {