132 lines
3.0 KiB
Go
132 lines
3.0 KiB
Go
/*
|
|
* 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
|
|
}
|