Add OTLP write handler benchmark

Signed-off-by: SungJin1212 <[email protected]>

Author: SungJin1212

pkg/util/push/otlp_test.go

@@ -220,13 +220,133 @@ func TestOTLPConvertToPromTS(t *testing.T) {
 	}
 }
 
+func BenchmarkOTLPWriteHandler(b *testing.B) {
+	cfg := distributor.OTLPConfig{
+		ConvertAllAttributes: false,
+		DisableTargetInfo:    false,
+	}
+	overrides, err := validation.NewOverrides(querier.DefaultLimitsConfig(), nil)
+	require.NoError(b, err)
+
+	exportRequest := generateOTLPWriteRequest()
+	mockPushFunc := func(context.Context, *cortexpb.WriteRequest) (*cortexpb.WriteResponse, error) {
+		return &cortexpb.WriteResponse{}, nil
+	}
+	handler := OTLPHandler(10000, overrides, cfg, nil, mockPushFunc)
+
+	b.Run("json with no compression", func(b *testing.B) {
+		ctx := context.Background()
+		ctx = user.InjectOrgID(ctx, "user-1")
+		body, err := exportRequest.MarshalJSON()
+		require.NoError(b, err)
+		buf := bytes.NewBuffer(body) // To reuse body
+
+		req, err := http.NewRequestWithContext(ctx, "", "", bytes.NewReader(buf.Bytes()))
+		require.NoError(b, err)
+		req.Header.Set("Content-Type", jsonContentType)
+
+		b.ResetTimer()
+		b.ReportAllocs()
+		for i := 0; i < b.N; i++ {
+			recorder := httptest.NewRecorder()
+			handler.ServeHTTP(recorder, req)
+
+			resp := recorder.Result()
+			require.Equal(b, http.StatusOK, resp.StatusCode)
+			req.Body = io.NopCloser(bytes.NewReader(buf.Bytes()))
+		}
+	})
+	b.Run("json with gzip", func(b *testing.B) {
+		ctx := context.Background()
+		ctx = user.InjectOrgID(ctx, "user-1")
+		body, err := exportRequest.MarshalJSON()
+		require.NoError(b, err)
+
+		var gzipBody bytes.Buffer
+		gz := gzip.NewWriter(&gzipBody)
+		_, err = gz.Write(body)
+		require.NoError(b, err)
+		require.NoError(b, gz.Close())
+		buf := bytes.NewBuffer(gzipBody.Bytes()) // To reuse body
+
+		req, err := http.NewRequestWithContext(ctx, "", "", bytes.NewReader(buf.Bytes()))
+		require.NoError(b, err)
+		req.Header.Set("Content-Type", jsonContentType)
+		req.Header.Set("Content-Encoding", "gzip")
+
+		b.ResetTimer()
+		b.ReportAllocs()
+		for i := 0; i < b.N; i++ {
+			recorder := httptest.NewRecorder()
+			handler.ServeHTTP(recorder, req)
+
+			resp := recorder.Result()
+			require.Equal(b, http.StatusOK, resp.StatusCode)
+			req.Body = io.NopCloser(bytes.NewReader(buf.Bytes()))
+		}
+	})
+	b.Run("proto with no compression", func(b *testing.B) {
+		ctx := context.Background()
+		ctx = user.InjectOrgID(ctx, "user-1")
+		body, err := exportRequest.MarshalProto()
+		require.NoError(b, err)
+		buf := bytes.NewBuffer(body) // To reuse body
+
+		req, err := http.NewRequestWithContext(ctx, "", "", bytes.NewReader(buf.Bytes()))
+		require.NoError(b, err)
+		req.Header.Set("Content-Type", pbContentType)
+
+		b.ResetTimer()
+		b.ReportAllocs()
+		for i := 0; i < b.N; i++ {
+
+			recorder := httptest.NewRecorder()
+			handler.ServeHTTP(recorder, req)
+
+			resp := recorder.Result()
+			require.Equal(b, http.StatusOK, resp.StatusCode)
+			req.Body = io.NopCloser(bytes.NewReader(buf.Bytes()))
+		}
+	})
+	b.Run("proto with gzip", func(b *testing.B) {
+		ctx := context.Background()
+		ctx = user.InjectOrgID(ctx, "user-1")
+		body, err := exportRequest.MarshalProto()
+		require.NoError(b, err)
+
+		var gzipBody bytes.Buffer
+		gz := gzip.NewWriter(&gzipBody)
+		_, err = gz.Write(body)
+		require.NoError(b, err)
+		require.NoError(b, gz.Close())
+		buf := bytes.NewBuffer(gzipBody.Bytes()) // To reuse body
+
+		req, err := http.NewRequestWithContext(ctx, "", "", bytes.NewReader(buf.Bytes()))
+		require.NoError(b, err)
+		req.Header.Set("Content-Type", pbContentType)
+		req.Header.Set("Content-Encoding", "gzip")
+
+		b.ResetTimer()
+		b.ReportAllocs()
+		for i := 0; i < b.N; i++ {
+
+			recorder := httptest.NewRecorder()
+			handler.ServeHTTP(recorder, req)
+
+			resp := recorder.Result()
+			require.Equal(b, http.StatusOK, resp.StatusCode)
+			req.Body = io.NopCloser(bytes.NewReader(buf.Bytes()))
+		}
+	})
+}
+
 func TestOTLPWriteHandler(t *testing.T) {
 	cfg := distributor.OTLPConfig{
 		ConvertAllAttributes: false,
 		DisableTargetInfo:    false,
 	}
 
-	exportRequest := generateOTLPWriteRequest(t)
+	exportRequest := generateOTLPWriteRequest()
 
 	tests := []struct {
 		description        string
@@ -368,7 +488,7 @@ func TestOTLPWriteHandler(t *testing.T) {
 	}
 }
 
-func generateOTLPWriteRequest(t *testing.T) pmetricotlp.ExportRequest {
+func generateOTLPWriteRequest() pmetricotlp.ExportRequest {
 	d := pmetric.NewMetrics()
 
 	// Generate One Counter, One Gauge, One Histogram, One Exponential-Histogram