109 lines
1.9 KiB
Go
109 lines
1.9 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/tomnomnom/linkheader"
|
|
|
|
resty "gopkg.in/resty.v0"
|
|
)
|
|
|
|
// Client represents the client state for the API.
|
|
type Client struct {
|
|
RestyClient *resty.Client
|
|
}
|
|
|
|
type ErrorResponse struct {
|
|
Detail *ErrorDetail `json:"error,omitempty"`
|
|
}
|
|
|
|
func (e *ErrorResponse) Error() string {
|
|
if e != nil && e.Detail != nil {
|
|
return e.Detail.Title
|
|
}
|
|
return "Unknown error"
|
|
}
|
|
|
|
type ErrorDetail struct {
|
|
Title string `json:"title,omitempty"`
|
|
}
|
|
|
|
// Config contains all the configuration data for the API Client
|
|
type Config struct {
|
|
APIKey string
|
|
BaseURL string
|
|
Debug bool
|
|
}
|
|
|
|
// New returns a new Client for the specified apiKey.
|
|
func New(config Config) Client {
|
|
r := resty.New()
|
|
|
|
baseURL := config.BaseURL
|
|
if baseURL == "" {
|
|
baseURL = "https://api.newrelic.com/v2"
|
|
}
|
|
|
|
r.SetHeader("X-Api-Key", config.APIKey)
|
|
r.SetHostURL(baseURL)
|
|
|
|
if config.Debug {
|
|
r.SetDebug(true)
|
|
}
|
|
|
|
c := Client{
|
|
RestyClient: r,
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
// Do exectes an API request with the specified parameters.
|
|
func (c *Client) Do(method string, path string, body interface{}, response interface{}) (string, error) {
|
|
r := c.RestyClient.R().
|
|
SetError(&ErrorResponse{})
|
|
|
|
if body != nil {
|
|
r = r.SetBody(body)
|
|
}
|
|
|
|
if response != nil {
|
|
r = r.SetResult(response)
|
|
}
|
|
|
|
apiResponse, err := r.Execute(method, path)
|
|
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
nextPath := ""
|
|
header := apiResponse.Header().Get("Link")
|
|
if header != "" {
|
|
links := linkheader.Parse(header)
|
|
|
|
for _, link := range links.FilterByRel("next") {
|
|
nextPath = link.URL
|
|
break
|
|
}
|
|
}
|
|
|
|
statusClass := apiResponse.StatusCode() / 100 % 10
|
|
|
|
if statusClass == 2 {
|
|
return nextPath, nil
|
|
}
|
|
|
|
rawError := apiResponse.Error()
|
|
|
|
if rawError != nil {
|
|
apiError := rawError.(*ErrorResponse)
|
|
|
|
if apiError.Detail != nil {
|
|
return "", apiError
|
|
}
|
|
}
|
|
|
|
return "", fmt.Errorf("Unexpected status %v returned from API", apiResponse.StatusCode())
|
|
}
|