Method Fetch

pkg/traceql/engine_test.go:366–404

(ctx context.Context, req FetchTagValuesRequest, cb FetchTagValuesCallback)

}

func (m *MockAutocompleteFetcher) Fetch(ctx context.Context, req FetchTagValuesRequest, cb FetchTagValuesCallback) error {
	rootExpr, err := Parse(m.query)
	if err != nil {
		return err
	}
	if err := rootExpr.validate(); err != nil {
		return err
	}

	for {
		spanset, err := m.iterator.Next(ctx)
		if err != nil && errors.Is(err, io.EOF) {
			return err
		}
		if spanset == nil {
			break
		}
		if len(spanset.Spans) == 0 {
			continue
		}

		evalSS, _ := rootExpr.Pipeline.evaluate([]*Spanset{spanset})

		for _, ss := range evalSS {
			for _, s := range ss.Spans {
				for attr, static := range s.AllAttributes() {
					if attr.Name != req.TagName.Name {
						continue
					}
					if cb(static) {
						return nil
					}
				}
			}
		}
	}

	return nil
}

type MockSpanSetFetcher struct {
	iterator        SpansetIterator