Add benchmark

substratusai · Jan 8, 2024 · d57436a · d57436a
1 parent b023ebc
commit d57436a
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/pkg/endpoints/endponts_bench_test.go b/pkg/endpoints/endponts_bench_test.go
@@ -0,0 +1,20 @@
+package endpoints
+
+import (
+	"context"
+	"testing"
+)
+
+func BenchmarkEndpointGroup(b *testing.B) {
+	e := newEndpointGroup()
+	e.setIPs(map[string]struct{}{"10.0.0.1": {}}, map[string]int32{"testPort": 1})
+	b.ResetTimer()
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			_, err := e.getBestHost(context.Background(), "testPort")
+			if err != nil {
+				b.Fatal(err)
+			}
+		}
+	})
+}