Function benchmarkQueryStream

pkg/ingester/ingester_test.go:4256–4306

(b *testing.B, samplesCount, seriesCount int)

}

func benchmarkQueryStream(b *testing.B, samplesCount, seriesCount int) {
	cfg := defaultIngesterTestConfig(b)

	// Create ingester.
	i, err := prepareIngesterWithBlocksStorage(b, cfg, prometheus.NewRegistry())
	require.NoError(b, err)
	require.NoError(b, services.StartAndAwaitRunning(context.Background(), i))
	defer services.StopAndAwaitTerminated(context.Background(), i) //nolint:errcheck

	// Wait until it's ACTIVE.
	test.Poll(b, 1*time.Second, ring.ACTIVE, func() any {
		return i.lifecycler.GetState()
	})

	// Push series.
	ctx := user.InjectOrgID(context.Background(), userID)

	samples := make([]cortexpb.Sample, 0, samplesCount)

	for i := range samplesCount {
		samples = append(samples, cortexpb.Sample{
			Value:       float64(i),
			TimestampMs: int64(i),
		})
	}

	for s := range seriesCount {
		_, err = i.Push(ctx, writeRequestSingleSeries(labels.FromStrings("__name__", "foo", "l", strconv.Itoa(s)), samples))
		require.NoError(b, err)
	}

	req := &client.QueryRequest{
		StartTimestampMs: 0,
		EndTimestampMs:   int64(samplesCount + 1),

		Matchers: []*client.LabelMatcher{{
			Type:  client.EQUAL,
			Name:  model.MetricNameLabel,
			Value: "foo",
		}},
	}

	mockStream := &mockQueryStreamServer{ctx: ctx}

	b.ReportAllocs()

	for b.Loop() {
		err := i.QueryStream(req, mockStream)
		require.NoError(b, err)
	}
}

func mockWriteRequest(t *testing.T, lbls labels.Labels, value float64, timestampMs int64) (*cortexpb.WriteRequest, *client.QueryStreamResponse) {
	samples := []cortexpb.Sample{