Function aggregateGenericAttributes

cmd/tempo-cli/cmd-analyse-block.go:668–761

(pf *parquet.File, definitionLevel int, keyPath string, valuePath string, intPath string, isArrayPath string)

}

func aggregateGenericAttributes(pf *parquet.File, definitionLevel int, keyPath string, valuePath string, intPath string, isArrayPath string) (attributeSummary, error) {
	makeIter := makeIterFunc(context.Background(), pf)

	required := []parquetquery.Iterator{
		makeIter(keyPath, parquetquery.NewSkipNilsPredicate(), "key"),
	}
	optional := []parquetquery.Iterator{
		makeIter(valuePath, parquetquery.NewSkipNilsPredicate(), "value"),
		makeIter(intPath, parquetquery.NewSkipNilsPredicate(), "int"),
	}
	if isArrayPath != "" {
		optional = append(optional, makeIter(isArrayPath, parquetquery.NewSkipNilsPredicate(), "isArray"))
	}

	attrIter, err := parquetquery.NewLeftJoinIterator(definitionLevel, required, optional, &attrStatsCollector{})
	if err != nil {
		return attributeSummary{}, err
	}
	defer attrIter.Close()

	var (
		attributes             = make(map[string]*stringAttributeSummary, 1000)
		stringArrayAttributes  = make(map[string]*stringAttributeSummary, 1000)
		integerAttributes      = make(map[string]*integerAttributeSummary, 1000)
		integerArrayAttributes = make(map[string]*integerAttributeSummary, 1000)
	)

	getString := func(name string, from map[string]*stringAttributeSummary) *stringAttributeSummary {
		v, ok := from[name]
		if !ok {
			v = &stringAttributeSummary{
				name:        name,
				cardinality: make(cardinality),
			}
			from[name] = v
		}
		return v
	}

	getInt := func(name string, from map[string]*integerAttributeSummary) *integerAttributeSummary {
		v, ok := from[name]
		if !ok {
			v = &integerAttributeSummary{
				name: name,
			}
			from[name] = v
		}
		return v
	}

	for res, err := attrIter.Next(); res != nil; res, err = attrIter.Next() {
		if err != nil {
			return attributeSummary{}, err
		}

		for _, e := range res.OtherEntries {
			stats, ok := e.Value.(*attrStats)
			if !ok {