benchmark with 3 different server slice sizes

Author: isopov

x/mongo/driver/topology/topology_test.go

@@ -1068,26 +1068,30 @@ func TestEqualTopologies(t *testing.T) {
 }
 
 func BenchmarkEqualTopologies(b *testing.B) {
-	servers := make([]description.Server, 100)
-	for i := 0; i < len(servers); i++ {
-		servers[i] = description.Server{
-			Addr:              address.Address("127.0.0." + strconv.Itoa(i) + ":27017"),
-			HeartbeatInterval: time.Duration(10) * time.Second,
-			LastWriteTime:     time.Date(2017, 2, 11, 14, 0, 0, 0, time.UTC),
-			LastUpdateTime:    time.Date(2017, 2, 11, 14, 0, 2, 0, time.UTC),
-			Kind:              description.ServerKindMongos,
-			WireVersion:       &description.VersionRange{Min: 6, Max: 21},
-		}
-	}
-	desc := description.Topology{
-		Servers: servers,
-	}
+	for _, bsize := range []int{10, 100, 1000} {
+		b.Run(strconv.Itoa(bsize), func(b *testing.B) {
+			servers := make([]description.Server, bsize)
+			for i := 0; i < len(servers); i++ {
+				servers[i] = description.Server{
+					Addr:              address.Address("127.0.0." + strconv.Itoa(i) + ":27017"),
+					HeartbeatInterval: time.Duration(10) * time.Second,
+					LastWriteTime:     time.Date(2017, 2, 11, 14, 0, 0, 0, time.UTC),
+					LastUpdateTime:    time.Date(2017, 2, 11, 14, 0, 2, 0, time.UTC),
+					Kind:              description.ServerKindMongos,
+					WireVersion:       &description.VersionRange{Min: 6, Max: 21},
+				}
+			}
+			desc := description.Topology{
+				Servers: servers,
+			}
 
-	b.ResetTimer()
-	b.RunParallel(func(p *testing.PB) {
-		b.ReportAllocs()
-		for p.Next() {
-			_ = equalTopologies(desc, desc)
-		}
-	})
+			b.ResetTimer()
+			b.RunParallel(func(p *testing.PB) {
+				b.ReportAllocs()
+				for p.Next() {
+					_ = equalTopologies(desc, desc)
+				}
+			})
+		})
+	}
 }