Node.js MongoDB Driver API

Source: lib/bulk/unordered.js

  1. 'use strict';
  3. const common = require('./common');
  4. const BulkOperationBase = common.BulkOperationBase;
  5. const Batch = common.Batch;
  6. const bson = common.bson;
  7. const utils = require('../utils');
  8. const toError = utils.toError;
  10. /**
  11.  * Add to internal list of Operations
  12.  *
  13.  * @ignore
  14.  * @param {UnorderedBulkOperation} bulkOperation
  15.  * @param {number} docType number indicating the document type
  16.  * @param {object} document
  17.  * @return {UnorderedBulkOperation}
  18.  */
  19. function addToOperationsList(bulkOperation, docType, document) {
  20.   // Get the bsonSize
  21.   const bsonSize = bson.calculateObjectSize(document, {
  22.     checkKeys: false,
  24.     // Since we don't know what the user selected for BSON options here,
  25.     // err on the safe side, and check the size with ignoreUndefined: false.
  26.     ignoreUndefined: false
  27.   });
  28.   // Throw error if the doc is bigger than the max BSON size
  29.   if (bsonSize >= bulkOperation.s.maxBatchSizeBytes)
  30.     throw toError('document is larger than the maximum size ' + bulkOperation.s.maxBatchSizeBytes);
  31.   // Holds the current batch
  32.   bulkOperation.s.currentBatch = null;
  33.   // Get the right type of batch
  34.   if (docType === common.INSERT) {
  35.     bulkOperation.s.currentBatch = bulkOperation.s.currentInsertBatch;
  36.   } else if (docType === common.UPDATE) {
  37.     bulkOperation.s.currentBatch = bulkOperation.s.currentUpdateBatch;
  38.   } else if (docType === common.REMOVE) {
  39.     bulkOperation.s.currentBatch = bulkOperation.s.currentRemoveBatch;
  40.   }
  42.   const maxKeySize = bulkOperation.s.maxKeySize;
  44.   // Create a new batch object if we don't have a current one
  45.   if (bulkOperation.s.currentBatch == null)
  46.     bulkOperation.s.currentBatch = new Batch(docType, bulkOperation.s.currentIndex);
  48.   // Check if we need to create a new batch
  49.   if (
  50.     bulkOperation.s.currentBatch.size + 1 >= bulkOperation.s.maxWriteBatchSize ||
  51.     bulkOperation.s.currentBatch.sizeBytes + maxKeySize + bsonSize >=
  52.       bulkOperation.s.maxBatchSizeBytes ||
  53.     bulkOperation.s.currentBatch.batchType !== docType
  54.   ) {
  55.     // Save the batch to the execution stack
  56.     bulkOperation.s.batches.push(bulkOperation.s.currentBatch);
  58.     // Create a new batch
  59.     bulkOperation.s.currentBatch = new Batch(docType, bulkOperation.s.currentIndex);
  60.   }
  62.   // We have an array of documents
  63.   if (Array.isArray(document)) {
  64.     throw toError('operation passed in cannot be an Array');
  65.   }
  67.   bulkOperation.s.currentBatch.operations.push(document);
  68.   bulkOperation.s.currentBatch.originalIndexes.push(bulkOperation.s.currentIndex);
  69.   bulkOperation.s.currentIndex = bulkOperation.s.currentIndex + 1;
  71.   // Save back the current Batch to the right type
  72.   if (docType === common.INSERT) {
  73.     bulkOperation.s.currentInsertBatch = bulkOperation.s.currentBatch;
  74.     bulkOperation.s.bulkResult.insertedIds.push({
  75.       index: bulkOperation.s.bulkResult.insertedIds.length,
  76.       _id: document._id
  77.     });
  78.   } else if (docType === common.UPDATE) {
  79.     bulkOperation.s.currentUpdateBatch = bulkOperation.s.currentBatch;
  80.   } else if (docType === common.REMOVE) {
  81.     bulkOperation.s.currentRemoveBatch = bulkOperation.s.currentBatch;
  82.   }
  84.   // Update current batch size
  85.   bulkOperation.s.currentBatch.size += 1;
  86.   bulkOperation.s.currentBatch.sizeBytes += maxKeySize + bsonSize;
  88.   // Return bulkOperation
  89.   return bulkOperation;
  90. }
  92. /**
  93.  * Create a new UnorderedBulkOperation instance (INTERNAL TYPE, do not instantiate directly)
  94.  * @class
  95.  * @extends BulkOperationBase
  96.  * @property {number} length Get the number of operations in the bulk.
  97.  * @return {UnorderedBulkOperation} a UnorderedBulkOperation instance.
  98.  */
  99. class UnorderedBulkOperation extends BulkOperationBase {
  100.   constructor(topology, collection, options) {
  101.     options = options || {};
  102.     options = Object.assign(options, { addToOperationsList });
  104.     super(topology, collection, options, false);
  105.   }
  106. }
  108. /**
  109.  * Returns an unordered batch object
  110.  * @ignore
  111.  */
  112. function initializeUnorderedBulkOp(topology, collection, options) {
  113.   return new UnorderedBulkOperation(topology, collection, options);
  114. }
  116. initializeUnorderedBulkOp.UnorderedBulkOperation = UnorderedBulkOperation;
  117. module.exports = initializeUnorderedBulkOp;
  118. module.exports.Bulk = UnorderedBulkOperation;