114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package v1
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
const MaxRPS = 100
|
|
|
|
var prefix = "/api/transport/v1"
|
|
|
|
// GetRequest performs GET request to the provided route.
|
|
func (c *MgClient) GetRequest(url string, parameters []byte) ([]byte, int, error) {
|
|
return makeRequest(
|
|
"GET",
|
|
fmt.Sprintf("%s%s%s", c.URL, prefix, url),
|
|
bytes.NewBuffer(parameters),
|
|
c,
|
|
)
|
|
}
|
|
|
|
// PostRequest performs POST request to the provided route.
|
|
func (c *MgClient) PostRequest(url string, parameters io.Reader) ([]byte, int, error) {
|
|
return makeRequest(
|
|
"POST",
|
|
fmt.Sprintf("%s%s%s", c.URL, prefix, url),
|
|
parameters,
|
|
c,
|
|
)
|
|
}
|
|
|
|
// PutRequest performs PUT request to the provided route.
|
|
func (c *MgClient) PutRequest(url string, parameters []byte) ([]byte, int, error) {
|
|
return makeRequest(
|
|
"PUT",
|
|
fmt.Sprintf("%s%s%s", c.URL, prefix, url),
|
|
bytes.NewBuffer(parameters),
|
|
c,
|
|
)
|
|
}
|
|
|
|
// DeleteRequest performs DELETE request to the provided route.
|
|
func (c *MgClient) DeleteRequest(url string, parameters []byte) ([]byte, int, error) {
|
|
return makeRequest(
|
|
"DELETE",
|
|
fmt.Sprintf("%s%s%s", c.URL, prefix, url),
|
|
bytes.NewBuffer(parameters),
|
|
c,
|
|
)
|
|
}
|
|
|
|
func (c *MgClient) WaitForRateLimit() {
|
|
if c.limiter != nil && c.Token != "" {
|
|
c.limiter.Obtain(c.Token)
|
|
}
|
|
}
|
|
|
|
func makeRequest(reqType, url string, buf io.Reader, c *MgClient) ([]byte, int, error) {
|
|
var res []byte
|
|
req, err := http.NewRequest(reqType, url, buf)
|
|
if err != nil {
|
|
return res, 0, err
|
|
}
|
|
|
|
req.Header.Set("Content-Type", "application/json")
|
|
req.Header.Set("X-Transport-Token", c.Token)
|
|
|
|
maxAttempt := 1
|
|
if c.limiter != nil && c.Token != "" {
|
|
maxAttempt = 3
|
|
}
|
|
|
|
attempt := 0
|
|
tryAgain:
|
|
c.WaitForRateLimit()
|
|
if c.Debug {
|
|
if strings.Contains(url, "/files/upload") {
|
|
c.writeLog("MG TRANSPORT API Request: %s %s %s [file data]", reqType, url, c.Token)
|
|
} else {
|
|
c.writeLog("MG TRANSPORT API Request: %s %s %s %v", reqType, url, c.Token, buf)
|
|
}
|
|
}
|
|
|
|
resp, err := c.httpClient.Do(req)
|
|
if err != nil {
|
|
return res, 0, NewCriticalHTTPError(err)
|
|
}
|
|
|
|
if resp.StatusCode == http.StatusTooManyRequests && attempt < maxAttempt {
|
|
attempt++
|
|
c.writeLog("MG TRANSPORT API Request rate limit hit on attempt %d, retrying", attempt)
|
|
goto tryAgain
|
|
}
|
|
|
|
if resp.StatusCode >= http.StatusInternalServerError {
|
|
err = NewServerError(resp)
|
|
return res, resp.StatusCode, err
|
|
}
|
|
|
|
res, err = buildLimitedRawResponse(resp)
|
|
if err != nil {
|
|
return res, 0, err
|
|
}
|
|
|
|
if c.Debug {
|
|
c.writeLog("MG TRANSPORT API Response: %s", res)
|
|
}
|
|
|
|
return res, resp.StatusCode, err
|
|
}
|