metadata: add benchmark test for FromIncomingContext and ValueFromInc…

…omingContext (#6117)
grpc · Mar 15, 2023 · 6f44ae8 · 6f44ae8
1 parent a1e657c
commit 6f44ae8
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/metadata/metadata_test.go b/metadata/metadata_test.go
@@ -335,3 +335,21 @@ func BenchmarkFromOutgoingContext(b *testing.B) {
 		FromOutgoingContext(ctx)
 	}
 }
+
+func BenchmarkFromIncomingContext(b *testing.B) {
+	md := Pairs("X-My-Header-1", "42")
+	ctx := NewIncomingContext(context.Background(), md)
+	b.ResetTimer()
+	for n := 0; n < b.N; n++ {
+		FromIncomingContext(ctx)
+	}
+}
+
+func BenchmarkValueFromIncomingContext(b *testing.B) {
+	md := Pairs("X-My-Header-1", "42")
+	ctx := NewIncomingContext(context.Background(), md)
+	b.ResetTimer()
+	for n := 0; n < b.N; n++ {
+		ValueFromIncomingContext(ctx, "x-my-header-1")
+	}
+}