Merge pull request #12 from cameront/master

allow setting http client to something non-default
mihasya · Dec 27, 2017 · c2a1624 · c2a1624
2 parents 0ec8d5e + 1bdf70c
commit c2a1624
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -65,6 +65,12 @@ type Batch struct {
 	Source      string        `json:"source"`
 }
 
+var client = http.DefaultClient
+
+func SetHTTPClient(c *http.Client) {
+	client = c
+}
+
 func (self *LibratoClient) PostMetrics(batch Batch) (err error) {
 	var (
 		js   []byte
@@ -87,7 +93,7 @@ func (self *LibratoClient) PostMetrics(batch Batch) (err error) {
 	req.Header.Set("Content-Type", "application/json")
 	req.SetBasicAuth(self.Email, self.Token)
 
-	if resp, err = http.DefaultClient.Do(req); err != nil {
+	if resp, err = client.Do(req); err != nil {
 		return
 	}
 	defer resp.Body.Close()