Method Fetch

tempodb/encoding/vparquet5/wal_block.go:713–758

(ctx context.Context, req traceql.FetchSpansRequest, _ common.SearchOptions)

}

func (b *walBlock) Fetch(ctx context.Context, req traceql.FetchSpansRequest, _ common.SearchOptions) (traceql.FetchSpansResponse, error) {
	ctx, span := tracer.Start(ctx, "walBlock.Fetch")
	defer span.End()

	// todo: this same method is called in backendBlock.Fetch. is there anyway to share this?
	err := checkConditions(req.Conditions)
	if err != nil {
		return traceql.FetchSpansResponse{}, fmt.Errorf("conditions invalid: %w", err)
	}

	var (
		blockFlushes = b.readFlushes()
		readers      = make([]*walReaderAt, 0, len(blockFlushes))
		iters        = make([]traceql.CommonIterator[*traceql.Spanset], 0, len(blockFlushes))
	)

	for _, page := range blockFlushes {
		file, err := page.file(ctx)
		if err != nil {
			return traceql.FetchSpansResponse{}, fmt.Errorf("error opening file %s: %w", page.path, err)
		}

		iter, err := fetch(ctx, req, file.parquetFile, file.parquetFile.RowGroups(), b.meta.DedicatedColumns)
		if err != nil {
			return traceql.FetchSpansResponse{}, fmt.Errorf("creating fetch iter: %w", err)
		}

		iters = append(iters, &pageFileClosingIterator[*traceql.Spanset]{iter: iter, pageFile: file})
		readers = append(readers, file.r)
	}

	return traceql.FetchSpansResponse{
		Results: &mergeIterator[*traceql.Spanset]{
			iters: iters,
		},
		// FIXME: can this be simplified with the common.MetadataCallback?? and Metrics Collector??
		Bytes: func() uint64 {
			// read value when callback is called
			var totalBytesRead uint64
			for _, r := range readers {
				totalBytesRead += r.BytesRead()
			}
			return totalBytesRead
		},
	}, nil
}

func (b *walBlock) FetchSpans(ctx context.Context, req traceql.FetchSpansRequest, _ common.SearchOptions) (traceql.FetchSpansOnlyResponse, error) {
	ctx, span := tracer.Start(ctx, "walBlock.FetchSpans")