ARSN: checkpoint

lib/storage/metadata/MetadataWrapper.js

@@ -292,6 +292,19 @@ class MetadataWrapper {
         });
     }
 
+    getObjectsMD(bucketName, objNamesWithParams, log, cb) {
+        log.debug('getting object from metadata');
+        this.client.getObjects(bucketName, objNamesWithParams, log, (err, data) => {
+            if (err) {
+                log.debug('error from metadata', { implName: this.implName,
+                    err });
+                return cb(err);
+            }
+            log.debug('object retrieved from metadata');
+            return cb(err, data);
+        });
+    }
+
     deleteObjectMD(bucketName, objName, params, log, cb, originOp = 's3:ObjectRemoved:Delete') {
         log.debug('deleting object from metadata');
         this.client.deleteObject(bucketName, objName, params, log, err => {

lib/storage/metadata/mongoclient/MongoClientInterface.js

@@ -1041,6 +1041,61 @@ class MongoClientInterface {
         ], cb);
     }
 
+    getObjects(bucketName, objects, log, callback) {
+        const c = this.getCollection(bucketName);
+        let vFormat = null;
+        // Function to process each document
+        const processDoc = (doc, objName, params, cb) => {
+            if (!doc && params && params.versionId) {
+                return cb(errors.NoSuchKey);
+            }
+            // If no master found then object is either non existent
+            // or last version is delete marker
+            if (!doc || doc.value.isPHD) {
+                return this.getLatestVersion(c, objName, vFormat, log, (err, doc) => {
+                    if (err && !err.is.NoSuchKey) {
+                        return cb(err);
+                    }
+                    return cb(null, doc || null);
+                });
+            }
+            MongoUtils.unserialize(doc.value);
+            return cb(null, doc.value);
+        };
+        this.getBucketVFormat(bucketName, log, (err, _vFormat) => {
+            if (err) {
+                return callback(err);
+            }
+            vFormat = _vFormat;
+            // Create keys, maintaining the context with each key
+            const keysAndObjects = objects.map(({ key: objName, params }) => {
+                const _key = params && params.versionId
+                    ? formatVersionKey(objName, params.versionId, vFormat)
+                    : formatMasterKey(objName, vFormat);
+                return { key: _key, objName, params };
+            });
+            // Extract keys and find documents
+            const keys = keysAndObjects.map(o => o.key);
+            return c.find({
+                _id: { $in: keys },
+                $or: [
+                    { 'value.deleted': { $exists: false } },
+                    { 'value.deleted': { $eq: false } },
+                ],
+            }).toArray().then(docs => {
+                // Create a Map to quickly find docs by their keys
+                const docByKey = new Map(docs.map(doc => [doc._id, doc]));
+                // Process each document using associated context (objName, params)
+                async.mapLimit(keysAndObjects, 5, ({ key, objName, params }, cb) => {
+                    const doc = docByKey.get(key);
+                    processDoc(doc, objName, params, cb);
+                }, callback);
+            }).catch(err => {
+                callback(err);
+            });
+        });
+    }
+
     /**
      * This function return the latest version of an object
      * by getting all keys related to an object's versions, ordering them
@@ -1199,7 +1254,7 @@ class MongoClientInterface {
                     'value.isPHD': true,
                     'value.versionId': mst.versionId,
                 };
-                this.internalDeleteObject(c, bucketName, masterKey, filter, log, err => {
+                this.internalDeleteObject(c, bucketName, masterKey, {}, filter, log, err => {
                     if (err) {
                         // the PHD master might get updated when a PUT is performed
                         // before the repair is done, we don't want to return an error
@@ -1270,7 +1325,7 @@ class MongoClientInterface {
                 .then(() => next())
                 .catch(err => next(err)),
             // delete version
-            next => this.internalDeleteObject(c, bucketName, versionKey, {}, log,
+            next => this.internalDeleteObject(c, bucketName, versionKey, params, {}, log,
                 err => {
                     // we don't return an error in case we don't find
                     // a version as we expect this case when dealing with
@@ -1307,7 +1362,7 @@ class MongoClientInterface {
      */
     deleteObjectVerNotMaster(c, bucketName, objName, params, log, cb, originOp = 's3:ObjectRemoved:Delete') {
         const versionKey = formatVersionKey(objName, params.versionId, params.vFormat);
-        this.internalDeleteObject(c, bucketName, versionKey, {}, log, err => {
+        this.internalDeleteObject(c, bucketName, versionKey, params, {}, log, err => {
             if (err) {
                 if (err.is.NoSuchKey) {
                     log.error(
@@ -1398,7 +1453,7 @@ class MongoClientInterface {
      */
     deleteObjectNoVer(c, bucketName, objName, params, log, cb, originOp = 's3:ObjectRemoved:Delete') {
         const masterKey = formatMasterKey(objName, params.vFormat);
-        this.internalDeleteObject(c, bucketName, masterKey, {}, log, err => {
+        this.internalDeleteObject(c, bucketName, masterKey, params, {}, log, err => {
             if (err) {
                 // Should not return an error when no object is found
                 if (err.is.NoSuchKey) {
@@ -1420,6 +1475,7 @@ class MongoClientInterface {
      * @param {Object} collection MongoDB collection
      * @param {string} bucketName bucket name
      * @param {string} key Key of the object to delete
+     * @param {object} params request params
      * @param {object} filter additional query filters
      * @param {Logger}log logger instance
      * @param {Function} cb callback containing error
@@ -1427,8 +1483,24 @@ class MongoClientInterface {
      * @param {String} [originOp=s3:ObjectRemoved:Delete] origin operation
      * @return {undefined}
      */
-    internalDeleteObject(collection, bucketName, key, filter, log, cb, originOp = 's3:ObjectRemoved:Delete') {
-        // filter used when finding and updating object
+    internalDeleteObject(collection, bucketName, key, params, filter, log, cb, originOp = 's3:ObjectRemoved:Delete') {
+        // filter used when deleting object
+        const deleteFilter = Object.assign({
+            _id: key,
+        }, filter);
+
+        if (params.shouldOnlyDelete) {
+            // If flag is true, directly delete object
+            return collection.deleteOne(deleteFilter)
+                .then(() => cb(null))
+                .catch(err => {
+                    log.error('internalDeleteObject: error deleting object',
+                        { bucket: bucketName, object: key, error: err.message });
+                    return cb(errors.InternalError);
+                });
+        }
+
+        // If flag is false, proceed with normal operations
         const findFilter = Object.assign({
             _id: key,
             $or: [
@@ -1436,12 +1508,13 @@ class MongoClientInterface {
                 { 'value.deleted': { $eq: false } },
             ],
         }, filter);
-        // filter used when deleting object
+
         const updateDeleteFilter = Object.assign({
             '_id': key,
             'value.deleted': true,
         }, filter);
-        async.waterfall([
+
+        return async.waterfall([
             // Adding delete flag when getting the object
             // to avoid having race conditions.
             next => collection.findOneAndUpdate(findFilter, {
@@ -1469,8 +1542,8 @@ class MongoClientInterface {
             }),
             // We update the full object to get the whole object metadata
             // in the oplog update event
-            (objMetadata, next) => collection.bulkWrite([
-                {
+            (objMetadata, next) => {
+                const operations = [{
                     updateOne: {
                         filter: updateDeleteFilter,
                         update: {
@@ -1478,12 +1551,17 @@ class MongoClientInterface {
                         },
                         upsert: false,
                     },
-                }, {
+                }];
+
+                // Add the delete operation if the shouldDelete flag is true
+                operations.push({
                     deleteOne: {
                         filter: updateDeleteFilter,
                     },
+                });
+
+                collection.bulkWrite(operations, { ordered: true }).then(() => next(null)).catch(() => next());
             },
-            ], { ordered: true }).then(() => next(null)).catch(() => next()),
         ], (err, res) => {
             if (err) {
                 if (err.is.NoSuchKey) {