Function BenchmarkSearch

tempodb/wal/wal_test.go:470–501

(b *testing.B)

}

func BenchmarkSearch(b *testing.B) {
	for _, enc := range encoding.AllEncodingsForWrites() {
		version := enc.Version()
		b.Run(version, func(b *testing.B) {
			runWALBenchmark(b, version, 1, func(ids [][]byte, objs []*tempopb.Trace, block common.WALBlock) {
				ctx := context.Background()

				for i := 0; i < b.N; i++ {
					j := i % len(ids)
					id, o := ids[j], objs[j]

					k, v := findFirstAttribute(o)
					require.NotEmpty(b, k)
					require.NotEmpty(b, v)

					resp, err := block.Search(ctx, &tempopb.SearchRequest{
						Tags: map[string]string{
							k: v,
						},
						Limit: 10,
					}, common.DefaultSearchOptions())
					if errors.Is(err, util.ErrUnsupported) {
						return
					}
					require.NoError(b, err)
					require.Equal(b, 1, len(resp.Traces))
					require.Equal(b, util.TraceIDToHexString(id), resp.Traces[0].TraceID)
				}
			})
		})
	}
}

func runWALBenchmark(b *testing.B, encoding string, flushCount int, runner func([][]byte, []*tempopb.Trace, common.WALBlock)) {
	runWALBenchmarkWithAppendMode(b, encoding, flushCount, false, runner)