PHP-1315: Implement Collection::findOneAndUpdate()

Author: bjori

examples/write.php

@@ -136,6 +136,10 @@
     echo "Kasparov\n";
     var_dump($result);
 
+    echo "Returning the old document where he was Russian\n";
+    $result = $collection->findOneAndUpdate($kasparov, array('$set' => array("citizen" => "Croatia")));
+    var_dump($result);
+
     echo "Deleting him, he isn't Croatian just yet\n";
     $result = $collection->findOneAndDelete(array("citizen" => "Croatia"));
     var_dump($result);

src/MongoDB/Collection.php

@@ -472,6 +472,67 @@ function getFindOneAndReplaceOptions() { /* {{{ */
 
     } /* }}} */
 
+    function findOneAndUpdate(array $filter, array $update, array $options = array()) { /* {{{ */
+        if (key($update)[0] != '$') {
+            throw new \RuntimeException("First key in \$update must be a \$operator");
+        }
+
+        $options = array_merge($this->getFindOneAndUpdateOptions(), $options);
+        $options = $this->_massageFindAndModifyOptions($options, $update);
+
+        $cmd = array(
+            "findandmodify" => $this->collname,
+            "query"         => $filter,
+        ) + $options;
+
+        $doc = $this->_runCommand($this->dbname, $cmd)->getResponseDocument();
+        if ($doc["ok"]) {
+            return $doc["value"];
+        }
+
+        throw $this->_generateCommandException($doc);
+    } /* }}} */
+    function getFindOneAndUpdateOptions() { /* {{{ */
+        return array(
+
+            /**
+             * The maximum amount of time to allow the query to run.
+             *
+             * @see http://docs.mongodb.org/manual/reference/command/findAndModify/
+             */
+            "maxTimeMS" => 0,
+
+            /**
+             * Limits the fields to return for all matching documents.
+             *
+             * @see http://docs.mongodb.org/manual/tutorial/project-fields-from-query-results
+             */
+            "projection" => array(),
+
+            /**
+             * When ReturnDocument.After, returns the updated or inserted document rather than the original.
+             * Defaults to ReturnDocument.Before.
+             *
+             * @see http://docs.mongodb.org/manual/reference/command/findAndModify/
+             */
+            "returnDocument" => self::FIND_ONE_AND_RETURN_BEFORE,
+
+            /**
+             * Determines which document the operation modifies if the query selects multiple documents.
+             *
+             * @see http://docs.mongodb.org/manual/reference/command/findAndModify/
+             */
+            "sort" => array(),
+
+            /**
+             * When true, creates a new document if no document matches the query. The default is false.
+             *
+             * @see http://docs.mongodb.org/manual/reference/command/findAndModify/
+             */
+            "upsert" => false,
+        );
+
+    } /* }}} */
 
     protected function _massageFindAndModifyOptions($options, $update = array()) { /* {{{ */
         $ret = array(