update test

db/workerinfo_test.go

@@ -141,14 +141,13 @@ func BenchmarkDbWorker_InsertWorker(b *testing.B) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	workers := make([]*DbWorkerInfo, 0, b.N)
-	for i := 0; i < b.N; i++ {
-		workers = append(workers, generateAWroker())
-	}
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
-		worker := workers[i]
+		b.StopTimer()
+		worker := generateAWroker()
+		b.StartTimer()
 		if _, err := db.InsertWorker(context.Background(), worker); err != nil {
 			panic(fmt.Sprintf("insert worker failed with err:%s", err))
 		}
@@ -161,6 +160,7 @@ func BenchmarkDbWorker_InsertWorker_Parallel(b *testing.B) {
 		log.Fatal(err)
 	}
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
 			worker := generateAWroker()
@@ -176,12 +176,17 @@ func BenchmarkDbWorker_UpdateHardware(b *testing.B) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	b.ResetTimer()
-	nhardware := generateAHardware()
-
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
+
+	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
+		b.StopTimer()
+
 		idx := rand.Intn(len(idlist))
+		nhardware := generateAHardware()
+
+		b.StartTimer()
 		if err := db.UpdateHardware(context.Background(), idlist[idx], nhardware); err != nil {
 			panic(fmt.Sprintf("update worker failed with err:%s", err))
 		}
@@ -193,11 +198,13 @@ func BenchmarkDbWorker_UpdateHardware_Parallel(b *testing.B) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	nhardware := generateAHardware()
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
+
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
 			idx := rand.Intn(len(idlist))
+			nhardware := generateAHardware()
 			if err := db.UpdateHardware(context.Background(), idlist[idx], nhardware); err != nil {
 				panic(fmt.Sprintf("update worker failed with err:%s", err))
 			}
@@ -211,11 +218,12 @@ func BenchmarkDbWorker_UpdateModel(b *testing.B) {
 		log.Fatal(err)
 	}
 	b.ResetTimer()
-	nresource := generateAmodel()
 
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
 	for i := 0; i < b.N; i++ {
 		idx := rand.Intn(len(idlist))
+		nresource := generateAmodel()
 		if err := db.UpdateModel(context.Background(), idlist[idx], nresource); err != nil {
 			panic(fmt.Sprintf("update worker failed with err:%s", err))
 		}
@@ -227,11 +235,13 @@ func BenchmarkDbWorker_UpdateModel_Parallel(b *testing.B) {
 	if err != nil {
 		log.Fatal(err)
 	}
-	nresource := generateAmodel()
+
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
 			idx := rand.Intn(len(idlist))
+			nresource := generateAmodel()
 			if err := db.UpdateModel(context.Background(), idlist[idx], nresource); err != nil {
 				panic(fmt.Sprintf("update worker failed with err:%s", err))
 			}
@@ -248,6 +258,7 @@ func BenchmarkDbWorker_UpdateNodeInfo(b *testing.B) {
 	nnodeinfo := generateANodeInfo()
 
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
 	for i := 0; i < b.N; i++ {
 		idx := rand.Intn(len(idlist))
 		if err := db.UpdateNodeInfo(context.Background(), idlist[idx], nnodeinfo); err != nil {
@@ -263,6 +274,7 @@ func BenchmarkDbWorker_UpdateNodeInfo_Parallel(b *testing.B) {
 	}
 	nnodeinfo := generateANodeInfo()
 	db := NewDBWorker(client, database, collection)
+	defer db.client.Disconnect(context.Background())
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
 			idx := rand.Intn(len(idlist))