初始化

API/internal/provider/apimart/client.go

@@ -0,0 +1,64 @@
+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
+}