Update Cycle operation.

spec/drupol/collection/CollectionSpec.php

@@ -323,16 +323,15 @@ public function it_can_cycle(): void
             ->beConstructedThrough('with', [['1', '2', '3']]);
 
         $this
-            ->cycle(1)
+            ->cycle(3)
             ->shouldIterateAs(['1', '2', '3']);
 
         $this
-            ->cycle(2)
+            ->cycle(6)
             ->shouldIterateAs(['1', '2', '3', '1', '2', '3']);
 
         $this
-            ->cycle()
-            ->limit(7)
+            ->cycle(7)
             ->shouldIterateAs(['1', '2', '3', '1', '2', '3', '1']);
     }
 

src/Collection.php

@@ -137,9 +137,9 @@ public function count(): int
      *
      * @return \drupol\collection\Contract\Collection
      */
-    public function cycle(int $count = 0): BaseInterface
+    public function cycle(int $length = 0): BaseInterface
     {
-        return $this->run(new Cycle($count));
+        return $this->run(new Cycle($length));
     }
 
     /**

src/Contract/Cycleable.php

@@ -10,9 +10,9 @@
 interface Cycleable
 {
     /**
-     * @param int $count
+     * @param int $length
      *
      * @return \drupol\collection\Contract\Collection<mixed>
      */
-    public function cycle(int $count = 0): Base;
+    public function cycle(int $length = 0): Base;
 }

src/Operation/Cycle.php

@@ -6,7 +6,10 @@
 
 use Closure;
 use drupol\collection\Contract\Operation;
+use drupol\collection\Iterator\IterableIterator;
 use Generator;
+use InfiniteIterator;
+use LimitIterator;
 
 /**
  * Class Cycle.
@@ -16,30 +19,36 @@ final class Cycle implements Operation
     /**
      * @var int
      */
-    private $count;
+    private $length;
 
     /**
      * Cycle constructor.
      *
-     * @param int $count
+     * @param int $length
      */
-    public function __construct(int $count = 0)
+    public function __construct(int $length = 0)
     {
-        $this->count = $count;
+        $this->length = $length;
     }
 
     /**
      * {@inheritdoc}
      */
     public function on(iterable $collection): Closure
     {
-        $count = $this->count;
+        $length = $this->length;
 
-        return static function () use ($collection, $count): Generator {
-            for ($j = 0 === $count ? 1 : 0; $j !== $count; ++$j) {
-                foreach ($collection as $value) {
-                    yield $value;
-                }
+        return static function () use ($collection, $length): Generator {
+            $cycleIterator = new LimitIterator(
+                new InfiniteIterator(
+                    new IterableIterator($collection)
+                ),
+                0,
+                $length
+            );
+
+            foreach ($cycleIterator as $value) {
+                yield $value;
             }
         };
     }