chore: seed migration objects batch 5

server/internal/library/imadmin/client.go

@@ -17,6 +17,7 @@ import (
 	"time"
 
 	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/os/gctx"
 )
 
@@ -198,6 +199,9 @@ func (c *Client) parseError(resp *http.Response) error {
 		Reason  string `json:"reason"`
 	}
 	_ = json.Unmarshal(body, &envelope)
+	if envelope.Message == "" && envelope.Reason == "" {
+		g.Log().Warningf(context.Background(), "IM API returned non-success status:%d url:%s body:%s", resp.StatusCode, resp.Request.URL.String(), string(body))
+	}
 	return stableError(envelope.Message, envelope.Reason)
 }
 

server/internal/library/imadmin/client_test.go

@@ -58,6 +58,39 @@ func TestListCodesWithInviteCodeUsesSignedSearchPost(t *testing.T) {
 	}
 }
 
+func TestInternalPostUsesInternalSignatureAndBody(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != http.MethodPost || r.URL.Path != "/internal/admin-query/tenants/default/users" {
+			t.Fatalf("unexpected request %s %s", r.Method, r.URL.Path)
+		}
+		body, err := io.ReadAll(r.Body)
+		if err != nil {
+			t.Fatalf("read body: %v", err)
+		}
+		if string(body) != `{"username":"alice"}` {
+			t.Fatalf("unexpected body %s", body)
+		}
+		if r.Header.Get("X-Service-Name") != "hotgo-admin" || r.Header.Get("X-Signature") == "" {
+			t.Fatalf("missing internal authentication headers")
+		}
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write([]byte(`{"code":0,"data":{"user_id":123,"username":"alice"}}`))
+	}))
+	defer server.Close()
+
+	client := NewClient(Config{InternalBaseUrl: server.URL, InternalServiceName: "hotgo-admin", InternalServiceSecret: "secret", Timeout: time.Second})
+	var out struct {
+		UserID   int64  `json:"user_id"`
+		Username string `json:"username"`
+	}
+	if err := client.InternalPost(context.Background(), "/internal/admin-query/tenants/default/users", map[string]string{"username": "alice"}, &out); err != nil {
+		t.Fatalf("InternalPost returned error: %v", err)
+	}
+	if out.UserID != 123 || out.Username != "alice" {
+		t.Fatalf("unexpected response: %+v", out)
+	}
+}
+
 func signedTestServer(t *testing.T, method string, path string, inspect func(t *testing.T, r *http.Request, body []byte)) *httptest.Server {
 	t.Helper()
 	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

server/internal/library/imadmin/internal.go

@@ -14,15 +14,29 @@ import (
 	"time"
 
 	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
 )
 
 func (c *Client) InternalGet(ctx context.Context, path string, out any) error {
-	req, err := c.newInternalRequest(ctx, http.MethodGet, path, nil)
+	return c.internalJSON(ctx, http.MethodGet, path, nil, out)
+}
+
+func (c *Client) InternalPost(ctx context.Context, path string, in any, out any) error {
+	body, err := json.Marshal(in)
+	if err != nil {
+		return gerror.Wrap(err, "序列化IM内部请求失败")
+	}
+	return c.internalJSON(ctx, http.MethodPost, path, body, out)
+}
+
+func (c *Client) internalJSON(ctx context.Context, method, path string, body []byte, out any) error {
+	req, err := c.newInternalRequest(ctx, method, path, body)
 	if err != nil {
 		return err
 	}
 	resp, err := c.http.Do(req)
 	if err != nil {
+		g.Log().Warningf(ctx, "IM internal query request failed url:%s err:%+v", req.URL.String(), err)
 		return gerror.New("IM内部查询服务暂不可用")
 	}
 	defer resp.Body.Close()
@@ -32,7 +46,7 @@ func (c *Client) InternalGet(ctx context.Context, path string, out any) error {
 	if out == nil {
 		return nil
 	}
-	body, err := io.ReadAll(io.LimitReader(resp.Body, maxResponseBody))
+	responseBody, err := io.ReadAll(io.LimitReader(resp.Body, maxResponseBody))
 	if err != nil {
 		return gerror.Wrap(err, "读取IM内部查询响应失败")
 	}
@@ -42,13 +56,13 @@ func (c *Client) InternalGet(ctx context.Context, path string, out any) error {
 		Reason  string          `json:"reason"`
 		Data    json.RawMessage `json:"data"`
 	}
-	if err = json.Unmarshal(body, &envelope); err == nil && len(envelope.Data) > 0 {
+	if err = json.Unmarshal(responseBody, &envelope); err == nil && len(envelope.Data) > 0 {
 		if envelope.Code != 0 {
 			return stableError(envelope.Message, envelope.Reason)
 		}
 		return json.Unmarshal(envelope.Data, out)
 	}
-	return json.Unmarshal(body, out)
+	return json.Unmarshal(responseBody, out)
 }
 
 func (c *Client) newInternalRequest(ctx context.Context, method string, path string, bodyBytes []byte) (*http.Request, error) {