Function BenchmarkHandleRPC_StreamRoundTrip

middleware/grpc_stats_bench_test.go:77–98

BenchmarkHandleRPC_StreamRoundTrip simulates a full streaming round-trip: Begin, N×(InPayload + OutPayload), End — matching QueryStream traffic.

(b *testing.B)

// BenchmarkHandleRPC_StreamRoundTrip simulates a full streaming round-trip:
// Begin, N×(InPayload + OutPayload), End — matching QueryStream traffic.
func BenchmarkHandleRPC_StreamRoundTrip(b *testing.B) {
	h := newBenchStatsHandler(b)
	const method = "/cortex.Querier/QueryStream"
	const msgsPerStream = 50 // typical number of chunks per query

	ctx := h.TagRPC(context.Background(), &stats.RPCTagInfo{FullMethodName: method})

	inPayload := &stats.InPayload{WireLength: 512}
	outPayload := &stats.OutPayload{WireLength: 8192}

	b.ResetTimer()
	b.ReportAllocs()

	for i := 0; i < b.N; i++ {
		h.HandleRPC(ctx, &stats.Begin{})
		for j := 0; j < msgsPerStream; j++ {
			h.HandleRPC(ctx, inPayload)
			h.HandleRPC(ctx, outPayload)
		}
		h.HandleRPC(ctx, &stats.End{})
	}
}