Add http timeout option

Author: ischasny

engine/engine.go

@@ -68,6 +68,9 @@ func New(resultCache cache.Interface, valueStore indexer.Interface, options ...O
 		dhMergeURL:  dhMergeURL,
 		dhMetaURL:   dhMetaURL,
 		vsNoNewMH:   opts.vsNoNewMH,
+		httpClient: http.Client{
+			Timeout: opts.httpClientTimeout,
+		},
 	}
 }
 

engine/option.go

@@ -3,24 +3,28 @@ package engine
 import (
 	"fmt"
 	"net/url"
+	"time"
 )
 
 const defaultDHBatchSize = 1024
+const defaultHttpTimeout = 5 * time.Second
 
 // config contains all options for configuring Engine.
 type config struct {
-	cacheOnPut  bool
-	dhBatchSize int
-	dhstoreURL  string
-	vsNoNewMH   bool
+	cacheOnPut        bool
+	dhBatchSize       int
+	dhstoreURL        string
+	vsNoNewMH         bool
+	httpClientTimeout time.Duration
 }
 
 type Option func(*config) error
 
 // getOpts creates a config and applies Options to it.
 func getOpts(opts []Option) (config, error) {
 	cfg := config{
-		dhBatchSize: defaultDHBatchSize,
+		dhBatchSize:       defaultDHBatchSize,
+		httpClientTimeout: defaultHttpTimeout,
 	}
 
 	for i, opt := range opts {
@@ -74,3 +78,11 @@ func WithVSNoNewMH(ok bool) Option {
 		return nil
 	}
 }
+
+// WithHttpClientTimeout sets http timeout for queries to DHStore
+func WithHttpClientTimeout(timeout time.Duration) Option {
+	return func(cfg *config) error {
+		cfg.httpClientTimeout = timeout
+		return nil
+	}
+}