From 3b710fdfdbcc825dabd29afea637df9f484299ae Mon Sep 17 00:00:00 2001
From: Monet Lee <monet_lee@163.com>
Date: Mon, 31 Mar 2025 16:45:52 +0800
Subject: [PATCH] fix: improve stress test tools parms. (#3265)

* feat: implement stress-test tools.

* revert config file.

* fix: improve tools parms.

* fix modify args.
---
 tools/stress-test/main.go | 35 +++++++++++++++++++++--------------
 1 file changed, 21 insertions(+), 14 deletions(-)

diff --git a/tools/stress-test/main.go b/tools/stress-test/main.go
index ee58c9749..aa52b69ed 100755
--- a/tools/stress-test/main.go
+++ b/tools/stress-test/main.go
@@ -14,6 +14,7 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/openimsdk/open-im-server/v3/pkg/apistruct"
 	"github.com/openimsdk/open-im-server/v3/pkg/common/config"
 	"github.com/openimsdk/protocol/auth"
 	"github.com/openimsdk/protocol/constant"
@@ -232,12 +233,15 @@ func (st *StressTest) SendMsg(ctx context.Context, userID string) error {
 		"content": fmt.Sprintf("index %d. The current time is %s", st.MsgCounter, time.Now().Format("2006-01-02 15:04:05.000")),
 	}
 
-	req := map[string]any{
-		"sendID":      userID,
-		"groupID":     st.DefaultGroupID,
-		"contentType": constant.Text,
-		"sessionType": constant.ReadGroupChatType,
-		"content":     contentObj,
+	req := &apistruct.SendMsgReq{
+		SendMsg: apistruct.SendMsg{
+			SendID:         userID,
+			SenderNickname: userID,
+			GroupID:        st.DefaultGroupID,
+			ContentType:    constant.Text,
+			SessionType:    constant.ReadGroupChatType,
+			Content:        contentObj,
+		},
 	}
 
 	_, err := st.PostRequest(ctx, ApiAddress+SendMsg, &req)
@@ -254,15 +258,18 @@ func (st *StressTest) SendMsg(ctx context.Context, userID string) error {
 func (st *StressTest) CreateGroup(ctx context.Context, userID string) (string, error) {
 	groupID := fmt.Sprintf("StressTestGroup_%d_%s", st.GroupCounter, time.Now().Format("20060102150405"))
 
-	req := map[string]any{
-		"memberUserIDs": TestTargetUserList,
-		"ownerUserID":   userID,
-		"groupInfo": map[string]any{
-			"groupID":   groupID,
-			"groupName": groupID,
-			"groupType": constant.WorkingGroup,
-		},
+	groupInfo := &sdkws.GroupInfo{
+		GroupID:   groupID,
+		GroupName: groupID,
+		GroupType: constant.WorkingGroup,
 	}
+
+	req := group.CreateGroupReq{
+		OwnerUserID:   userID,
+		MemberUserIDs: TestTargetUserList,
+		GroupInfo:     groupInfo,
+	}
+
 	resp := group.CreateGroupResp{}
 
 	response, err := st.PostRequest(ctx, ApiAddress+CreateGroup, &req)