Function BenchmarkIterators

tempodb/encoding/vparquet4/block_traceql_test.go:1538–1586

BenchmarkIterators is a convenient method to run benchmarks on various iterator constructions directly when working on optimizations. Replace the iterator at the beginning of the benchmark loop with any combination desired.

(b *testing.B)

// BenchmarkIterators is a convenient method to run benchmarks on various iterator constructions directly when working on optimizations.
// Replace the iterator at the beginning of the benchmark loop with any combination desired.
func BenchmarkIterators(b *testing.B) {
	ctx := context.TODO()
	opts := common.DefaultSearchOptions()
	opts.StartPage = 3
	opts.TotalPages = 2

	block := blockForBenchmarks(b)
	pf, _, err := block.openForSearch(ctx, opts)
	require.NoError(b, err)

	rgs := pf.RowGroups()
	rgs = rgs[3:5]

	var instrPred *parquetquery.InstrumentedPredicate
	makeIterInternal := makeIterFunc(ctx, rgs, pf)
	makeIter := func(columnName string, predicate parquetquery.Predicate, selectAs string) parquetquery.Iterator {
		instrPred = &parquetquery.InstrumentedPredicate{
			Pred: predicate,
		}

		return makeIterInternal(columnName, predicate, selectAs)
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		iter := makeIter(columnPathSpanAttrKey, parquetquery.NewSubstringPredicate("e"), "foo")
		count := 0
		for {
			res, err := iter.Next()
			if err != nil {
				panic(err)
			}
			if res == nil {
				break
			}
			count++
		}
		iter.Close()
		if instrPred != nil {
			b.ReportMetric(float64(count), "count")
			b.ReportMetric(float64(instrPred.InspectedColumnChunks), "stats_cc")
			b.ReportMetric(float64(instrPred.KeptColumnChunks), "stats_cc_kept")
			b.ReportMetric(float64(instrPred.InspectedPages), "stats_ip")
			b.ReportMetric(float64(instrPred.KeptPages), "stats_ip_kept")
			b.ReportMetric(float64(instrPred.InspectedValues), "stats_v")
			b.ReportMetric(float64(instrPred.KeptValues), "stats_v_kept")
		}
	}
}

func BenchmarkBackendBlockQueryRange(b *testing.B) {
	testCases := []string{