Function BenchmarkHandleRPC_InPayload

middleware/grpc_stats_bench_test.go:36–54

BenchmarkHandleRPC_InPayload measures the per-message hot path for incoming payloads on a streaming RPC (the dominant case in QueryStream).

(b *testing.B)

// BenchmarkHandleRPC_InPayload measures the per-message hot path for incoming
// payloads on a streaming RPC (the dominant case in QueryStream).
func BenchmarkHandleRPC_InPayload(b *testing.B) {
	h := newBenchStatsHandler(b)
	const method = "/cortex.Querier/QueryStream"

	// Simulate stream open: TagRPC stores the method name in ctx.
	ctx := h.TagRPC(context.Background(), &stats.RPCTagInfo{FullMethodName: method})

	// Simulate Begin (stream open).
	h.HandleRPC(ctx, &stats.Begin{})

	inPayload := &stats.InPayload{WireLength: 4096}

	b.ResetTimer()
	b.ReportAllocs()

	for i := 0; i < b.N; i++ {
		h.HandleRPC(ctx, inPayload)
	}
}

// BenchmarkHandleRPC_OutPayload measures the per-message hot path for outgoing
// payloads (server-side streaming responses).