update test

sortset/sortset_test.go

@@ -6,6 +6,7 @@ import (
 	"github.com/redis/go-redis/v9"
 	"math/rand"
 	"strconv"
+	"sync"
 	"testing"
 )
 
@@ -43,12 +44,22 @@ func TestSortSet_AddWorker(t *testing.T) {
 	})
 	sortSet := NewSortSet(client)
 	fmt.Println("start add worker")
-	w := initData()
-	for i := 0; i < len(w); i++ {
-		if err := sortSet.AddWorker(w[i]); err != nil {
-			panic(fmt.Sprintf("add worker failed: %v", err))
-		}
+	wg := sync.WaitGroup{}
+	for i := 0; i < 8; i++ {
+		wg.Add(1)
+		go func(max int) {
+			defer wg.Done()
+			count := 0
+			for count < max {
+				mid := getRandId(1000)
+				if err := sortSet.AddWorker(getAWorker(mid)); err != nil {
+					panic(fmt.Sprintf("add worker failed: %v", err))
+				}
+				count++
+			}
+		}(1000 * 100000 / 8)
 	}
+	wg.Wait()
 	fmt.Println("add worker done")
 }