terraform/vendor/github.com/zorkian/go-datadog-api/request.go

132 lines
3.0 KiB
Go
Raw Normal View History

/*
* Datadog API for Go
*
* Please see the included LICENSE file for licensing information.
*
* Copyright 2013 by authors and contributors.
*/
package datadog
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"io"
"io/ioutil"
"net/http"
"os"
"strings"
"time"
"github.com/cenkalti/backoff"
)
// uriForAPI is to be called with something like "/v1/events" and it will give
// the proper request URI to be posted to.
func (self *Client) uriForAPI(api string) string {
url := os.Getenv("DATADOG_HOST")
if url == "" {
url = "https://app.datadoghq.com"
}
if strings.Index(api, "?") > -1 {
return url + "/api" + api + "&api_key=" +
self.apiKey + "&application_key=" + self.appKey
} else {
return url + "/api" + api + "?api_key=" +
self.apiKey + "&application_key=" + self.appKey
}
}
// doJsonRequest is the simplest type of request: a method on a URI that returns
// some JSON result which we unmarshal into the passed interface.
func (self *Client) doJsonRequest(method, api string,
reqbody, out interface{}) error {
// Handle the body if they gave us one.
var bodyreader io.Reader
if method != "GET" && reqbody != nil {
bjson, err := json.Marshal(reqbody)
if err != nil {
return err
}
bodyreader = bytes.NewReader(bjson)
}
req, err := http.NewRequest(method, self.uriForAPI(api), bodyreader)
if err != nil {
return err
}
if bodyreader != nil {
req.Header.Add("Content-Type", "application/json")
}
// Perform the request and retry it if it's not a POST request
var resp *http.Response
if method == "POST" {
resp, err = self.HttpClient.Do(req)
} else {
resp, err = self.doRequestWithRetries(req, 60*time.Second)
}
if err != nil {
return err
}
defer resp.Body.Close()
if resp.StatusCode < 200 || resp.StatusCode > 299 {
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return err
}
return fmt.Errorf("API error %s: %s", resp.Status, body)
}
// If they don't care about the body, then we don't care to give them one,
// so bail out because we're done.
if out == nil {
return nil
}
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return err
}
// If we got no body, by default let's just make an empty JSON dict. This
// saves us some work in other parts of the code.
if len(body) == 0 {
body = []byte{'{', '}'}
}
err = json.Unmarshal(body, &out)
if err != nil {
return err
}
return nil
}
// doRequestWithRetries performs an HTTP request repeatedly for maxTime or until
// no error and no HTTP response code higher than 299 is returned.
func (self *Client) doRequestWithRetries(req *http.Request, maxTime time.Duration) (*http.Response, error) {
var (
err error
resp *http.Response
bo = backoff.NewExponentialBackOff()
)
bo.MaxElapsedTime = maxTime
err = backoff.Retry(func() error {
resp, err = self.HttpClient.Do(req)
if err != nil {
return err
}
if resp.StatusCode < 200 || resp.StatusCode > 299 {
return errors.New("API error: " + resp.Status)
}
return nil
}, bo)
return resp, err
}