初始化2

bj_power_mes/internal/wmsclient/client.go

@@ -0,0 +1,71 @@
+package wmsclient
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"net/http"
+	"time"
+)
+
+// Client WMS 内部接口客户端（半成品流转等）
+type Client struct {
+	baseURL string
+	token   string
+	hc      *http.Client
+}
+
+func New(baseURL, token string) *Client {
+	if baseURL == "" {
+		baseURL = "http://127.0.0.1:9091"
+	}
+	return &Client{
+		baseURL: baseURL,
+		token:   token,
+		hc:      &http.Client{Timeout: 5 * time.Second},
+	}
+}
+
+// SemiInbound 调 WMS /api/internal/semi/inbound 半成品入库
+func (c *Client) SemiInbound(ctx context.Context, sn, orderNo string, doneProcessCodes []int, operator string) error {
+	return c.post(ctx, "/api/internal/semi/inbound", map[string]any{
+		"sn": sn, "orderNo": orderNo, "doneProcessCodes": doneProcessCodes, "operator": operator,
+	})
+}
+
+// SemiOutbound 调 WMS /api/internal/semi/outbound 半成品出库（重上线）
+func (c *Client) SemiOutbound(ctx context.Context, sn, orderNo string, doneProcessCodes []int, operator string) error {
+	return c.post(ctx, "/api/internal/semi/outbound", map[string]any{
+		"sn": sn, "orderNo": orderNo, "doneProcessCodes": doneProcessCodes, "operator": operator,
+	})
+}
+
+func (c *Client) post(ctx context.Context, path string, body any) error {
+	b, err := json.Marshal(body)
+	if err != nil {
+		return err
+	}
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.baseURL+path, bytes.NewReader(b))
+	if err != nil {
+		return err
+	}
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("X-API-TOKEN", c.token)
+	resp, err := c.hc.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode >= 300 {
+		return &RespError{Status: resp.StatusCode}
+	}
+	return nil
+}
+
+type RespError struct {
+	Status int
+}
+
+func (e *RespError) Error() string {
+	return "wms call failed with status " + string(rune('0'+e.Status))
+}