optimizations: Optimize union steps

tontinton · Dec 6, 2024 · 5a9ff8e · 5a9ff8e
1 parent 3443539
commit 5a9ff8e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/optimizations/mod.rs b/src/optimizations/mod.rs
@@ -95,6 +95,15 @@ impl Optimizer {
             }
         }
 
+        // Don't forget to optimize union steps too!
         steps
+            .into_iter()
+            .map(|mut step| {
+                if let WorkflowStep::Union(ref mut workflow) = step {
+                    workflow.steps = self.optimize(workflow.steps.clone());
+                }
+                step
+            })
+            .collect()
     }
 }
diff --git a/src/workflow/mod.rs b/src/workflow/mod.rs
@@ -73,7 +73,7 @@ pub enum WorkflowStep {
 
 #[derive(Debug, Clone)]
 pub struct Workflow {
-    steps: Vec<WorkflowStep>,
+    pub steps: Vec<WorkflowStep>,
 }
 
 fn rx_stream(mut rx: mpsc::Receiver<Log>) -> LogStream {