From 111bf0fcaf73ef7d9c0749e9b99eb5a219e6cf02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicol=C3=A1s=20Pazos?= <npazosmendez@gmail.com>
Date: Mon, 2 Oct 2023 17:02:48 -0300
Subject: [PATCH] add basic reduce remote write handler benchmark

---
 storage/remote/write_handler_test.go | 30 ++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/storage/remote/write_handler_test.go b/storage/remote/write_handler_test.go
index 20059474e..ec04985a0 100644
--- a/storage/remote/write_handler_test.go
+++ b/storage/remote/write_handler_test.go
@@ -182,6 +182,36 @@ func BenchmarkRemoteWritehandler(b *testing.B) {
 	}
 }
 
+// TODO(npazosmendez): add benchmarks with realistc scenarios
+func BenchmarkReducedRemoteWriteHandler(b *testing.B) {
+	const labelValue = "abcdefg'hijlmn234!@#$%^&*()_+~`\"{}[],./<>?hello0123hiOlá你好Dzieńdobry9Zd8ra765v4stvuyte"
+	reqs := []*http.Request{}
+	for i := 0; i < b.N; i++ {
+		pool := newLookupPool()
+		num := strings.Repeat(strconv.Itoa(i), 16)
+		buf, _, err := buildReducedWriteRequest([]prompb.ReducedTimeSeries{{
+			Labels: []prompb.LabelRef{
+				{NameRef: pool.intern("__name__"), ValueRef: pool.intern("test_metric")},
+				{NameRef: pool.intern("test_label_name_" + num), ValueRef: pool.intern(labelValue + num)},
+			},
+			Histograms: []prompb.Histogram{HistogramToHistogramProto(0, &testHistogram)},
+		}}, pool.getTable(), nil, nil)
+		require.NoError(b, err)
+		req, err := http.NewRequest("", "", bytes.NewReader(buf))
+		require.NoError(b, err)
+		reqs = append(reqs, req)
+	}
+
+	appendable := &mockAppendable{}
+	handler := NewWriteHandler(log.NewNopLogger(), nil, appendable, true)
+	recorder := httptest.NewRecorder()
+
+	b.ResetTimer()
+	for _, req := range reqs {
+		handler.ServeHTTP(recorder, req)
+	}
+}
+
 func TestCommitErr(t *testing.T) {
 	buf, _, err := buildWriteRequest(writeRequestFixture.Timeseries, nil, nil, nil)
 	require.NoError(t, err)