Move configure_iterator to PathRoute

pangloss · Sep 28, 2012 · 8aa11cd · 8aa11cd
1 parent 10d3c8a
commit 8aa11cd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/lib/pacer/core/graph/path_route.rb b/lib/pacer/core/graph/path_route.rb
@@ -28,5 +28,17 @@ def subgraph(target_graph = nil, opts = {})
       end
       target_graph
     end
+
+    protected
+
+    def configure_iterator(iter)
+      if respond_to? :graph
+        pipe = Pacer::Pipes::PathWrappingPipe.new(graph)
+        pipe.setStarts iter
+        pipe
+      else
+        iter
+      end
+    end
   end
 end
diff --git a/lib/pacer/transform/path.rb b/lib/pacer/transform/path.rb
@@ -26,16 +26,6 @@ def attach_pipe(end_pipe)
         pipe.setStarts end_pipe if end_pipe
         pipe
       end
-
-      def configure_iterator(iter)
-        if respond_to? :graph
-          pipe = Pacer::Pipes::PathWrappingPipe.new(graph)
-          pipe.setStarts iter
-          pipe
-        else
-          iter
-        end
-      end
     end
   end
 end