Support $elemMatch in query projections (closes #101)

doctrine · Apr 23, 2013 · d51a44d · d51a44d
1 parent 68a820f
commit d51a44d
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/lib/Doctrine/MongoDB/Query/Builder.php b/lib/Doctrine/MongoDB/Query/Builder.php
@@ -366,6 +366,22 @@ public function selectSlice($fieldName, $skip, $limit = null)
         return $this;
     }
 
+    /**
+     * Select a matching embedded document from an array field.
+     *
+     * @param string $fieldName
+     * @param array|Expr $expression
+     * @return Builder
+     */
+    public function selectElemMatch($fieldName, $expression)
+    {
+        if ($expression instanceof Expr) {
+            $expression = $expression->getQuery();
+        }
+        $this->query['select'][$fieldName] = array($this->cmd . 'elemMatch' => $expression);
+        return $this;
+    }
+
     /**
      * Set the current field to operate on.
      *

diff --git a/tests/Doctrine/MongoDB/Tests/Query/BuilderTest.php b/tests/Doctrine/MongoDB/Tests/Query/BuilderTest.php
@@ -576,6 +576,26 @@ private function provideProjections($include)
         );
     }
 
+    public function testSelectElemMatchWithArray()
+    {
+        $qb = $this->getTestQueryBuilder()
+            ->selectElemMatch('addresses', array('state' => 'ny'));
+
+        $expected = array('addresses' => array('$elemMatch' => array('state' => 'ny')));
+
+        $this->assertEquals($expected, $qb->debug('select'));
+    }
+
+    public function testSelectElemMatchWithExpr()
+    {
+        $qb = $this->getTestQueryBuilder();
+        $qb->selectElemMatch('addresses', $qb->expr()->field('state')->equals('ny'));
+
+        $expected = array('addresses' => array('$elemMatch' => array('state' => 'ny')));
+
+        $this->assertEquals($expected, $qb->debug('select'));
+    }
+
     private function getTestQueryBuilder()
     {
         return $this->conn->selectCollection('db', 'users')->createQueryBuilder();