diff --git a/test/stress-test-v2/main.go b/test/stress-test-v2/main.go index 0c309b9c9..0dd069547 100644 --- a/test/stress-test-v2/main.go +++ b/test/stress-test-v2/main.go @@ -161,7 +161,7 @@ func (st *StressTest) PostRequest(ctx context.Context, url string, reqbody any) if baseResp.ErrCode != 0 { 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 } @@ -530,9 +530,20 @@ func main() { // Create 100K groups st.Wg.Add(1) 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 func() { + st.Mutex.Lock() st.Create100kGroupCounter++ + st.Mutex.Unlock() }() groupID := fmt.Sprintf("v2_StressTest_Group_100K_%d", idx) @@ -556,7 +567,7 @@ func main() { } 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 } @@ -567,7 +578,7 @@ func main() { } 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 } @@ -602,9 +613,20 @@ func main() { // Create 999 groups st.Wg.Add(1) 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 func() { + st.Mutex.Lock() st.Create999GroupCounter++ + st.Mutex.Unlock() }() groupID := fmt.Sprintf("v2_StressTest_Group_1K_%d", idx) @@ -627,7 +649,7 @@ func main() { } 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 } @@ -638,7 +660,7 @@ func main() { } 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 }