Add #compact_paths to easily get rid of nil payloads.

pangloss · Oct 1, 2012 · 16c5930 · 16c5930
1 parent d8c4ee6
commit 16c5930
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/pacer/core/graph/path_route.rb b/lib/pacer/core/graph/path_route.rb
@@ -30,7 +30,7 @@ def subgraph(target_graph = nil, opts = {})
     end
 
     def payloads
-      map do |path|
+      map element_type: :path do |path|
         path.map do |e|
           if e.is_a? Pacer::Payload::Element
             e.payload
@@ -41,6 +41,12 @@ def payloads
       end
     end
 
+    def compact_paths
+      map element_type: :path do |path|
+        path.compact
+      end
+    end
+
     protected
 
     def configure_iterator(iter)