Method validateTraceSearch

cmd/tempo-vulture/validation.go:321–379

(
	_ context.Context,
	writtenTrace *util.TraceInfo,
	actualTrace *tempopb.Trace,
	httpClient httpclient.TempoHTTPClient,
)

}

func (vs *ValidationService) validateTraceSearch(
	_ context.Context,
	writtenTrace *util.TraceInfo,
	actualTrace *tempopb.Trace,
	httpClient httpclient.TempoHTTPClient,
) error {
	vs.logTraceAttributes(actualTrace, writtenTrace.HexID())

	// Get a random attribute from the expected trace
	attr := util.RandomAttrFromTrace(actualTrace)
	if attr == nil {
		return fmt.Errorf("no searchable attribute found in trace")
	}

	searchQuery := fmt.Sprintf(`{.%s = "%s"}`, attr.Key, util.StringifyAnyValue(attr.Value))
	vs.logger.Info("Searching for trace",
		zap.String("traceID", writtenTrace.HexID()),
		zap.String("searchQuery", searchQuery),
	)

	start := writtenTrace.Timestamp().Add(-30 * time.Minute).Unix()
	end := writtenTrace.Timestamp().Add(30 * time.Minute).Unix()

	searchResp, searchErr := httpClient.SearchTraceQLWithRange(searchQuery, start, end)
	if searchErr != nil {
		return fmt.Errorf("search API failed: %w", searchErr)
	}

	vs.logger.Info("Search response received",
		zap.Int("traces_found", len(searchResp.Traces)),
	)

	// Check if our trace ID is in the search results
	found := false
	for _, trace := range searchResp.Traces {
		vs.logger.Info("written trace", zap.String("trace_id", writtenTrace.HexID()))
		vs.logger.Info("found trace", zap.String("trace_id", trace.TraceID))

		equal, err := util.EqualHexStringTraceIDs(writtenTrace.HexID(), trace.TraceID)
		if err != nil {
			return fmt.Errorf("error comparing trace IDs: %w", err)
		}
		if !equal {
			return fmt.Errorf("trace IDs do not match")
		}

		if equal {
			found = true
			break
		}
	}

	if found {
		vs.logger.Info("Found trace via search")
	} else {
		return fmt.Errorf("trace not found in search results")
	}
	return nil