update test

db/workerinfo_test.go

@@ -17,7 +17,8 @@ import (
 )
 
 var (
-	idlist = make([]string, 0, 1000000)
+	maxModelId = 10000
+	idlist     = make([]string, 0, 1000000)
 	//workers    = make([]*DbWorkerInfo, 0, 1000000)
 	database                = "test"
 	collection              = "workers"
@@ -197,7 +198,7 @@ func generateAHardware() *types.HardwareInfo {
 
 func generateAInstallModel() *types.InstalledModel {
 	return &types.InstalledModel{
-		ModelID:       getRandId(100),
+		ModelID:       getRandId(maxModelId),
 		DiskSize:      101,
 		InstalledTime: time.Now().Unix(),
 		LastRunTime:   time.Now().Unix(),
@@ -206,7 +207,7 @@ func generateAInstallModel() *types.InstalledModel {
 
 func generateARunningModel() *types.RunningModel {
 	return &types.RunningModel{
-		ModelID:       getRandId(100),
+		ModelID:       getRandId(maxModelId),
 		GpuSeq:        rand.Intn(3),
 		GpuRAM:        generateAGpuRam(),
 		StartedTime:   time.Now().Unix(),
@@ -217,16 +218,17 @@ func generateARunningModel() *types.RunningModel {
 }
 
 func generateAModel() *types.ModelInfo {
-	return &types.ModelInfo{
-		InstalledModels: []*types.InstalledModel{
-			generateAInstallModel(),
-			generateAInstallModel(),
-			generateAInstallModel(),
-		},
-		RunningModels: []*types.RunningModel{
-			generateARunningModel(),
-		},
+	m := &types.ModelInfo{
+		InstalledModels: make([]*types.InstalledModel, 0, 1000),
+		RunningModels:   make([]*types.RunningModel, 0, 1000),
+	}
+	for i := 0; i < 500; i++ {
+		m.InstalledModels = append(m.InstalledModels, generateAInstallModel())
+		if len(m.RunningModels) < 500 {
+			m.RunningModels = append(m.RunningModels, generateARunningModel())
+		}
 	}
+	return m
 }
 
 func BenchmarkGenerateWorker(b *testing.B) {
@@ -393,7 +395,7 @@ func BenchmarkDbWorker_FindWorkerByInstallModelAndSortByGpuRam(b *testing.B) {
 	defer db.client.Disconnect(context.Background())
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
-		installedModelId := getRandId(100)
+		installedModelId := getRandId(maxModelId)
 		performance := generateAGpuPerformance()
 		ram := generateAGpuRam()
 		if w, err := db.FindWorkerByInstallModelAndSortByGpuRam(context.Background(), installedModelId, performance, ram, 10); err != nil {
@@ -413,7 +415,7 @@ func BenchmarkDbWorker_FindWorkerByInstallModelAndSortByGpuRam_Parallel(b *testi
 	defer db.client.Disconnect(context.Background())
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
-			installedModelId := getRandId(100)
+			installedModelId := getRandId(maxModelId)
 			performance := generateAGpuPerformance()
 			ram := generateAGpuRam()
 			if w, err := db.FindWorkerByInstallModelAndSortByGpuRam(context.Background(), installedModelId, performance, ram, 10); err != nil {
@@ -434,7 +436,7 @@ func BenchmarkDbWorker_FindWorkerByRunningModelAndSortByWaitTime(b *testing.B) {
 	defer db.client.Disconnect(context.Background())
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
-		runningModelId := getRandId(100)
+		runningModelId := getRandId(maxModelId)
 		if w, err := db.FindWorkerByRunningModelAndSortByWaitTime(context.Background(), runningModelId, 10); err != nil {
 			panic(fmt.Sprintf("find worker failed with err:%s", err))
 		} else if len(w) == 0 {
@@ -452,7 +454,7 @@ func BenchmarkDbWorker_FindWorkerByRunningModelAndSortByWaitTime_Parallel(b *tes
 	defer db.client.Disconnect(context.Background())
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
-			runningModelId := getRandId(100)
+			runningModelId := getRandId(maxModelId)
 			if w, err := db.FindWorkerByRunningModelAndSortByWaitTime(context.Background(), runningModelId, 10); err != nil {
 				panic(fmt.Sprintf("find worker failed with err:%s", err))
 			} else if len(w) == 0 {