update test

sortset/sortset_test.go

@@ -21,43 +21,63 @@ func getAWorker(id string) *WorkerInfo {
 	}
 }
 
-func initData(id string) []*WorkerInfo {
-	w := make([]*WorkerInfo, 100000)
-	for i := 0; i < 100000; i++ {
-		w[i] = &WorkerInfo{
-			WorkerId: uuid.NewString(),
-			ModelId:  id,
-			WaitTime: rand.Intn(100),
+func initData() []*WorkerInfo {
+	tw := make([]*WorkerInfo, 0, 1000*100000)
+	for id := 0; id < 1000; id++ {
+		idx := strconv.Itoa(id)
+		for i := 0; i < 100000; i++ {
+			w := &WorkerInfo{
+				WorkerId: uuid.NewString(),
+				ModelId:  idx,
+				WaitTime: rand.Intn(100),
+			}
+			tw = append(tw, w)
 		}
 	}
-	return w
+	return tw
 }
-func BenchmarkSortSet_AddWorker(b *testing.B) {
+
+func TestSortSet_AddWorker(t *testing.T) {
 	client := redis.NewClient(&redis.Options{
 		Addr: "0.0.0.0:6379",
 	})
 	sortSet := NewSortSet(client)
-	mid := "13"
-	w := initData(mid)
-	b.ResetTimer()
-	// start benchmark
-	for i := 0; i < b.N; i++ {
-		if err := sortSet.AddWorker(w[i%100000]); err != nil {
+	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))
 		}
 	}
+	fmt.Println("add worker done")
 }
 
+//
+//func BenchmarkSortSet_AddWorker(b *testing.B) {
+//	client := redis.NewClient(&redis.Options{
+//		Addr: "0.0.0.0:6379",
+//	})
+//	sortSet := NewSortSet(client)
+//	w := initData()
+//	b.ResetTimer()
+//	// start benchmark
+//	for i := 0; i < b.N; i++ {
+//		if err := sortSet.AddWorker(w[i%100000]); err != nil {
+//			panic(fmt.Sprintf("add worker failed: %v", err))
+//		}
+//	}
+//}
+
 func BenchmarkSortSet_AddWorkerParallel(b *testing.B) {
 	client := redis.NewClient(&redis.Options{
 		Addr: "0.0.0.0:6379",
 	})
 	sortSet := NewSortSet(client)
-	mid := "18"
 	b.ResetTimer()
 	// start benchmark
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
+			mid := getRandId(1000)
 			if err := sortSet.AddWorker(getAWorker(mid)); err != nil {
 				panic(fmt.Sprintf("add worker failed: %v", err))
 			}
@@ -65,31 +85,32 @@ func BenchmarkSortSet_AddWorkerParallel(b *testing.B) {
 	})
 }
 
-func BenchmarkSortSet_FindWorker(b *testing.B) {
-	client := redis.NewClient(&redis.Options{
-		Addr: "0.0.0.0:6379",
-	})
-	sortSet := NewSortSet(client)
-	mid := "13"
-	b.ResetTimer()
-	// start benchmark
-	for i := 0; i < b.N; i++ {
-		if _, err := sortSet.FindWorker(mid, 10); err != nil {
-			panic(fmt.Sprintf("find worker failed: %v", err))
-		}
-	}
-}
+//
+//func BenchmarkSortSet_FindWorker(b *testing.B) {
+//	client := redis.NewClient(&redis.Options{
+//		Addr: "0.0.0.0:6379",
+//	})
+//	sortSet := NewSortSet(client)
+//	mid := "13"
+//	b.ResetTimer()
+//	// start benchmark
+//	for i := 0; i < b.N; i++ {
+//		if _, err := sortSet.FindWorker(mid, 10); err != nil {
+//			panic(fmt.Sprintf("find worker failed: %v", err))
+//		}
+//	}
+//}
 
 func BenchmarkSortSet_FindWorkerParallel(b *testing.B) {
 	client := redis.NewClient(&redis.Options{
 		Addr: "0.0.0.0:6379",
 	})
 	sortSet := NewSortSet(client)
-	mid := "18"
 	b.ResetTimer()
 	// start benchmark
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
+			mid := getRandId(1000)
 			if _, err := sortSet.FindWorker(mid, 10); err != nil {
 				panic(fmt.Sprintf("find worker failed: %v", err))
 			}