65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package apimart
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"juhe-factory/api/internal/provider"
|
|
)
|
|
|
|
type Client struct {
|
|
HTTPClient *http.Client
|
|
}
|
|
|
|
func NewClient(client *http.Client) Client {
|
|
return Client{HTTPClient: client}
|
|
}
|
|
|
|
func (c Client) requestJSON(ctx context.Context, method, baseURL, path, apiKey, requestID string, payload map[string]any) ([]byte, error) {
|
|
var body io.Reader
|
|
if payload != nil {
|
|
encoded, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
body = bytes.NewReader(encoded)
|
|
}
|
|
request, err := http.NewRequestWithContext(ctx, method, strings.TrimRight(baseURL, "/")+path, body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
request.Header.Set("Authorization", "Bearer "+apiKey)
|
|
if payload != nil {
|
|
request.Header.Set("Content-Type", "application/json")
|
|
}
|
|
if requestID != "" {
|
|
request.Header.Set("Idempotency-Key", requestID)
|
|
request.Header.Set("X-Request-ID", requestID)
|
|
}
|
|
response, err := c.HTTPClient.Do(request)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer response.Body.Close()
|
|
data, err := io.ReadAll(io.LimitReader(response.Body, 4*1024*1024))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if response.StatusCode < 200 || response.StatusCode >= 300 {
|
|
return nil, &provider.HTTPError{StatusCode: response.StatusCode, Body: safeBody(data), RetryAfter: response.Header.Get("Retry-After")}
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func safeBody(data []byte) string {
|
|
value := strings.TrimSpace(string(data))
|
|
if len(value) > 2000 {
|
|
value = value[:2000]
|
|
}
|
|
return value
|
|
}
|