mongodb 3.2.5 → 3.3.0-beta2

package/lib/operations/replace_one.js

@@ -0,0 +1,47 @@
+'use strict';
+
+const OperationBase = require('./operation').OperationBase;
+const updateDocuments = require('./common_functions').updateDocuments;
+
+class ReplaceOneOperation extends OperationBase {
+  constructor(collection, filter, doc, options) {
+    super(options);
+
+    this.collection = collection;
+    this.filter = filter;
+    this.doc = doc;
+  }
+
+  execute(callback) {
+    const coll = this.collection;
+    const filter = this.filter;
+    const doc = this.doc;
+    const options = this.options;
+
+    // Set single document update
+    options.multi = false;
+
+    // Execute update
+    updateDocuments(coll, filter, doc, options, (err, r) => replaceCallback(err, r, doc, callback));
+  }
+}
+
+function replaceCallback(err, r, doc, callback) {
+  if (callback == null) return;
+  if (err && callback) return callback(err);
+  if (r == null) return callback(null, { result: { ok: 1 } });
+
+  r.modifiedCount = r.result.nModified != null ? r.result.nModified : r.result.n;
+  r.upsertedId =
+    Array.isArray(r.result.upserted) && r.result.upserted.length > 0
+      ? r.result.upserted[0] // FIXME(major): should be `r.result.upserted[0]._id`
+      : null;
+  r.upsertedCount =
+    Array.isArray(r.result.upserted) && r.result.upserted.length ? r.result.upserted.length : 0;
+  r.matchedCount =
+    Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? 0 : r.result.n;
+  r.ops = [doc];
+  if (callback) callback(null, r);
+}
+
+module.exports = ReplaceOneOperation;

package/lib/operations/set_profiling_level.js

@@ -0,0 +1,48 @@
+'use strict';
+
+const CommandOperation = require('./command');
+const levelValues = new Set(['off', 'slow_only', 'all']);
+
+class SetProfilingLevelOperation extends CommandOperation {
+  constructor(db, level, options) {
+    let profile = 0;
+
+    if (level === 'off') {
+      profile = 0;
+    } else if (level === 'slow_only') {
+      profile = 1;
+    } else if (level === 'all') {
+      profile = 2;
+    }
+
+    super(db, options);
+    this.level = level;
+    this.profile = profile;
+  }
+
+  _buildCommand() {
+    const profile = this.profile;
+
+    // Set up the profile number
+    const command = { profile };
+
+    return command;
+  }
+
+  execute(callback) {
+    const level = this.level;
+
+    if (!levelValues.has(level)) {
+      return callback(new Error('Error: illegal profiling level value ' + level));
+    }
+
+    super.execute((err, doc) => {
+      if (err == null && doc.ok === 1) return callback(null, level);
+      return err != null
+        ? callback(err, null)
+        : callback(new Error('Error with profile command'), null);
+    });
+  }
+}
+
+module.exports = SetProfilingLevelOperation;

package/lib/operations/stats.js

@@ -0,0 +1,45 @@
+'use strict';
+
+const Aspect = require('./operation').Aspect;
+const CommandOperation = require('./command');
+const defineAspects = require('./operation').defineAspects;
+
+/**
+ * Get all the collection statistics.
+ *
+ * @class
+ * @property {Collection} a Collection instance.
+ * @property {object} [options] Optional settings. See Collection.prototype.stats for a list of options.
+ */
+class StatsOperation extends CommandOperation {
+  /**
+   * Construct a Stats operation.
+   *
+   * @param {Collection} a Collection instance.
+   * @param {object} [options] Optional settings. See Collection.prototype.stats for a list of options.
+   */
+  constructor(collection, options) {
+    super(collection.s.db, options, collection);
+  }
+
+  _buildCommand() {
+    const collection = this.collection;
+    const options = this.options;
+
+    // Build command object
+    const command = {
+      collStats: collection.collectionName
+    };
+
+    // Check if we have the scale value
+    if (options['scale'] != null) {
+      command['scale'] = options['scale'];
+    }
+
+    return command;
+  }
+}
+
+defineAspects(StatsOperation, Aspect.READ_OPERATION);
+
+module.exports = StatsOperation;

package/lib/operations/to_array.js

@@ -0,0 +1,68 @@
+'use strict';
+
+const Aspect = require('./operation').Aspect;
+const defineAspects = require('./operation').defineAspects;
+const handleCallback = require('../utils').handleCallback;
+const loadCursor = require('../dynamic_loaders').loadCursor;
+const OperationBase = require('./operation').OperationBase;
+const push = Array.prototype.push;
+
+class ToArrayOperation extends OperationBase {
+  constructor(cursor) {
+    super();
+
+    this.cursor = cursor;
+  }
+
+  execute(callback) {
+    const cursor = this.cursor;
+
+    let Cursor = loadCursor();
+
+    const items = [];
+
+    // Reset cursor
+    cursor.rewind();
+    cursor.s.state = Cursor.INIT;
+
+    // Fetch all the documents
+    const fetchDocs = () => {
+      cursor._next((err, doc) => {
+        if (err) {
+          return cursor._endSession
+            ? cursor._endSession(() => handleCallback(callback, err))
+            : handleCallback(callback, err);
+        }
+        if (doc == null) {
+          return cursor.close({ skipKillCursors: true }, () =>
+            handleCallback(callback, null, items)
+          );
+        }
+
+        // Add doc to items
+        items.push(doc);
+
+        // Get all buffered objects
+        if (cursor.bufferedCount() > 0) {
+          let docs = cursor.readBufferedDocuments(cursor.bufferedCount());
+
+          // Transform the doc if transform method added
+          if (cursor.s.transforms && typeof cursor.s.transforms.doc === 'function') {
+            docs = docs.map(cursor.s.transforms.doc);
+          }
+
+          push.apply(items, docs);
+        }
+
+        // Attempt a fetch
+        fetchDocs();
+      });
+    };
+
+    fetchDocs();
+  }
+}
+
+defineAspects(ToArrayOperation, Aspect.SKIP_SESSION);
+
+module.exports = ToArrayOperation;

package/lib/operations/update_many.js

@@ -0,0 +1,29 @@
+'use strict';
+
+const OperationBase = require('./operation').OperationBase;
+const updateCallback = require('./common_functions').updateCallback;
+const updateDocuments = require('./common_functions').updateDocuments;
+
+class UpdateManyOperation extends OperationBase {
+  constructor(collection, filter, update, options) {
+    super(options);
+
+    this.collection = collection;
+    this.filter = filter;
+    this.update = update;
+  }
+
+  execute(callback) {
+    const coll = this.collection;
+    const filter = this.filter;
+    const update = this.update;
+    const options = this.options;
+
+    // Set single document update
+    options.multi = true;
+    // Execute update
+    updateDocuments(coll, filter, update, options, (err, r) => updateCallback(err, r, callback));
+  }
+}
+
+module.exports = UpdateManyOperation;

package/lib/operations/update_one.js

@@ -0,0 +1,44 @@
+'use strict';
+
+const OperationBase = require('./operation').OperationBase;
+const updateDocuments = require('./common_functions').updateDocuments;
+
+class UpdateOneOperation extends OperationBase {
+  constructor(collection, filter, update, options) {
+    super(options);
+
+    this.collection = collection;
+    this.filter = filter;
+    this.update = update;
+  }
+
+  execute(callback) {
+    const coll = this.collection;
+    const filter = this.filter;
+    const update = this.update;
+    const options = this.options;
+
+    // Set single document update
+    options.multi = false;
+    // Execute update
+    updateDocuments(coll, filter, update, options, (err, r) => updateCallback(err, r, callback));
+  }
+}
+
+function updateCallback(err, r, callback) {
+  if (callback == null) return;
+  if (err) return callback(err);
+  if (r == null) return callback(null, { result: { ok: 1 } });
+  r.modifiedCount = r.result.nModified != null ? r.result.nModified : r.result.n;
+  r.upsertedId =
+    Array.isArray(r.result.upserted) && r.result.upserted.length > 0
+      ? r.result.upserted[0] // FIXME(major): should be `r.result.upserted[0]._id`
+      : null;
+  r.upsertedCount =
+    Array.isArray(r.result.upserted) && r.result.upserted.length ? r.result.upserted.length : 0;
+  r.matchedCount =
+    Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? 0 : r.result.n;
+  callback(null, r);
+}
+
+module.exports = UpdateOneOperation;

package/lib/operations/validate_collection.js

@@ -0,0 +1,40 @@
+'use strict';
+
+const CommandOperation = require('./command');
+
+class ValidateCollectionOperation extends CommandOperation {
+  constructor(admin, collectionName, options) {
+    // Decorate command with extra options
+    let command = { validate: collectionName };
+    const keys = Object.keys(options);
+    for (let i = 0; i < keys.length; i++) {
+      if (options.hasOwnProperty(keys[i]) && keys[i] !== 'session') {
+        command[keys[i]] = options[keys[i]];
+      }
+    }
+
+    super(admin.s.db, options, null, command);
+
+    this.collectionName;
+  }
+
+  execute(callback) {
+    const collectionName = this.collectionName;
+
+    super.execute((err, doc) => {
+      if (err != null) return callback(err, null);
+
+      if (doc.ok === 0) return callback(new Error('Error with validate command'), null);
+      if (doc.result != null && doc.result.constructor !== String)
+        return callback(new Error('Error with validation data'), null);
+      if (doc.result != null && doc.result.match(/exception|corrupt/) != null)
+        return callback(new Error('Error: invalid collection ' + collectionName), null);
+      if (doc.valid != null && !doc.valid)
+        return callback(new Error('Error: invalid collection ' + collectionName), null);
+
+      return callback(null, doc);
+    });
+  }
+}
+
+module.exports = ValidateCollectionOperation;

package/lib/read_concern.js

@@ -0,0 +1,55 @@
+'use strict';
+
+/**
+ * The **ReadConcern** class is a class that represents a MongoDB ReadConcern.
+ * @class
+ * @property {string} level The read concern level
+ * @see https://docs.mongodb.com/manual/reference/read-concern/index.html
+ */
+class ReadConcern {
+  /**
+   * Constructs a ReadConcern from the read concern properties.
+   * @param {string} [level] The read concern level ({'local'|'available'|'majority'|'linearizable'|'snapshot'})
+   */
+  constructor(level) {
+    if (level != null) {
+      this.level = level;
+    }
+  }
+
+  /**
+   * Construct a ReadConcern given an options object.
+   *
+   * @param {object} options The options object from which to extract the write concern.
+   * @return {ReadConcern}
+   */
+  static fromOptions(options) {
+    if (options == null) {
+      return;
+    }
+
+    if (options.readConcern) {
+      return new ReadConcern(options.readConcern.level);
+    }
+
+    return new ReadConcern(options.level);
+  }
+
+  static get MAJORITY() {
+    return 'majority';
+  }
+
+  static get AVAILABLE() {
+    return 'available';
+  }
+
+  static get LINEARIZABLE() {
+    return 'linearizable';
+  }
+
+  static get SNAPSHOT() {
+    return 'snapshot';
+  }
+}
+
+module.exports = ReadConcern;

package/lib/topologies/mongos.js

@@ -1,8 +1,8 @@
 'use strict';
 
 const TopologyBase = require('./topology_base').TopologyBase;
-const MongoError = require('mongodb-core').MongoError;
-const CMongos = require('mongodb-core').Mongos;
+const MongoError = require('../core').MongoError;
+const CMongos = require('../core').Mongos;
 const Cursor = require('../cursor');
 const Server = require('./server');
 const Store = require('./topology_base').Store;
@@ -165,7 +165,7 @@ class Mongos extends TopologyBase {
         ? options.socketOptions
         : options;
 
-    // Translate all the options to the mongodb-core ones
+    // Translate all the options to the core types
     clonedOptions = translateOptions(clonedOptions, socketOptions);
 
     // Build default client information

package/lib/topologies/native_topology.js

@@ -1,6 +1,6 @@
 'use strict';
 
-const Topology = require('mongodb-core').Topology;
+const Topology = require('../core').Topology;
 const ServerCapabilities = require('./topology_base').ServerCapabilities;
 const Cursor = require('../cursor');
 const translateOptions = require('../utils').translateOptions;
@@ -30,7 +30,7 @@ class NativeTopology extends Topology {
         ? options.socketOptions
         : options;
 
-    // Translate all the options to the mongodb-core ones
+    // Translate all the options to the core types
     clonedOptions = translateOptions(clonedOptions, socketOptions);
 
     super(servers, clonedOptions);
@@ -47,6 +47,26 @@ class NativeTopology extends Topology {
     this.s.sCapabilities = new ServerCapabilities(this.lastIsMaster());
     return this.s.sCapabilities;
   }
+
+  // Command
+  command(ns, cmd, options, callback) {
+    super.command(ns.toString(), cmd, options, callback);
+  }
+
+  // Insert
+  insert(ns, ops, options, callback) {
+    super.insert(ns.toString(), ops, options, callback);
+  }
+
+  // Update
+  update(ns, ops, options, callback) {
+    super.update(ns.toString(), ops, options, callback);
+  }
+
+  // Remove
+  remove(ns, ops, options, callback) {
+    super.remove(ns.toString(), ops, options, callback);
+  }
 }
 
 module.exports = NativeTopology;

package/lib/topologies/replset.js

@@ -2,10 +2,10 @@
 
 const Server = require('./server');
 const Cursor = require('../cursor');
-const MongoError = require('mongodb-core').MongoError;
+const MongoError = require('../core').MongoError;
 const TopologyBase = require('./topology_base').TopologyBase;
 const Store = require('./topology_base').Store;
-const CReplSet = require('mongodb-core').ReplSet;
+const CReplSet = require('../core').ReplSet;
 const MAX_JS_INT = require('../utils').MAX_JS_INT;
 const translateOptions = require('../utils').translateOptions;
 const filterOptions = require('../utils').filterOptions;
@@ -172,7 +172,7 @@ class ReplSet extends TopologyBase {
         ? options.socketOptions
         : options;
 
-    // Translate all the options to the mongodb-core ones
+    // Translate all the options to the core types
     clonedOptions = translateOptions(clonedOptions, socketOptions);
 
     // Build default client information

package/lib/topologies/server.js

@@ -1,10 +1,10 @@
 'use strict';
 
-const CServer = require('mongodb-core').Server;
+const CServer = require('../core').Server;
 const Cursor = require('../cursor');
 const TopologyBase = require('./topology_base').TopologyBase;
 const Store = require('./topology_base').Store;
-const MongoError = require('mongodb-core').MongoError;
+const MongoError = require('../core').MongoError;
 const MAX_JS_INT = require('../utils').MAX_JS_INT;
 const translateOptions = require('../utils').translateOptions;
 const filterOptions = require('../utils').filterOptions;
@@ -165,7 +165,7 @@ class Server extends TopologyBase {
         ? options.socketOptions
         : options;
 
-    // Translate all the options to the mongodb-core ones
+    // Translate all the options to the core types
     clonedOptions = translateOptions(clonedOptions, socketOptions);
 
     // Build default client information
@@ -177,7 +177,7 @@ class Server extends TopologyBase {
 
     // Define the internal properties
     this.s = {
-      // Create an instance of a server instance from mongodb-core
+      // Create an instance of a server instance from core module
       coreTopology: new CServer(clonedOptions),
       // Server capabilities
       sCapabilities: null,

package/lib/topologies/topology_base.js

@@ -1,11 +1,11 @@
 'use strict';
 
 const EventEmitter = require('events'),
-  MongoError = require('mongodb-core').MongoError,
+  MongoError = require('../core').MongoError,
   f = require('util').format,
   os = require('os'),
   translateReadPreference = require('../utils').translateReadPreference,
-  ClientSession = require('mongodb-core').Sessions.ClientSession;
+  ClientSession = require('../core').Sessions.ClientSession;
 
 // The store of ops
 var Store = function(topology, storeOptions) {
@@ -313,22 +313,22 @@ class TopologyBase extends EventEmitter {
 
   // Command
   command(ns, cmd, options, callback) {
-    this.s.coreTopology.command(ns, cmd, translateReadPreference(options), callback);
+    this.s.coreTopology.command(ns.toString(), cmd, translateReadPreference(options), callback);
   }
 
   // Insert
   insert(ns, ops, options, callback) {
-    this.s.coreTopology.insert(ns, ops, options, callback);
+    this.s.coreTopology.insert(ns.toString(), ops, options, callback);
   }
 
   // Update
   update(ns, ops, options, callback) {
-    this.s.coreTopology.update(ns, ops, options, callback);
+    this.s.coreTopology.update(ns.toString(), ops, options, callback);
   }
 
   // Remove
   remove(ns, ops, options, callback) {
-    this.s.coreTopology.remove(ns, ops, options, callback);
+    this.s.coreTopology.remove(ns.toString(), ops, options, callback);
   }
 
   // IsConnected

package/lib/url_parser.js

@@ -1,10 +1,11 @@
 'use strict';
 
-const ReadPreference = require('mongodb-core').ReadPreference,
+const ReadPreference = require('./core').ReadPreference,
   parser = require('url'),
   f = require('util').format,
-  Logger = require('mongodb-core').Logger,
+  Logger = require('./core').Logger,
   dns = require('dns');
+const ReadConcern = require('./read_concern');
 
 module.exports = function(url, options, callback) {
   if (typeof options === 'function') (callback = options), (options = {});
@@ -441,7 +442,7 @@ function parseConnectionString(url, options) {
         dbOptions.native_parser = value === 'true';
         break;
       case 'readConcernLevel':
-        dbOptions.readConcern = { level: value };
+        dbOptions.readConcern = new ReadConcern(value);
         break;
       case 'connectTimeoutMS':
         serverOptions.socketOptions.connectTimeoutMS = parseInt(value, 10);