mirrors/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-08-05 21:57:09 +02:00
Code Issues Projects Releases Wiki Activity

remote write 2.0: add write handler benchmark for V2 format (#16687)

Browse Source 
Signed-off-by: Sandro Jijavadze <sandrojijavadze@protonmail.com>
This commit is contained in:
buda 2025-07-24 00:51:06 +04:00 committed by GitHub
parent 9b00c296a8
commit f8b3fce845
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 49 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
storage/remote/write_handler_test.go
View File

@ -23,7 +23,6 @@ import (
"net/http"
"net/http/httptest"
"strconv"
"strings"
"testing"
"time"
@ -667,31 +666,58 @@ func TestOutOfOrderHistogram_V1Message(t *testing.T) {
}
func BenchmarkRemoteWriteHandler(b *testing.B) {
const labelValue = "abcdefg'hijlmn234!@#$%^&*()_+~`\"{}[],./<>?hello0123hiOlá你好Dzieńdobry9Zd8ra765v4stvuyte"
var reqs []*http.Request
for i := 0; i < b.N; i++ {
num := strings.Repeat(strconv.Itoa(i), 16)
buf, _, _, err := buildWriteRequest(nil, []prompb.TimeSeries{{
Labels: []prompb.Label{
{Name: "__name__", Value: "test_metric"},
{Name: "test_label_name_" + num, Value: labelValue + num},
},
Histograms: []prompb.Histogram{prompb.FromIntHistogram(0, &testHistogram)},
}}, nil, nil, nil, nil, "snappy")
require.NoError(b, err)
req, err := http.NewRequest("", "", bytes.NewReader(buf))
require.NoError(b, err)
reqs = append(reqs, req)
labelStrings := []string{
"__name__", "test_metric",
"test_label_name", "abcdefg'hijlmn234!@#$%^&*()_+~`\"{}[],./<>?hello0123hiOlá你好Dzieńdobry9Zd8ra765v4stvuyte",
}
v1Labels := prompb.FromLabels(labels.FromStrings(labelStrings...), nil)
appendable := &mockAppendable{}
// TODO: test with other proto format(s)
handler := NewWriteHandler(promslog.NewNopLogger(), nil, appendable, []config.RemoteWriteProtoMsg{config.RemoteWriteProtoMsgV1}, false)
recorder := httptest.NewRecorder()
testCases := []struct {
name string
payloadFunc func() ([]byte, error)
protoFormat config.RemoteWriteProtoMsg
}{
{
name: "V1 Write",
payloadFunc: func() ([]byte, error) {
buf, _, _, err := buildWriteRequest(nil, []prompb.TimeSeries{{
Labels: v1Labels,
Histograms: []prompb.Histogram{prompb.FromIntHistogram(0, &testHistogram)},
}}, nil, nil, nil, nil, "snappy")
return buf, err
},
protoFormat: config.RemoteWriteProtoMsgV1,
},
{
name: "V2 Write",
payloadFunc: func() ([]byte, error) {
buf, _, _, err := buildV2WriteRequest(promslog.NewNopLogger(), []writev2.TimeSeries{{
LabelsRefs: []uint32{0, 1, 2, 3},
Histograms: []writev2.Histogram{writev2.FromIntHistogram(0, &testHistogram)},
}}, labelStrings,
nil, nil, nil, "snappy")
return buf, err
},
protoFormat: config.RemoteWriteProtoMsgV2,
},
}
for _, tc := range testCases {
b.Run(tc.name, func(b *testing.B) {
appendable := &mockAppendable{}
handler := NewWriteHandler(promslog.NewNopLogger(), nil, appendable, []config.RemoteWriteProtoMsg{tc.protoFormat}, false)
b.ResetTimer()
for i := 0; i < b.N; i++ {
b.StopTimer()
buf, err := tc.payloadFunc()
require.NoError(b, err)
req, err := http.NewRequest("", "", bytes.NewReader(buf))
require.NoError(b, err)
b.StartTimer()
b.ResetTimer()
for _, req := range reqs {
handler.ServeHTTP(recorder, req)
recorder := httptest.NewRecorder()
handler.ServeHTTP(recorder, req)
}
})
}
}