From 6466f08ce755c4a4399cef8ca9ffc410398ef9a0 Mon Sep 17 00:00:00 2001
From: Anderson Borges <andersonvaleriano.borges@gmail.com>
Date: Fri, 19 Jan 2018 10:41:27 -0200
Subject: [PATCH] Added createIfNotExist function

---
 Library/Phalcon/Mvc/MongoCollection.php | 71 +++++++++++++++++++++++++
 1 file changed, 71 insertions(+)

diff --git a/Library/Phalcon/Mvc/MongoCollection.php b/Library/Phalcon/Mvc/MongoCollection.php
index 1570b7b42..8ca5703a3 100644
--- a/Library/Phalcon/Mvc/MongoCollection.php
+++ b/Library/Phalcon/Mvc/MongoCollection.php
@@ -535,6 +535,77 @@ public function create()
          */
         return $this->_postSave(self::$_disableEvents, $success, false);
     }
+  
+    /**
+     * {@inheritdoc}
+     *
+     * @param array $data
+     */
+    public function createIfNotExist(array $criteria)
+    {
+        if (empty($criteria)) {
+            throw new Exception("Criteria parameter must be array with one or more attributes of the model");
+        }
+      
+        /**
+         * Choose a collection according to the collection name
+         */
+        $collection = $this->prepareCU();
+      
+        /**
+         * Assume non-existence to fire beforeCreate events - no update does occur anyway
+         */
+        $exists = false;
+
+        /**
+         * Reset current operation
+         */
+        $this->_operationMade = self::OP_NONE;
+
+        /**
+         * The messages added to the validator are reset here
+         */
+        $this->_errorMessages = [];
+
+        /**
+         * Execute the preSave hook
+         */
+        if ($this->_preSave($this->_dependencyInjector, self::$_disableEvents, $exists) === false) {
+            return false;
+        }
+
+        $keys = array_flip($criteria);
+        $data = $this->toArray();
+
+        if (array_diff_key($keys, $data)) {
+            throw new \Exception("Criteria parameter must be array with one or more attributes of the model");
+        }
+
+        $query = array_intersect_key($data, $keys);
+
+        $success = false;
+
+        $status = $collection->findOneAndUpdate($query,
+            ['$setOnInsert' => $data],
+            ['new' => true, 'upsert' => true]);
+
+        if ($status == null) {
+            $doc = $collection->findOne($query);
+
+            if (is_object($doc)) {
+                $success = true;
+                $this->_operationMade = self::OP_CREATE;
+                $this->_id = $doc['_id'];
+            }
+        } else {
+            $this->appendMessage(new Message("Document already exists"));
+        }
+
+        /**
+         * Call the postSave hooks
+         */
+        return $this->_postSave(self::$_disableEvents, $success, $exists);
+    }
 
     /**
      * {@inheritdoc}