Propagate context into datasource read functions (#1282)

Signed-off-by: Dave Henderson <dhenderson@gmail.com>

1 files changed, 3 insertions, 2 deletions

diff --git a/data/datasource_http.go b/data/datasource_http.go
index e809768e..ed1c97b7 100644
--- a/data/datasource_http.go
+++ b/data/datasource_http.go
@@ -1,6 +1,7 @@
 package data
 
 import (
+	"context"
 	"io/ioutil"
 	"mime"
 	"net/http"
@@ -21,7 +22,7 @@ func buildURL(base *url.URL, args ...string) (*url.URL, error) {
 	return base.ResolveReference(p), nil
 }
 
-func readHTTP(source *Source, args ...string) ([]byte, error) {
+func readHTTP(ctx context.Context, source *Source, args ...string) ([]byte, error) {
 	if source.hc == nil {
 		source.hc = &http.Client{Timeout: time.Second * 5}
 	}
@@ -29,7 +30,7 @@ func readHTTP(source *Source, args ...string) ([]byte, error) {
 	if err != nil {
 		return nil, err
 	}
-	req, err := http.NewRequest("GET", u.String(), nil)
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, u.String(), nil)
 	if err != nil {
 		return nil, err
 	}