From 2228d7ce2650ba4cc5811a761d9f54e09ea6d99e Mon Sep 17 00:00:00 2001
From: Maciej Borsz <maciejborsz@google.com>
Date: Wed, 17 Feb 2021 15:03:53 +0100
Subject: [PATCH] Add BenchmarkSerializeObject* benchmarks

---
 .../handlers/responsewriters/writers_test.go  | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/staging/src/k8s.io/apiserver/pkg/endpoints/handlers/responsewriters/writers_test.go b/staging/src/k8s.io/apiserver/pkg/endpoints/handlers/responsewriters/writers_test.go
index 3b2b4097024..266e4edc724 100644
--- a/staging/src/k8s.io/apiserver/pkg/endpoints/handlers/responsewriters/writers_test.go
+++ b/staging/src/k8s.io/apiserver/pkg/endpoints/handlers/responsewriters/writers_test.go
@@ -366,6 +366,45 @@ func TestSerializeObject(t *testing.T) {
 	}
 }
 
+func benchmarkSerializeObject(b *testing.B, size int) {
+	largePayload := bytes.Repeat([]byte("0123456789abcdef"), size/16+1)
+	req := &http.Request{
+		Header: http.Header{
+			"Accept-Encoding": []string{"gzip"},
+		},
+		URL: &url.URL{Path: "/path"},
+	}
+	defer featuregatetesting.SetFeatureGateDuringTest(b, utilfeature.DefaultFeatureGate, features.APIResponseCompression, true)()
+
+	encoder := &fakeEncoder{
+		buf: largePayload,
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		recorder := httptest.NewRecorder()
+		SerializeObject("application/json", encoder, recorder, req, http.StatusOK, nil /* object */)
+		result := recorder.Result()
+		if result.StatusCode != http.StatusOK {
+			b.Fatalf("incorrect status code: got %v;  want: %v", result.StatusCode, http.StatusOK)
+		}
+	}
+}
+
+func BenchmarkSerializeObject10KB(b *testing.B) {
+	benchmarkSerializeObject(b, 10*1024)
+}
+
+func BenchmarkSerializeObject10MB(b *testing.B) {
+	benchmarkSerializeObject(b, 10*1024*1024)
+}
+func BenchmarkSerializeObject100MB(b *testing.B) {
+	benchmarkSerializeObject(b, 100*1024*1024)
+}
+func BenchmarkSerializeObject1GB(b *testing.B) {
+	benchmarkSerializeObject(b, 1024*1024*1024)
+}
+
 type fakeResponseRecorder struct {
 	*httptest.ResponseRecorder
 	fe                 *fakeEncoder