Merge pull request #287 from ets-cfuhrman-pfe/fuhrmanator/singleton-mongo-connection

Fixes #286 - Faire un vrai singleton pour la connexion à la BD

server/app.js

@@ -127,4 +127,11 @@ async function start() {
   });
 }
 
+// Graceful shutdown on SIGINT (Ctrl+C)
+process.on('SIGINT', async () => {
+  console.log('Shutting down...');
+  await db.closeConnection();
+  process.exit(0);
+});
+
 start();

server/config/db.js

@@ -1,28 +1,53 @@
 const { MongoClient } = require('mongodb');
-const dotenv = require('dotenv')
+const dotenv = require('dotenv');
 
 dotenv.config();
 
 class DBConnection {
-
     constructor() {
         this.mongoURI = process.env.MONGO_URI;
         this.databaseName = process.env.MONGO_DATABASE;
+        this.client = null;
         this.connection = null;
     }
 
+    // Connect to the database, but don't reconnect if already connected
     async connect() {
-        const client = new MongoClient(this.mongoURI);
+        if (this.connection) {
-        this.connection = await client.connect();
+            console.log('Using existing MongoDB connection');
+            return this.connection;
+        }
+
+        try {
+            // Create the MongoClient only if the connection does not exist
+            this.client = new MongoClient(this.mongoURI);
+            await this.client.connect();
+            this.connection = this.client.db(this.databaseName);
+            console.log('MongoDB connected');
+            return this.connection;
+        } catch (error) {
+            console.error('MongoDB connection error:', error);
+            throw new Error('Failed to connect to MongoDB');
+        }
     }
 
+    // Return the current database connection
     getConnection() {
         if (!this.connection) {
-            throw new Error('Connexion MongoDB non établie');
+            throw new Error('MongoDB connection not established');
+        }
+        return this.connection;
+    }
+
+    // Close the MongoDB connection gracefully
+    async closeConnection() {
+        if (this.client) {
+            await this.client.close();
+            console.log('MongoDB connection closed');
         }
-        return this.connection.db(this.databaseName);
     }
 }
 
+// Exporting the singleton instance
 const instance = new DBConnection();
-module.exports = instance;
+module.exports = instance;