First commit

2023-06-05 20:06:13 -05:00
commit 4e3c08d1c4
672 changed files with 179969 additions and 0 deletions
--- a/node_modules/mongodb/lib/bulk/unordered.js
+++ b/node_modules/mongodb/lib/bulk/unordered.js
@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UnorderedBulkOperation = void 0;
+const BSON = require("../bson");
+const error_1 = require("../error");
+const common_1 = require("./common");
+/** @public */
+class UnorderedBulkOperation extends common_1.BulkOperationBase {
+    /** @internal */
+    constructor(collection, options) {
+        super(collection, options, false);
+    }
+    handleWriteError(callback, writeResult) {
+        if (this.s.batches.length) {
+            return false;
+        }
+        return super.handleWriteError(callback, writeResult);
+    }
+    addToOperationsList(batchType, document) {
+        // Get the bsonSize
+        const bsonSize = BSON.calculateObjectSize(document, {
+            checkKeys: false,
+            // Since we don't know what the user selected for BSON options here,
+            // err on the safe side, and check the size with ignoreUndefined: false.
+            ignoreUndefined: false
+        });
+        // Throw error if the doc is bigger than the max BSON size
+        if (bsonSize >= this.s.maxBsonObjectSize) {
+            // TODO(NODE-3483): Change this to MongoBSONError
+            throw new error_1.MongoInvalidArgumentError(`Document is larger than the maximum size ${this.s.maxBsonObjectSize}`);
+        }
+        // Holds the current batch
+        this.s.currentBatch = undefined;
+        // Get the right type of batch
+        if (batchType === common_1.BatchType.INSERT) {
+            this.s.currentBatch = this.s.currentInsertBatch;
+        }
+        else if (batchType === common_1.BatchType.UPDATE) {
+            this.s.currentBatch = this.s.currentUpdateBatch;
+        }
+        else if (batchType === common_1.BatchType.DELETE) {
+            this.s.currentBatch = this.s.currentRemoveBatch;
+        }
+        const maxKeySize = this.s.maxKeySize;
+        // Create a new batch object if we don't have a current one
+        if (this.s.currentBatch == null) {
+            this.s.currentBatch = new common_1.Batch(batchType, this.s.currentIndex);
+        }
+        // Check if we need to create a new batch
+        if (
+        // New batch if we exceed the max batch op size
+        this.s.currentBatch.size + 1 >= this.s.maxWriteBatchSize ||
+            // New batch if we exceed the maxBatchSizeBytes. Only matters if batch already has a doc,
+            // since we can't sent an empty batch
+            (this.s.currentBatch.size > 0 &&
+                this.s.currentBatch.sizeBytes + maxKeySize + bsonSize >= this.s.maxBatchSizeBytes) ||
+            // New batch if the new op does not have the same op type as the current batch
+            this.s.currentBatch.batchType !== batchType) {
+            // Save the batch to the execution stack
+            this.s.batches.push(this.s.currentBatch);
+            // Create a new batch
+            this.s.currentBatch = new common_1.Batch(batchType, this.s.currentIndex);
+        }
+        // We have an array of documents
+        if (Array.isArray(document)) {
+            throw new error_1.MongoInvalidArgumentError('Operation passed in cannot be an Array');
+        }
+        this.s.currentBatch.operations.push(document);
+        this.s.currentBatch.originalIndexes.push(this.s.currentIndex);
+        this.s.currentIndex = this.s.currentIndex + 1;
+        // Save back the current Batch to the right type
+        if (batchType === common_1.BatchType.INSERT) {
+            this.s.currentInsertBatch = this.s.currentBatch;
+            this.s.bulkResult.insertedIds.push({
+                index: this.s.bulkResult.insertedIds.length,
+                _id: document._id
+            });
+        }
+        else if (batchType === common_1.BatchType.UPDATE) {
+            this.s.currentUpdateBatch = this.s.currentBatch;
+        }
+        else if (batchType === common_1.BatchType.DELETE) {
+            this.s.currentRemoveBatch = this.s.currentBatch;
+        }
+        // Update current batch size
+        this.s.currentBatch.size += 1;
+        this.s.currentBatch.sizeBytes += maxKeySize + bsonSize;
+        return this;
+    }
+}
+exports.UnorderedBulkOperation = UnorderedBulkOperation;
+//# sourceMappingURL=unordered.js.map