Update MongoDB connector

lib/cache.js

@@ -23,7 +23,7 @@ const config = require("./config");
 const MAX_CACHE_TTL = config.get("MAX_CACHE_TTL");
 
 let redisClient;
-let mongoDb;
+let mongoClient;
 let mongoCollection;
 let mongoTimeOffset = 0;
 
@@ -49,15 +49,17 @@ function connect(callback) {
 
     mongodb.MongoClient.connect(
       MONGODB_CONNECTION_URL,
-      (err, db) => {
+      { useNewUrlParser: true },
+      (err, client) => {
         if (err) return void callback(err);
 
-        mongoDb = db;
+        const db = client.db();
+        mongoClient = client;
         mongoCollection = db.collection("cache");
-        mongoCollection.ensureIndex({ expire: 1 }, { expireAfterSeconds: 0 });
+        mongoCollection.createIndex({ expire: 1 }, { expireAfterSeconds: 0 });
 
         const now = Date.now();
-        mongoDb.command({ hostInfo: 1 }, (err, res) => {
+        db.command({ hostInfo: 1 }, (err, res) => {
           if (err) return void callback(err);
           mongoTimeOffset = res.system.currentTime.getTime() - now;
           exports.get = mongoGet;
@@ -74,8 +76,7 @@ function connect(callback) {
 
 function disconnect() {
   if (redisClient) redisClient.quit();
-
-  if (mongoDb) mongoDb.close();
+  if (mongoClient) mongoClient.close();
 }
 
 function redisGet(key, callback) {
@@ -171,8 +172,8 @@ function mongoGet(key, callback) {
 
 function mongoDel(key, callback) {
   if (Array.isArray(key))
-    mongoCollection.remove({ _id: { $in: key } }, callback);
-  else mongoCollection.remove({ _id: key }, callback);
+    mongoCollection.deleteMany({ _id: { $in: key } }, callback);
+  else mongoCollection.deleteOne({ _id: key }, callback);
 }
 
 function mongoSet(key, value, ttl, callback) {
@@ -184,24 +185,23 @@ function mongoSet(key, value, ttl, callback) {
   if (!ttl) ttl = MAX_CACHE_TTL;
 
   const expire = new Date(Date.now() - mongoTimeOffset + ttl * 1000);
-  mongoCollection.save({ _id: key, value: value, expire: expire }, callback);
+  mongoCollection.replaceOne(
+    { _id: key },
+    { _id: key, value: value, expire: expire },
+    { upsert: true },
+    callback
+  );
 }
 
 function mongoPop(key, callback) {
-  mongoCollection.findAndModify(
-    { _id: key },
-    null,
-    null,
-    { remove: true },
-    (err, res) => {
-      if (err || !res["value"]) return void callback(err);
+  mongoCollection.findOneAndDelete({ _id: key }, (err, res) => {
+    if (err || !res["value"]) return void callback(err);
 
-      if (res["value"]["expire"] > new Date(Date.now() - mongoTimeOffset))
-        return void callback(null, res["value"]["value"]);
+    if (res["value"]["expire"] > new Date(Date.now() - mongoTimeOffset))
+      return void callback(null, res["value"]["value"]);
 
-      callback();
-    }
-  );
+    callback();
+  });
 }
 
 function mongoLock(lockName, ttl, callback) {
@@ -211,13 +211,13 @@ function mongoLock(lockName, ttl, callback) {
 
   function unlockOrExtend(extendTtl) {
     if (!extendTtl) {
-      mongoCollection.remove({ _id: lockName, value: token }, (err, res) => {
+      mongoCollection.deleteOne({ _id: lockName, value: token }, (err, res) => {
         if (err || res["result"]["n"] !== 1)
           throw err || new Error("Lock expired");
       });
     } else {
       const expire = new Date(Date.now() - mongoTimeOffset + extendTtl * 1000);
-      mongoCollection.update(
+      mongoCollection.updateOne(
         { _id: lockName, value: token },
         { expire: expire },
         (err, res) => {
@@ -231,7 +231,7 @@ function mongoLock(lockName, ttl, callback) {
   const expireTest = new Date(Date.now() - mongoTimeOffset);
   const expireSet = new Date(Date.now() - mongoTimeOffset + ttl * 1000);
 
-  mongoCollection.update(
+  mongoCollection.updateOne(
     { _id: lockName, expire: { $lte: expireTest } },
     { $set: { value: token, expire: expireSet } },
     { upsert: true },