fix: make sure trace.Spans() consider tree object when flat is nil

kubeshop · Sep 6, 2023 · 74f2b36 · 74f2b36
1 parent 3cce590
commit 74f2b36
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server/traces/trace_entities.go b/server/traces/trace_entities.go
@@ -194,6 +194,11 @@ func (t *Trace) Spans() []Span {
 		return []Span{}
 	}
 
+	if len(t.Flat) == 0 {
+		t.Flat = make(map[trace.SpanID]*Span)
+		flattenSpans(t.Flat, t.RootSpan)
+	}
+
 	spans := make([]Span, 0, len(t.Flat))
 	for _, span := range t.Flat {
 		spans = append(spans, *span)