Replaced mongodb.ObjectId() with the functions from mongo_utils.js

Signed-off-by: Aayush Chouhan <[email protected]>

Author: aayushchouhan09

src/agent/block_store_speed.js

@@ -4,7 +4,7 @@
 // const _ = require('lodash');
 const argv = require('minimist')(process.argv);
 const cluster = require('cluster');
-const mongodb = require('mongodb');
+const mongo_utils = require('../util/mongo_utils');
 
 const api = require('../api');
 const config = require('../../config');
@@ -61,7 +61,7 @@ async function worker(client) {
 }
 
 async function write_block(client) {
-    const block_id = new mongodb.ObjectId();
+    const block_id = new mongo_utils.ObjectId();
     return client.block_store.write_block({
         [RPC_BUFFERS]: { data: Buffer.allocUnsafe(argv.size) },
         block_md: {

src/sdk/nb.d.ts

@@ -2,6 +2,7 @@ export as namespace nb;
 
 import * as fs from 'fs';
 import * as mongodb from 'mongodb';
+import * as mongo_utils from '../util/mongo_utils';
 import { EventEmitter } from 'events';
 import { Readable, Writable } from 'stream';
 import { IncomingMessage, ServerResponse } from 'http';
@@ -46,7 +47,7 @@ interface Base {
     toString?(): string;
 }
 
-type ID = mongodb.ObjectID;
+type ID = string;
 type DBBuffer = mongodb.Binary | Buffer;
 
 interface System extends Base {
@@ -719,8 +720,8 @@ interface DBClient {
     populate(docs: object[] | object, doc_path: string, collection: DBCollection, fields: object): Promise<object[] | object>;
     resolve_object_ids_recursive(idmap: object, item: object): object;
     resolve_object_ids_paths(idmap: object, item: object, paths: string[], allow_missing: boolean): object;
-    new_object_id(): mongodb.ObjectId;
-    parse_object_id(id_str: string): mongodb.ObjectId;
+    new_object_id(): mongo_utils.ObjectId;
+    parse_object_id(id_str: string): mongo_utils.ObjectId;
     fix_id_type(doc: object[] | object): object[] | object;
     is_object_id(id: object[] | object): boolean;
     is_err_duplicate_key(err: object): boolean;

src/server/analytic_services/activity_log_store.js

@@ -1,7 +1,7 @@
 /* Copyright (C) 2016 NooBaa */
 'use strict';
 
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 const _ = require('lodash');
 
 const db_client = require('../../util/db_client');
@@ -25,7 +25,7 @@ class ActivityLogStore {
     }
 
     make_activity_log_id(id_str) {
-        return new mongodb.ObjectID(id_str);
+        return new mongo_utils.ObjectId(id_str);
     }
 
 

src/server/analytic_services/history_data_store.js

@@ -1,7 +1,7 @@
 /* Copyright (C) 2016 NooBaa */
 'use strict';
 
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 
 // const dbg = require('../../util/debug_module')(__filename);
 const config = require('../../../config.js');
@@ -30,7 +30,7 @@ class HistoryDataStore {
             const time_stamp = new Date();
             const record_expiration_date = new Date(time_stamp.getTime() - config.STATISTICS_COLLECTOR_EXPIRATION);
             const record = {
-                _id: new mongodb.ObjectId(),
+                _id: new mongo_utils.ObjectId(),
                 time_stamp,
                 system_snapshot: item,
                 history_type: 'SYSTEM'

src/server/bg_services/mirror_writer.js

@@ -3,7 +3,7 @@
 
 const util = require('util');
 const _ = require('lodash');
-const { ObjectId } = require('mongodb');
+const { ObjectId } = require('../../util/mongo_utils');
 
 const config = require('../../../config');
 const dbg = require('../../util/debug_module')(__filename);

src/server/node_services/nodes_store.js

@@ -2,7 +2,7 @@
 'use strict';
 
 const _ = require('lodash');
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 
 const dbg = require('../../util/debug_module')(__filename);
 const node_schema = require('./node_schema');
@@ -24,7 +24,7 @@ class NodesStore {
     }
 
     make_node_id(id_str) {
-        return new mongodb.ObjectId(id_str);
+        return new mongo_utils.ObjectId(id_str);
     }
 
     is_connected() {

src/server/notifications/alerts_log_store.js

@@ -1,7 +1,7 @@
 /* Copyright (C) 2016 NooBaa */
 'use strict';
 
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 const _ = require('lodash');
 const P = require('../../util/promise');
 const db_client = require('../../util/db_client');
@@ -23,7 +23,7 @@ class AlertsLogStore {
     }
 
     make_alert_log_id(id_str) {
-        return new mongodb.ObjectID(id_str);
+        return new mongo_utils.ObjectId(id_str);
     }
 
     create(alert_log) {
@@ -94,12 +94,12 @@ class AlertsLogStore {
 
         let _id;
         if (ids) {
-            const obj_ids = ids.map(id => new mongodb.ObjectID(id));
+            const obj_ids = ids.map(id => new mongo_utils.ObjectId(id));
             _id = { $in: obj_ids };
         } else if (till) {
-            _id = { $lt: new mongodb.ObjectID(till) };
+            _id = { $lt: new mongo_utils.ObjectId(till) };
         } else if (since) {
-            _id = { $gt: new mongodb.ObjectID(since) };
+            _id = { $gt: new mongo_utils.ObjectId(since) };
         }
 
         return _.omitBy({

src/server/object_services/md_store.js

@@ -7,7 +7,7 @@
 const _ = require('lodash');
 const assert = require('assert');
 const moment = require('moment');
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 const mime = require('mime-types');
 
 const P = require('../../util/promise');
@@ -92,18 +92,18 @@ class MDStore {
         if (zero_suffix) {
             suffix = '0'.repeat(16);
         } else {
-            suffix = String(new mongodb.ObjectId()).slice(8, 24);
+            suffix = String(new mongo_utils.ObjectId()).slice(8, 24);
         }
         const hex_id = padded_hex_time + suffix;
         assert(padded_hex_time.length === 8);
         assert(suffix.length === 16);
         assert(hex_id.length === 24);
         assert(parseInt(padded_hex_time, 16) === Math.floor(time / 1000));
-        return new mongodb.ObjectId(hex_id);
+        return new mongo_utils.ObjectId(hex_id);
     }
 
     is_valid_md_id(id_str) {
-        return mongodb.ObjectId.isValid(id_str);
+        return mongo_utils.ObjectId.isValid(id_str);
     }
 
     /////////////
@@ -2113,7 +2113,7 @@ function sort_list_uploads_with_delimiter(a, b) {
  * @returns {nb.ID}
  */
 function make_md_id(id_str) {
-    return new mongodb.ObjectId(id_str);
+    return new mongo_utils.ObjectId(id_str);
 }
 
 

src/server/system_services/config_file_store.js

@@ -4,7 +4,7 @@
 const _ = require('lodash');
 const dbg = require('../../util/debug_module')(__filename);
 const db_client = require('../../util/db_client');
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 const config_file_schema = require('./schemas/config_file_schema');
 const config_file_indexes = require('./schemas/config_file_indexes');
 
@@ -26,7 +26,7 @@ class ConfigFileStore {
     async insert(item) {
         dbg.log0(`insert`, item);
         _.defaults(item, {
-            _id: new mongodb.ObjectId()
+            _id: new mongo_utils.ObjectId()
         });
         // There shouldn't be more than one record, this is being on the safe side
         this._config_files.validate(item);

src/server/system_services/replication_store.js

@@ -2,7 +2,7 @@
 'use strict';
 
 const _ = require('lodash');
-const mongodb = require('mongodb');
+const mongo_utils = require('../../util/mongo_utils');
 const db_client = require('../../util/db_client');
 const dbg = require('../../util/debug_module')(__filename);
 const replication_schema = require('./schemas/replication_configuration_schema');
@@ -25,7 +25,7 @@ class ReplicationStore {
         item = _.omitBy(item, _.isNil);
         dbg.log1(`insert_replication`, item);
         const record = {
-            _id: new mongodb.ObjectId(),
+            _id: new mongo_utils.ObjectId(),
             ...item
         };
         this._replicationconfigs.validate(record);