Function makeTestTenantIndex

tempodb/backend/tenantindex_benchmark_test.go:77–124

(numBlocks int)

}

func makeTestTenantIndex(numBlocks int) *TenantIndex {
	const numDistinctDedicatedCols = 10

	var (
		maxSupportedSpanColumns     = maxSupportedColumns[DedicatedColumnTypeString][DedicatedColumnScopeSpan]
		maxSupportedResourceColumns = maxSupportedColumns[DedicatedColumnTypeString][DedicatedColumnScopeResource]
	)

	dedicatedCols := make([]DedicatedColumns, 0, numDistinctDedicatedCols)
	for range numDistinctDedicatedCols {
		num := 0
		cols := make([]DedicatedColumn, 0, maxSupportedSpanColumns+maxSupportedResourceColumns)
		for range maxSupportedSpanColumns {
			num += rand.IntN(10)
			cols = append(cols, DedicatedColumn{
				Scope: DedicatedColumnScopeSpan,
				Name:  fmt.Sprintf("ded-span-%d", num),
				Type:  DedicatedColumnTypeString,
			})
		}
		for range maxSupportedResourceColumns {
			num += rand.IntN(10)
			cols = append(cols, DedicatedColumn{
				Scope: DedicatedColumnScopeResource,
				Name:  fmt.Sprintf("ded-res-%d", num),
				Type:  DedicatedColumnTypeString,
			})
		}
		dedicatedCols = append(dedicatedCols, cols)
	}

	blocks := make([]*BlockMeta, 0, numBlocks)
	compactedBlocks := make([]*CompactedBlockMeta, 0, numBlocks)
	for i := range numBlocks {
		meta := NewBlockMeta("test-tenant", uuid.New(), "vParquet4")
		meta.DedicatedColumns = dedicatedCols[i%numDistinctDedicatedCols]
		blocks = append(blocks, meta)

		compactedMeta := &CompactedBlockMeta{
			BlockMeta:     *NewBlockMeta("test-tenant", uuid.New(), "vParquet4"),
			CompactedTime: time.Now(),
		}
		compactedMeta.DedicatedColumns = dedicatedCols[i%numDistinctDedicatedCols]
		compactedBlocks = append(compactedBlocks, compactedMeta)
	}

	return newTenantIndex(blocks, compactedBlocks)
}