feat: initialize shared tenant workbench data

server/internal/library/imworkbench/client.go

@@ -65,27 +65,41 @@ func DefaultClient(ctx context.Context) *Client {
 }
 
 func (c *Client) Get(ctx context.Context, path string, values url.Values, out any) error {
+	return c.GetForTenant(ctx, path, values, 0, "", out)
+}
+
+func (c *Client) GetForTenant(ctx context.Context, path string, values url.Values, tenantID int64, tenantCode string, out any) error {
 	if values != nil {
 		if query := values.Encode(); query != "" {
 			path += "?" + query
 		}
 	}
-	return c.doJSON(ctx, http.MethodGet, path, nil, out)
+	return c.doJSON(ctx, http.MethodGet, path, nil, tenantID, tenantCode, out)
 }
 
 func (c *Client) Post(ctx context.Context, path string, in any, out any) error {
+	return c.PostForTenant(ctx, path, in, 0, "", out)
+}
+
+func (c *Client) PostForTenant(ctx context.Context, path string, in any, tenantID int64, tenantCode string, out any) error {
 	body, err := json.Marshal(in)
 	if err != nil {
 		return gerror.Wrap(err, "序列化工作台请求失败")
 	}
-	return c.doJSON(ctx, http.MethodPost, path, body, out)
+	return c.doJSON(ctx, http.MethodPost, path, body, tenantID, tenantCode, out)
 }
 
-func (c *Client) doJSON(ctx context.Context, method string, path string, body []byte, out any) error {
+func (c *Client) doJSON(ctx context.Context, method string, path string, body []byte, tenantID int64, tenantCode string, out any) error {
 	req, err := c.newRequest(ctx, method, path, body)
 	if err != nil {
 		return err
 	}
+	if tenantID > 0 {
+		req.Header.Set("X-Tenant-ID", strconv.FormatInt(tenantID, 10))
+	}
+	if tenantCode = strings.TrimSpace(tenantCode); tenantCode != "" {
+		req.Header.Set("X-Tenant-Code", tenantCode)
+	}
 	resp, err := c.http.Do(req)
 	if err != nil {
 		g.Log().Warningf(ctx, "IM workbench request failed url:%s err:%+v", req.URL.String(), err)

server/internal/library/imworkbench/client_test.go

@@ -0,0 +1,37 @@
+package imworkbench
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+func TestPostForTenantAddsTenantHeaders(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if got := r.Header.Get("X-Tenant-ID"); got != "23" {
+			t.Fatalf("X-Tenant-ID = %q, want 23", got)
+		}
+		if got := r.Header.Get("X-Tenant-Code"); got != "merchant-23" {
+			t.Fatalf("X-Tenant-Code = %q, want merchant-23", got)
+		}
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write([]byte(`{"code":0,"data":{"tenant_id":23}}`))
+	}))
+	defer server.Close()
+
+	client := &Client{
+		conf: Config{BaseURL: server.URL, ServiceName: "test", Secret: "secret", Timeout: time.Second},
+		http: server.Client(),
+	}
+	var response struct {
+		TenantID int64 `json:"tenant_id"`
+	}
+	if err := client.PostForTenant(context.Background(), "/initialize", map[string]any{"tenant_id": 23}, 23, "merchant-23", &response); err != nil {
+		t.Fatal(err)
+	}
+	if response.TenantID != 23 {
+		t.Fatalf("tenant_id = %d, want 23", response.TenantID)
+	}
+}