Update Run operation.

loophp · Jul 22, 2020 · 5a5a1bd · 5a5a1bd
1 parent 7c5ee8f
commit 5a5a1bd
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Transformation/Run.php b/src/Transformation/Run.php
@@ -7,6 +7,7 @@
 use loophp\collection\Contract\Operation;
 use loophp\collection\Contract\Transformation;
 use loophp\collection\Iterator\ClosureIterator;
+use loophp\collection\Iterator\IterableIterator;
 
 final class Run implements Transformation
 {
@@ -25,10 +26,16 @@ public function __construct(Operation ...$operations)
      */
     public function __invoke(iterable $collection): ClosureIterator
     {
+        $iterableIterator = new IterableIterator($collection);
+
         return (
             new FoldLeft(
-                static function (iterable $collection, Operation $operation): ClosureIterator {
-                    return new ClosureIterator($operation(), $collection, ...array_values($operation->getArguments()));
+                static function (iterable $collection, Operation $operation) use ($iterableIterator): ClosureIterator {
+                    return new ClosureIterator(
+                        $operation(),
+                        $iterableIterator,
+                        ...array_values($operation->getArguments())
+                    );
                 },
                 $collection
             )