feat(stage context): Get all stage results in a list

spinnaker · Sep 29, 2017 · cda9bae · cda9bae
1 parent 8d8dc93
commit cda9bae
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/orca-core/src/main/groovy/com/netflix/spinnaker/orca/pipeline/model/StageContext.java b/orca-core/src/main/groovy/com/netflix/spinnaker/orca/pipeline/model/StageContext.java
@@ -16,11 +16,14 @@
 
 package com.netflix.spinnaker.orca.pipeline.model;
 
+import com.google.common.collect.ForwardingMap;
+
+import javax.annotation.Nullable;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.Optional;
-import javax.annotation.Nullable;
-import com.google.common.collect.ForwardingMap;
+import java.util.stream.Collectors;
 
 public class StageContext extends ForwardingMap<String, Object> {
 
@@ -53,4 +56,16 @@ public StageContext(Stage<?> stage) {
         );
     }
   }
+
+  /*
+   * Gets all objects matching 'key', sorted by proximity to the current stage
+   */
+  public List<Object> getAll(Object key) {
+    return stage
+        .ancestors()
+        .stream()
+        .filter(it -> it.getOutputs().containsKey(key))
+        .map(it -> it.getOutputs().get(key))
+        .collect(Collectors.toList());
+  }
 }