Added functionality to retrieve ids of all docs that match a given filter

Author: martin-schilling

src/DocumentStore.php

@@ -136,4 +136,11 @@ public function getDoc(string $collectionName, string $docId): ?array;
      * @throws UnknownCollection
      */
     public function filterDocs(string $collectionName, Filter $filter, int $skip = null, int $limit = null, OrderBy $orderBy = null): \Traversable;
+
+    /**
+     * @param string $collectionName
+     * @param Filter $filter
+     * @return array
+     */
+    public function filterDocIds(string $collectionName, Filter $filter): array;
 }

src/InMemoryDocumentStore.php

@@ -323,6 +323,27 @@ public function filterDocs(
         return new \ArrayIterator($filteredDocs);
     }
 
+    /**
+     * @param string $collectionName
+     * @param Filter $filter
+     * @return array
+     */
+    public function filterDocIds(
+        string $collectionName,
+        Filter $filter
+    ): array {
+        $this->assertHasCollection($collectionName);
+
+        $docIds = [];
+        foreach ($this->inMemoryConnection['documents'][$collectionName] as $docId => $doc) {
+            if ($filter->match($doc, (string)$docId)) {
+                $docIds[] = $docId;
+            }
+        }
+
+        return $docIds;
+    }
+
     private function hasDoc(string $collectionName, string $docId): bool
     {
         if (! $this->hasCollection($collectionName)) {

tests/InMemoryDocumentStoreTest.php

@@ -4,8 +4,12 @@
 namespace EventEngineTest\DocumentStore;
 
 use EventEngine\DocumentStore\FieldIndex;
+use EventEngine\DocumentStore\Filter\AndFilter;
 use EventEngine\DocumentStore\Filter\AnyFilter;
 use EventEngine\DocumentStore\Filter\EqFilter;
+use EventEngine\DocumentStore\Filter\GtFilter;
+use EventEngine\DocumentStore\Filter\LtFilter;
+use EventEngine\DocumentStore\Filter\OrFilter;
 use EventEngine\DocumentStore\InMemoryDocumentStore;
 use EventEngine\DocumentStore\MultiFieldIndex;
 use EventEngine\Persistence\InMemoryConnection;
@@ -105,6 +109,25 @@ public function it_updates_a_subset_of_a_doc()
         $this->assertEquals(42, $filteredDocs[0]['some']['other']['nested']);
     }
 
+    /**
+     * @test
+     */
+    public function it_retrieves_doc_ids_by_filter()
+    {
+        $this->store->addCollection('test');
+
+        $this->store->addDoc('test', 'a', ['number' => 10]);
+        $this->store->addDoc('test', 'b', ['number' => 20]);
+        $this->store->addDoc('test', 'c', ['number' => 30]);
+
+        $result = $this->store->filterDocIds('test', new OrFilter(
+            new GtFilter('number', 21),
+            new LtFilter('number', 19)
+        ));
+
+        $this->assertEquals(['a', 'c'], $result);
+    }
+
     /**
      * @test
      */