Make PersistentCollection implement Selectable

Implementing ReadableCollection without Selectable is deprecated

doctrine/collections#518

Author: GromNaN

src/PersistentCollection.php

@@ -5,6 +5,7 @@
 namespace Doctrine\ODM\MongoDB;
 
 use Doctrine\Common\Collections\Collection as BaseCollection;
+use Doctrine\Common\Collections\Selectable;
 use Doctrine\ODM\MongoDB\PersistentCollection\PersistentCollectionInterface;
 use Doctrine\ODM\MongoDB\PersistentCollection\PersistentCollectionTrait;
 

src/PersistentCollection/PersistentCollectionInterface.php

@@ -5,6 +5,7 @@
 namespace Doctrine\ODM\MongoDB\PersistentCollection;
 
 use Doctrine\Common\Collections\Collection;
+use Doctrine\Common\Collections\Selectable;
 use Doctrine\ODM\MongoDB\DocumentManager;
 use Doctrine\ODM\MongoDB\MongoDBException;
 use Doctrine\ODM\MongoDB\UnitOfWork;
@@ -21,8 +22,9 @@
  * @template TKey of array-key
  * @template T of object
  * @template-extends Collection<TKey, T>
+ * @template-extends Selectable<TKey, T>
  */
-interface PersistentCollectionInterface extends Collection
+interface PersistentCollectionInterface extends Collection, Selectable
 {
     /**
      * Sets the document manager and unit of work (used during merge operations).

src/PersistentCollection/PersistentCollectionTrait.php

@@ -7,11 +7,14 @@
 use BadMethodCallException;
 use Closure;
 use Doctrine\Common\Collections\Collection as BaseCollection;
+use Doctrine\Common\Collections\Criteria;
+use Doctrine\Common\Collections\Selectable;
 use Doctrine\ODM\MongoDB\DocumentManager;
 use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
 use Doctrine\ODM\MongoDB\MongoDBException;
 use Doctrine\ODM\MongoDB\UnitOfWork;
 use Doctrine\ODM\MongoDB\Utility\CollectionHelper;
+use LogicException;
 use ReturnTypeWillChange;
 use Traversable;
 
@@ -775,4 +778,15 @@ public function reduce(Closure $func, $initial = null)
 
         return $this->coll->reduce($func, $initial);
     }
+
+    public function matching(Criteria $criteria)
+    {
+        $this->initialize();
+
+        if (! $this->coll instanceof Selectable) {
+            throw new LogicException('The backed collection must implement Selectable to use matching().');
+        }
+
+        return $this->coll->matching($criteria);
+    }
 }

tests/Tests/PersistentCollectionTest.php

@@ -7,6 +7,7 @@
 use Closure;
 use Doctrine\Common\Collections\ArrayCollection;
 use Doctrine\Common\Collections\Collection;
+use Doctrine\Common\Collections\Criteria;
 use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
 use Doctrine\ODM\MongoDB\MongoDBException;
 use Doctrine\ODM\MongoDB\PersistentCollection;
@@ -332,6 +333,27 @@ public function testIsEmptyUsesCountWhenCollectionIsNotInitialized(): void
         self::assertTrue($pcoll->isEmpty());
     }
 
+    public function testMatchingIsForwarded(): void
+    {
+        $criteria = new Criteria();
+        $criteria->where($criteria->expr()->eq('field', 'value'));
+
+        $expectedResult = new ArrayCollection([new stdClass()]);
+
+        // ArrayCollection implements both Collection and Selectable
+        // When doctrine/collections 2.6+ is required, we can mock Collection directly
+        $collection = $this->createMock(ArrayCollection::class);
+        $collection->expects($this->once())
+            ->method('matching')
+            ->with($criteria)
+            ->willReturn($expectedResult);
+
+        $pcoll  = new PersistentCollection($collection, $this->dm, $this->uow);
+        $result = $pcoll->matching($criteria);
+
+        self::assertSame($expectedResult, $result);
+    }
+
     /** @return Collection<int, object>&MockObject */
     private function getMockCollection()
     {