mg-transport-api-client-go/v1/request.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
}