Fixes #286 - Faire un vrai singleton pour la connexion à la base de données

This commit is contained in:
C. Fuhrman 2025-03-10 15:20:07 -04:00
parent ba055070a3
commit dcaee719d1
2 changed files with 39 additions and 7 deletions

View file

@ -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();

View file

@ -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);
this.connection = await client.connect();
if (this.connection) {
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;