support v2 proto for BenchmarkSampleSend

Signed-off-by: Callum Styan <callumstyan@gmail.com>
This commit is contained in:
Callum Styan 2024-09-18 10:59:21 -07:00
parent b6107cc888
commit 161b79c75e

View file

@ -1326,7 +1326,9 @@ func BenchmarkSampleSend(b *testing.B) {
cfg.MaxShards = 20
// todo: test with new proto type(s)
m := newTestQueueManager(b, cfg, mcfg, defaultFlushDeadline, c, config.RemoteWriteProtoMsgV1)
for _, format := range []config.RemoteWriteProtoMsg{config.RemoteWriteProtoMsgV1, config.RemoteWriteProtoMsgV2} {
b.Run(string(format), func(b *testing.B) {
m := newTestQueueManager(b, cfg, mcfg, defaultFlushDeadline, c, format)
m.StoreSeries(series, 0)
// These should be received by the client.
@ -1341,6 +1343,8 @@ func BenchmarkSampleSend(b *testing.B) {
}
// Do not include shutdown
b.StopTimer()
})
}
}
// Check how long it takes to add N series, including external labels processing.