fix(tf): execute extensions before preprocess steps

kubewharf · Aug 10, 2023 · dd4aea7 · dd4aea7
1 parent a68ffa5
commit dd4aea7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/frontend/tf/transform.go b/pkg/frontend/tf/transform.go
@@ -101,10 +101,6 @@ func (transformer *Transformer) Transform(
 		}
 	}
 
-	for _, step := range config.Steps {
-		step.Run(tree)
-	}
-
 	newSpans, err := extensionProcessor.ProcessExtensions(ctx, transformer, config.Extensions, trace.Spans, start, end)
 	if err != nil {
 		return fmt.Errorf("cannot prepare extension trace: %w", err)
@@ -113,6 +109,10 @@ func (transformer *Transformer) Transform(
 	newSpans = append(newSpans, tree.GetSpans()...)
 	tree = tftree.NewSpanTree(newSpans)
 
+	for _, step := range config.Steps {
+		step.Run(tree)
+	}
+
 	trace.Spans = tree.GetSpans()
 
 	return nil