Initial Commit

node_modules/mongodb/lib/operations/create_index.js

@@ -0,0 +1,92 @@
+'use strict';
+
+const Aspect = require('./operation').Aspect;
+const CommandOperation = require('./command');
+const defineAspects = require('./operation').defineAspects;
+const handleCallback = require('../utils').handleCallback;
+const MongoError = require('../core').MongoError;
+const parseIndexOptions = require('../utils').parseIndexOptions;
+
+const keysToOmit = new Set([
+  'name',
+  'key',
+  'writeConcern',
+  'w',
+  'wtimeout',
+  'j',
+  'fsync',
+  'readPreference',
+  'session'
+]);
+
+class CreateIndexOperation extends CommandOperation {
+  constructor(db, name, fieldOrSpec, options) {
+    super(db, options);
+
+    // Build the index
+    const indexParameters = parseIndexOptions(fieldOrSpec);
+    // Generate the index name
+    const indexName = typeof options.name === 'string' ? options.name : indexParameters.name;
+    // Set up the index
+    const indexesObject = { name: indexName, key: indexParameters.fieldHash };
+
+    this.name = name;
+    this.fieldOrSpec = fieldOrSpec;
+    this.indexes = indexesObject;
+  }
+
+  _buildCommand() {
+    const options = this.options;
+    const name = this.name;
+    const indexes = this.indexes;
+
+    // merge all the options
+    for (let optionName in options) {
+      if (!keysToOmit.has(optionName)) {
+        indexes[optionName] = options[optionName];
+      }
+    }
+
+    // Create command, apply write concern to command
+    const cmd = { createIndexes: name, indexes: [indexes] };
+
+    return cmd;
+  }
+
+  execute(callback) {
+    const db = this.db;
+    const options = this.options;
+    const indexes = this.indexes;
+
+    // Get capabilities
+    const capabilities = db.s.topology.capabilities();
+
+    // Did the user pass in a collation, check if our write server supports it
+    if (options.collation && capabilities && !capabilities.commandsTakeCollation) {
+      // Create a new error
+      const error = new MongoError('server/primary/mongos does not support collation');
+      error.code = 67;
+      // Return the error
+      return callback(error);
+    }
+
+    // Ensure we have a callback
+    if (options.writeConcern && typeof callback !== 'function') {
+      throw MongoError.create({
+        message: 'Cannot use a writeConcern without a provided callback',
+        driver: true
+      });
+    }
+
+    // Attempt to run using createIndexes command
+    super.execute((err, result) => {
+      if (err == null) return handleCallback(callback, err, indexes.name);
+
+      return handleCallback(callback, err, result);
+    });
+  }
+}
+
+defineAspects(CreateIndexOperation, Aspect.WRITE_OPERATION);
+
+module.exports = CreateIndexOperation;