Method FindTraceByID

tempodb/encoding/vparquet4/wal_block.go:544–597

(ctx context.Context, id common.ID, opts common.SearchOptions)

}

func (b *walBlock) FindTraceByID(ctx context.Context, id common.ID, opts common.SearchOptions) (*tempopb.TraceByIDResponse, error) {
	ctx, span := tracer.Start(ctx, "walBlock.FindTraceByID")
	defer span.End()

	metrics := &tempopb.TraceByIDMetrics{}
	trs := make([]*tempopb.Trace, 0)

	for _, page := range b.readFlushes() {
		if rowNumber, ok := page.ids.Get(id); ok {
			file, err := page.file(ctx)
			if err != nil {
				return nil, fmt.Errorf("error opening file %s: %w", page.path, err)
			}

			defer file.Close()
			pf := file.parquetFile

			r := parquet.NewReader(pf)
			defer r.Close()

			err = r.SeekToRow(rowNumber)
			if err != nil {
				return nil, fmt.Errorf("seek to row: %w", err)
			}

			tr := new(Trace)
			err = r.Read(tr)
			if err != nil {
				return nil, fmt.Errorf("error reading row from backend: %w", err)
			}

			trp := ParquetTraceToTempopbTrace(b.meta, tr)

			trs = append(trs, trp)

			metrics.InspectedBytes += file.r.BytesRead()
		}
	}

	combiner := trace.NewCombiner(opts.MaxBytes, false)
	for i, tr := range trs {
		_, err := combiner.ConsumeWithFinal(tr, i == len(trs)-1)
		if err != nil {
			return nil, err
		}
	}

	tr, _ := combiner.Result()
	response := &tempopb.TraceByIDResponse{
		Trace:   tr,
		Metrics: metrics,
	}
	return response, nil
}

func (b *walBlock) Search(ctx context.Context, req *tempopb.SearchRequest, _ common.SearchOptions) (*tempopb.SearchResponse, error) {
	ctx, span := tracer.Start(ctx, "walBlock.Search")