Add implementation of Countable for CachingIterator

doctrine · Jun 27, 2022 · 5224b08 · 5224b08
1 parent 8f968ba
commit 5224b08
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/lib/Doctrine/ODM/MongoDB/Iterator/CachingIterator.php b/lib/Doctrine/ODM/MongoDB/Iterator/CachingIterator.php
@@ -4,11 +4,13 @@
 
 namespace Doctrine\ODM\MongoDB\Iterator;
 
+use Countable;
 use Generator;
 use ReturnTypeWillChange;
 use RuntimeException;
 use Traversable;
 
+use function count;
 use function current;
 use function key;
 use function next;
@@ -26,7 +28,7 @@
  * @template TValue
  * @template-implements Iterator<TValue>
  */
-final class CachingIterator implements Iterator
+final class CachingIterator implements Countable, Iterator
 {
     /** @var array<mixed, TValue> */
     private $items = [];
@@ -55,6 +57,18 @@ public function __construct(Traversable $iterator)
         $this->storeCurrentItem();
     }
 
+    /**
+     * @see https://php.net/countable.count
+     * @return integer
+     */
+    #[ReturnTypeWillChange]
+    public function count()
+    {
+        $this->exhaustIterator();
+
+        return count($this->items);
+    }
+
     public function __destruct()
     {
         $this->iterator = null;