blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-05 20:11:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve stress test tools parms. (#3265)

Browse Source 
* feat: implement stress-test tools.

* revert config file.

* fix: improve tools parms.

* fix modify args.
This commit is contained in:
Monet Lee 2025-03-31 16:45:52 +08:00 committed by GitHub
parent 4dc9b45861
commit 3b710fdfdb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tools/stress-test/main.go
View File

@ -14,6 +14,7 @@ import (
"syscall" "syscall"
"time" "time"
"github.com/openimsdk/open-im-server/v3/pkg/apistruct"
"github.com/openimsdk/open-im-server/v3/pkg/common/config" "github.com/openimsdk/open-im-server/v3/pkg/common/config"
"github.com/openimsdk/protocol/auth" "github.com/openimsdk/protocol/auth"
"github.com/openimsdk/protocol/constant" "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")), "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{ req := &apistruct.SendMsgReq{
"sendID": userID, SendMsg: apistruct.SendMsg{
"groupID": st.DefaultGroupID, SendID: userID,
"contentType": constant.Text, SenderNickname: userID,
"sessionType": constant.ReadGroupChatType, GroupID: st.DefaultGroupID,
"content": contentObj, ContentType: constant.Text,
SessionType: constant.ReadGroupChatType,
Content: contentObj,
},
} }
_, err := st.PostRequest(ctx, ApiAddress+SendMsg, &req) _, 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) { func (st *StressTest) CreateGroup(ctx context.Context, userID string) (string, error) {
groupID := fmt.Sprintf("StressTestGroup_%d_%s", st.GroupCounter, time.Now().Format("20060102150405")) groupID := fmt.Sprintf("StressTestGroup_%d_%s", st.GroupCounter, time.Now().Format("20060102150405"))
req := map[string]any{ groupInfo := &sdkws.GroupInfo{
"memberUserIDs": TestTargetUserList, GroupID: groupID,
"ownerUserID": userID, GroupName: groupID,
"groupInfo": map[string]any{ GroupType: constant.WorkingGroup,
"groupID": groupID,
"groupName": groupID,
"groupType": constant.WorkingGroup,
},
} }
req := group.CreateGroupReq{
OwnerUserID: userID,
MemberUserIDs: TestTargetUserList,
GroupInfo: groupInfo,
}
resp := group.CreateGroupResp{} resp := group.CreateGroupResp{}
response, err := st.PostRequest(ctx, ApiAddress+CreateGroup, &req) response, err := st.PostRequest(ctx, ApiAddress+CreateGroup, &req)