improve stress tools.

pull/3295/head
Monet Lee 5 months ago
parent c8c3723f90
commit 13cd8c759a

@ -161,7 +161,7 @@ func (st *StressTest) PostRequest(ctx context.Context, url string, reqbody any)
if baseResp.ErrCode != 0 { if baseResp.ErrCode != 0 {
err = fmt.Errorf(baseResp.ErrMsg) err = fmt.Errorf(baseResp.ErrMsg)
log.ZError(ctx, "Failed to send request", err, "url", url, "reqbody", reqbody, "resp", baseResp) // log.ZError(ctx, "Failed to send request", err, "url", url, "reqbody", reqbody, "resp", baseResp)
return nil, err return nil, err
} }
@ -530,9 +530,20 @@ func main() {
// Create 100K groups // Create 100K groups
st.Wg.Add(1) st.Wg.Add(1)
go func(idx int) { go func(idx int) {
startTime := time.Now()
defer func() {
elapsedTime := time.Since(startTime)
log.ZInfo(st.Ctx, "100K group creation completed",
"groupID", fmt.Sprintf("v2_StressTest_Group_100K_%d", idx),
"index", idx,
"duration", elapsedTime.String())
}()
defer st.Wg.Done() defer st.Wg.Done()
defer func() { defer func() {
st.Mutex.Lock()
st.Create100kGroupCounter++ st.Create100kGroupCounter++
st.Mutex.Unlock()
}() }()
groupID := fmt.Sprintf("v2_StressTest_Group_100K_%d", idx) groupID := fmt.Sprintf("v2_StressTest_Group_100K_%d", idx)
@ -556,7 +567,7 @@ func main() {
} }
if len(InviteUserIDs) == 0 { if len(InviteUserIDs) == 0 {
log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID) // log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
continue continue
} }
@ -567,7 +578,7 @@ func main() {
} }
if len(InviteUserIDs) == 0 { if len(InviteUserIDs) == 0 {
log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID) // log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
continue continue
} }
@ -602,9 +613,20 @@ func main() {
// Create 999 groups // Create 999 groups
st.Wg.Add(1) st.Wg.Add(1)
go func(idx int) { go func(idx int) {
startTime := time.Now()
defer func() {
elapsedTime := time.Since(startTime)
log.ZInfo(st.Ctx, "999 group creation completed",
"groupID", fmt.Sprintf("v2_StressTest_Group_1K_%d", idx),
"index", idx,
"duration", elapsedTime.String())
}()
defer st.Wg.Done() defer st.Wg.Done()
defer func() { defer func() {
st.Mutex.Lock()
st.Create999GroupCounter++ st.Create999GroupCounter++
st.Mutex.Unlock()
}() }()
groupID := fmt.Sprintf("v2_StressTest_Group_1K_%d", idx) groupID := fmt.Sprintf("v2_StressTest_Group_1K_%d", idx)
@ -627,7 +649,7 @@ func main() {
} }
if len(InviteUserIDs) == 0 { if len(InviteUserIDs) == 0 {
log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID) // log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
continue continue
} }
@ -638,7 +660,7 @@ func main() {
} }
if len(InviteUserIDs) == 0 { if len(InviteUserIDs) == 0 {
log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID) // log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
continue continue
} }

Loading…
Cancel
Save