Function blockForBenchmarks

tempodb/encoding/vparquet4/block_traceql_test.go:2643–2671

(b testing.TB)

}

func blockForBenchmarks(b testing.TB) *backendBlock {
	id, ok := os.LookupEnv("BENCH_BLOCKID")
	if !ok {
		b.Fatal("BENCH_BLOCKID is not set. These benchmarks are designed to run against a block on local disk. Set BENCH_BLOCKID to the guid of the block to run benchmarks against. e.g. `export BENCH_BLOCKID=030c8c4f-9d47-4916-aadc-26b90b1d2bc4`")
	}

	path, ok := os.LookupEnv("BENCH_PATH")
	if !ok {
		b.Fatal("BENCH_PATH is not set. These benchmarks are designed to run against a block on local disk. Set BENCH_PATH to the root of the backend such that the block to benchmark is at <BENCH_PATH>/<BENCH_TENANTID>/<BENCH_BLOCKID>.")
	}

	tenantID, ok := os.LookupEnv("BENCH_TENANTID")

	if !ok {
		tenantID = "1"
	}

	blockID := uuid.MustParse(id)
	r, _, _, err := local.New(&local.Config{
		Path: path,
	})
	require.NoError(b, err)

	rr := backend.NewReader(r)
	meta, err := rr.BlockMeta(context.Background(), blockID, tenantID)
	require.NoError(b, err)

	return newBackendBlock(meta, rr)
}