client: Allow configuration of http client

Signed-off-by: yolossn <nssvlr@gmail.com>

api/client.go

@@ -40,6 +40,8 @@ type Config struct {
 	// The address of the Prometheus to connect to.
 	Address string
 
+	Client *http.Client
+
 	// RoundTripper is used by the Client to drive HTTP requests. If not
 	// provided, DefaultRoundTripper will be used.
 	RoundTripper http.RoundTripper
@@ -52,6 +54,16 @@ func (cfg *Config) roundTripper() http.RoundTripper {
 	return cfg.RoundTripper
 }
 
+func (cfg *Config) client() *http.Client {
+	if cfg.Client == nil {
+		// Set the roundtripper for old code
+		return &http.Client{
+			Transport: cfg.roundTripper(),
+		}
+	}
+	return cfg.Client
+}
+
 // Client is the interface for an API client.
 type Client interface {
 	URL(ep string, args map[string]string) *url.URL
@@ -70,7 +82,7 @@ func NewClient(cfg Config) (Client, error) {
 
 	return &httpClient{
 		endpoint: u,
-		client:   http.Client{Transport: cfg.roundTripper()},
+		client:   *cfg.client(),
 	}, nil
 }