IWF-124: Add filtering for NewOutputCollector

indeedeng · Dec 18, 2024 · 4748f34 · 4748f34
1 parent f2825b4
commit 4748f34
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/service/interpreter/outputCollector.go b/service/interpreter/outputCollector.go
@@ -7,11 +7,22 @@ type OutputCollector struct {
 }
 
 func NewOutputCollector(initOutputs []iwfidl.StateCompletionOutput) *OutputCollector {
+	filteredOutputs := []iwfidl.StateCompletionOutput{}
+
 	if initOutputs == nil {
-		initOutputs = []iwfidl.StateCompletionOutput{}
-	}
-	return &OutputCollector{
-		outputs: initOutputs,
+		return &OutputCollector{
+			outputs: filteredOutputs,
+		}
+	} else {
+		for _, output := range initOutputs {
+			if output.CompletedStateOutput != nil {
+				filteredOutputs = append(filteredOutputs, output)
+			}
+		}
+
+		return &OutputCollector{
+			outputs: filteredOutputs,
+		}
 	}
 }