Principal::map() - If we know the callback is one of us, we can use a…

… shortcut
sanmai · Mar 25, 2018 · ed1b434 · ed1b434
1 parent a8e3690
commit ed1b434
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/Pipeline/Principal.php b/src/Pipeline/Principal.php
@@ -43,6 +43,13 @@ public function __construct(\Traversable $input = null)
 
     public function map(callable $func)
     {
+        // If we know the callback is one of us, we can use a shortcut
+        if ($func instanceof self) {
+            $this->pipeline = call_user_func($func);
+
+            return $this;
+        }
+
         if (!$this->pipeline) {
             $this->pipeline = call_user_func($func);
 

diff --git a/tests/Pipeline/SimpleTest.php b/tests/Pipeline/SimpleTest.php
@@ -181,8 +181,7 @@ public function testPipelineInPipeline()
         });
 
         $pipeline2 = new Simple();
-        $pipeline2->map($pipeline1);
-        $pipeline2->filter(function ($i) {
+        $pipeline2->map($pipeline1)->filter(function ($i) {
             return $i % 2 != 0;
         });