84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
|
package mes
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"net/http"
|
|||
|
|
"net/url"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Client MES 内部 API HTTP 客户端。
|
|||
|
|
// 自动携带请求头：X-API-TOKEN（项目间 token）、Content-Type: application/json。
|
|||
|
|
// 任何非 2xx 状态返回包含响应体的错误信息。
|
|||
|
|
type Client struct {
|
|||
|
|
baseURL string
|
|||
|
|
token string
|
|||
|
|
httpClient *http.Client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func New(baseURL, token string) *Client {
|
|||
|
|
return &Client{
|
|||
|
|
baseURL: strings.TrimRight(baseURL, "/"),
|
|||
|
|
token: token,
|
|||
|
|
httpClient: &http.Client{Timeout: 10 * time.Second},
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Get 发送 GET 请求，query 为空则忽略。
|
|||
|
|
func (c *Client) Get(path string, query url.Values) ([]byte, http.Header, error) {
|
|||
|
|
u := c.baseURL + path
|
|||
|
|
if len(query) > 0 {
|
|||
|
|
u += "?" + query.Encode()
|
|||
|
|
}
|
|||
|
|
return c.do(http.MethodGet, u, nil)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Post 发送 POST 请求，body 序列化为 JSON（body 为 nil 则不发送请求体）。
|
|||
|
|
func (c *Client) Post(path string, body any) ([]byte, http.Header, error) {
|
|||
|
|
var reader io.Reader
|
|||
|
|
if body != nil {
|
|||
|
|
b, err := json.Marshal(body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, fmt.Errorf("序列化请求体失败: %w", err)
|
|||
|
|
}
|
|||
|
|
reader = bytes.NewReader(b)
|
|||
|
|
}
|
|||
|
|
return c.do(http.MethodPost, c.baseURL+path, reader)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Client) do(method, u string, body io.Reader) ([]byte, http.Header, error) {
|
|||
|
|
req, err := http.NewRequest(method, u, body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
req.Header.Set("X-API-TOKEN", c.token)
|
|||
|
|
req.Header.Set("Content-Type", "application/json;charset=utf-8")
|
|||
|
|
|
|||
|
|
resp, err := c.httpClient.Do(req)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
respBody, err := io.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp.Header, err
|
|||
|
|
}
|
|||
|
|
if resp.StatusCode < 200 || resp.StatusCode > 299 {
|
|||
|
|
return respBody, resp.Header, fmt.Errorf("MES %s %s 返回 %d：%s",
|
|||
|
|
method, u, resp.StatusCode, truncate(string(respBody), 200))
|
|||
|
|
}
|
|||
|
|
return respBody, resp.Header, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func truncate(s string, n int) string {
|
|||
|
|
s = strings.TrimSpace(s)
|
|||
|
|
if len(s) > n {
|
|||
|
|
return s[:n] + "..."
|
|||
|
|
}
|
|||
|
|
return s
|
|||
|
|
}
|