Method FindTraceByID

modules/querier/querier.go:180–313

FindTraceByID implements tempopb.Querier.

(ctx context.Context, req *tempopb.TraceByIDRequest, timeStart int64, timeEnd int64)

// FindTraceByID implements tempopb.Querier.
func (q *Querier) FindTraceByID(ctx context.Context, req *tempopb.TraceByIDRequest, timeStart int64, timeEnd int64) (*tempopb.TraceByIDResponse, error) {
	if !validation.ValidTraceID(req.TraceID) {
		return nil, errors.New("invalid trace id")
	}

	userID, err := validation.ExtractValidTenantID(ctx)
	if err != nil {
		return nil, fmt.Errorf("error extracting org id in Querier.FindTraceByID: %w", err)
	}

	ctx, span := tracer.Start(ctx, "Querier.FindTraceByID")
	defer span.End()

	span.SetAttributes(attribute.String("queryMode", req.QueryMode))

	maxBytes := q.limits.MaxBytesPerTrace(userID)
	combiner := trace.NewCombiner(maxBytes, req.AllowPartialTrace)
	var inspectedBytes uint64

	if req.QueryMode == QueryModeIngesters || req.QueryMode == QueryModeAll {
		// Get responses from all live stores in parallel.
		span.AddEvent("searching live-stores")
		forEach := func(funcCtx context.Context, client tempopb.QuerierClient) (any, error) {
			return client.FindTraceByID(funcCtx, req)
		}
		partialTraces, err := q.forLiveStoreRing(ctx, forEach)
		if err != nil {
			return nil, fmt.Errorf("error querying live-stores in Querier.FindTraceByID: %w", err)
		}

		var spanCountTotal, traceCountTotal int64
		found := false

		for _, partialTrace := range partialTraces {
			resp := partialTrace.(*tempopb.TraceByIDResponse)
			if resp.Trace == nil {
				continue
			}

			found = true

			spanCount, err := combiner.Consume(resp.Trace)
			if err != nil {
				return nil, fmt.Errorf("error combining live-store results in Querier.FindTraceByID: %w", err)
			}

			spanCountTotal += int64(spanCount)
			traceCountTotal++
			if resp.Metrics != nil {
				inspectedBytes += resp.Metrics.InspectedBytes
			}
		}

		span.AddEvent("done searching live-stores", oteltrace.WithAttributes(
			attribute.Bool("found", found),
			attribute.Int64("combinedSpans", spanCountTotal),
			attribute.Int64("combinedTraces", traceCountTotal),
		))