[db] compatability fixes

Author: ar2rsawseen

plugins/pluginManager.js

@@ -1440,6 +1440,17 @@ var pluginManager = function pluginManager() {
                     return;
                 }
                 return function(err, res) {
+                    if (res) {
+                        if (!res.result) {
+                            res.result = {};
+                        }
+                        if (!res.result.ok) {
+                            res.result.ok = !err;
+                        }
+                        if (!res.result.nModified) {
+                            res.result.nModified = res.modifiedCount || 0;
+                        }
+                    }
                     if (err) {
                         if (retry && err.code === 11000) {
                             if (typeof retry === "function") {
@@ -1740,10 +1751,17 @@ var pluginManager = function pluginManager() {
                 logDbRead.d("aggregate " + collection + " %j %j" + at, query, options);
                 logDbRead.d("From connection %j", countlyDb._cly_debug);
                 var cursor = this._aggregate(query, options);
+                cursor._count = cursor.count;
+                cursor.count = function() {
+                    return ob.countDocuments.apply(ob, arguments);
+                };
                 cursor._toArray = cursor.toArray;
                 cursor.toArray = function(cb) {
                     return handlePromiseErrors(cursor._toArray(logForReads(cb, e, copyArguments(args, "aggregate"))), e, copyArguments(arguments, "aggregate"));
                 };
+                cursor.isClosed = function() {
+                    return cursor.closed || cursor.killed;
+                };
                 if (typeof callback === "function") {
                     return cursor.toArray(callback);
                 }
@@ -1782,10 +1800,17 @@ var pluginManager = function pluginManager() {
                 logDbRead.d("find " + collection + " %j %j" + at, query, options);
                 logDbRead.d("From connection %j", countlyDb._cly_debug);
                 var cursor = this._find(query, options);
+                cursor._count = cursor.count;
+                cursor.count = function() {
+                    return ob.countDocuments.apply(ob, arguments);
+                };
                 cursor._toArray = cursor.toArray;
                 cursor.toArray = function(callback) {
                     return handlePromiseErrors(cursor._toArray(logForReads(callback, e, copyArguments(args, "find"))), e, copyArguments(arguments, "find"));
                 };
+                cursor.isClosed = function() {
+                    return cursor.closed || cursor.killed;
+                };
                 return cursor;
             };
 
@@ -1841,6 +1866,20 @@ var pluginManager = function pluginManager() {
                 return ob.findOneAndDelete(query, options, callback);
             };
 
+            ob._drop = ob.drop;
+            ob.drop = function() {
+                if (!arguments.length) {
+                    return ob._drop().catch(function(ex) {
+                        if (ex.code !== 26) {
+                            throw ex;
+                        }
+                    });
+                }
+                else {
+                    return ob._drop.apply(ob, arguments);
+                }
+            };
+
 
             countlyDb._collection_cache[collection] = ob;