Spaces:
Runtime error
Runtime error
| ; | |
| 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 |