Update Reverse operation.

loophp · Jul 6, 2020 · a9b6c9a · a9b6c9a
1 parent 0bd0e42
commit a9b6c9a
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/Operation/Reverse.php b/src/Operation/Reverse.php
@@ -8,16 +8,20 @@
 use Generator;
 use loophp\collection\Contract\Operation;
 use loophp\collection\Transformation\All;
+use loophp\collection\Transformation\Run;
+use loophp\collection\Transformation\Transform;
 
 final class Reverse extends AbstractOperation implements Operation
 {
     public function __invoke(): Closure
     {
         return static function (iterable $collection): Generator {
-            $all = (new All())($collection);
+            $all = (new Transform(new All()))((new Run(new Wrap()))($collection));
 
-            for (end($all); null !== ($key = key($all)); prev($all)) {
-                yield $key => current($all);
+            for (end($all); null !== key($all); prev($all)) {
+                $item = current($all);
+
+                yield key($item) => current($item);
             }
         };
     }