Function autocompleteIter

tempodb/encoding/vparquet4/block_autocomplete.go:268–324

autocompleteIter creates an iterator that will collect values for a given attribute/tag.

(ctx context.Context, tr tagRequest, pf *parquet.File, opts common.SearchOptions, dc backend.DedicatedColumns)

// autocompleteIter creates an iterator that will collect values for a given attribute/tag.
func autocompleteIter(ctx context.Context, tr tagRequest, pf *parquet.File, opts common.SearchOptions, dc backend.DedicatedColumns) (parquetquery.Iterator, error) {
	// categorizeConditions conditions into span-level or resource-level
	catConditions, _, err := categorizeConditions(tr.conditions)
	if err != nil {
		return nil, err
	}

	rgs := rowGroupsFromFile(pf, opts)
	makeIter := makeIterFunc(ctx, rgs, pf)
	makeNilIter := makeNilIterFunc(ctx, rgs, pf)

	var currentIter parquetquery.Iterator

	if len(catConditions.event) > 0 || tr.keysRequested(traceql.AttributeScopeEvent) {
		currentIter, err = createDistinctEventIterator(makeIter, makeNilIter, tr, currentIter, catConditions.event)
		if err != nil {
			return nil, errors.Wrap(err, "creating event iterator")
		}
	}

	if len(catConditions.link) > 0 || tr.keysRequested(traceql.AttributeScopeLink) {
		currentIter, err = createDistinctLinkIterator(makeIter, makeNilIter, tr, currentIter, catConditions.link)
		if err != nil {
			return nil, errors.Wrap(err, "creating link iterator")
		}
	}

	if len(catConditions.span) > 0 || tr.keysRequested(traceql.AttributeScopeSpan) {
		currentIter, err = createDistinctSpanIterator(makeIter, makeNilIter, tr, currentIter, catConditions.span, dc)
		if err != nil {
			return nil, errors.Wrap(err, "creating span iterator")
		}
	}

	if len(catConditions.instrumentation) > 0 || tr.keysRequested(traceql.AttributeScopeInstrumentation) {
		currentIter, err = createDistinctScopeIterator(makeIter, makeNilIter, tr, currentIter, catConditions.instrumentation)
		if err != nil {
			return nil, errors.Wrap(err, "creating instrumentation iterator")
		}
	}

	if len(catConditions.resource) > 0 || tr.keysRequested(traceql.AttributeScopeResource) {
		currentIter, err = createDistinctResourceIterator(makeIter, makeNilIter, tr, currentIter, catConditions.resource, dc)
		if err != nil {
			return nil, errors.Wrap(err, "creating resource iterator")
		}
	}

	if len(catConditions.trace) > 0 {
		currentIter, err = createDistinctTraceIterator(makeIter, tr, currentIter, catConditions.trace)
		if err != nil {
			return nil, errors.Wrap(err, "creating trace iterator")
		}
	}

	return currentIter, nil
}