Update db connection logic

Author: ronnmabunga

index.js

@@ -31,12 +31,12 @@ app.use((err, req, res, next) => {
     res.status(statusCode).send({ success: false, message: errorMessage });
 });
 
+connectDB(MONGO_STRING);
+
 if (require.main === module) {
-    connectDB(MONGO_STRING);
     app.listen(PORT, () => {
         logger.info(`Blog API: Now online on port ${PORT}`);
     });
 } else {
-    // logger.turnOffConsoleLogging();
     module.exports = app;
 }

test/test.js

@@ -7,6 +7,7 @@ require("dotenv").config();
 const MONGO_STRING = `${process.env.DEMO1_MONGO_STRING}`;
 chai.use(http);
 let app = require("../index");
+// logger.turnOffConsoleLogging();
 
 describe(`API Tests`, function () {
     let userToken;
@@ -24,7 +25,7 @@ describe(`API Tests`, function () {
             })
             .end((err, res) => {
                 adminToken = res.body.access;
-                // console.log("adminToken: ", adminToken);
+                console.log("adminToken: ", adminToken);
             });
         chai.request(app)
             .post("/users/login")
@@ -35,7 +36,7 @@ describe(`API Tests`, function () {
             })
             .end((err, res) => {
                 userToken = res.body.access;
-                // console.log("userToken: ", userToken);
+                console.log("userToken: ", userToken);
             });
     });
     after(async () => {

utils/mongoDBConn.js

@@ -3,8 +3,19 @@ const logger = require("./logger");
 
 const connectDB = async (uri) => {
     try {
-        await mongoose.connect(uri);
-        mongoose.connection.once("open", () => logger.info("Blog API: connected to MongoDB"));
+        if (mongoose.connection.readyState === 0) {
+            await mongoose.connect(uri);
+            mongoose.connection.once("open", () => {
+                logger.info("Blog API: Connected to MongoDB");
+            });
+        } else if (mongoose.connection.readyState === 2) {
+            await new Promise((resolve) => {
+                mongoose.connection.once("open", resolve);
+            });
+            logger.info("Blog API: Connection is now open after waiting.");
+        } else {
+            logger.info("Blog API: Connection is already open.");
+        }
     } catch (err) {
         logger.error(err.message);
         process.exit(1);
@@ -13,7 +24,9 @@ const connectDB = async (uri) => {
 
 const disconnectDB = async () => {
     await mongoose.connection.close();
-    logger.info("Blog API: Disconnected from MongoDB");
+    mongoose.connection.on("disconnected", () => {
+        logger.info("Blog API: Disconnected from MongoDB");
+    });
 };
 
 module.exports = { connectDB, disconnectDB };