Method Search

modules/livestore/instance_search.go:193–297

(ctx context.Context, req *tempopb.SearchRequest)

}

func (i *instance) Search(ctx context.Context, req *tempopb.SearchRequest) (*tempopb.SearchResponse, error) {
	ctx, span := tracer.Start(ctx, "instance.Search")
	defer span.End()

	maxResults := int(req.Limit)
	// if limit is not set, use a safe default
	if maxResults == 0 {
		maxResults = 20
	}

	span.AddEvent("SearchRequest", oteltrace.WithAttributes(attribute.String("request", req.String())))

	mostRecent := false
	if len(req.Query) > 0 {
		rootExpr, err := traceql.ParseNoOptimizations(req.Query)
		if err != nil {
			return nil, fmt.Errorf("error parsing query: %w", err)
		}

		ok := false
		if mostRecent, ok = rootExpr.Hints.GetBool(traceql.HintMostRecent, false); !ok {
			mostRecent = false
		}
	}

	var (
		resultsMtx = sync.Mutex{}
		combiner   = traceql.NewMetadataCombiner(maxResults, mostRecent)
		metrics    = &tempopb.SearchMetrics{}
		opts       = common.DefaultSearchOptions()
	)

	search := func(ctx context.Context, blockMeta *backend.BlockMeta, b block) error {
		var resp *tempopb.SearchResponse
		var err error

		// if the combiner is complete for the block's end time, we can skip searching it
		if combiner.IsCompleteFor(uint32(blockMeta.EndTime.Unix())) {
			return errComplete
		}

		if api.IsTraceQLQuery(req) {
			f := traceql.NewSpansetFetcherWrapperBoth(
				func(ctx context.Context, req traceql.FetchSpansRequest) (traceql.FetchSpansResponse, error) {
					return b.Fetch(ctx, req, opts)
				},
				func(ctx context.Context, req traceql.FetchSpansRequest) (traceql.FetchSpansOnlyResponse, error) {
					return b.FetchSpans(ctx, req, opts)
				},
			)
			// note: we are creating new engine for each wal block,
			// and engine.ExecuteSearch is parsing the query for each block
			var searchOpts []traceql.CompileOption
			if i.overrides.UnsafeQueryHints(i.tenantID) {
				searchOpts = append(searchOpts, traceql.WithUnsafeHints(true))
			}
			for _, name := range req.SkipASTTransformations {
				searchOpts = append(searchOpts, traceql.WithSkipOptimization(name))