add test

sortset/sortset_test.go

@@ -13,6 +13,14 @@ func getRandId(max int) string {
 	return strconv.Itoa(rand.Intn(max) + 1)
 }
 
+func getAWorker(id string) *WorkerInfo {
+	return &WorkerInfo{
+		WorkerId: uuid.NewString(),
+		ModelId:  id,
+		WaitTime: rand.Intn(100),
+	}
+}
+
 func initData(id string) []*WorkerInfo {
 	w := make([]*WorkerInfo, 100000)
 	for i := 0; i < 100000; i++ {
@@ -40,6 +48,23 @@ func BenchmarkSortSet_AddWorker(b *testing.B) {
 	}
 }
 
+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() {
+			if err := sortSet.AddWorker(getAWorker(mid)); err != nil {
+				panic(fmt.Sprintf("add worker failed: %v", err))
+			}
+		}
+	})
+}
+
 func BenchmarkSortSet_FindWorker(b *testing.B) {
 	client := redis.NewClient(&redis.Options{
 		Addr: "0.0.0.0:6379",
@@ -54,3 +79,20 @@ func BenchmarkSortSet_FindWorker(b *testing.B) {
 		}
 	}
 }
+
+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() {
+			if _, err := sortSet.FindWorker(mid, 10); err != nil {
+				panic(fmt.Sprintf("find worker failed: %v", err))
+			}
+		}
+	})
+}