Merge pull request #180 from TogetherCrew/add-connect-to-DB

feat: add connection class

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@togethercrew.dev/db",
-  "version": "3.0.73",
+  "version": "3.0.75",
   "description": "All interactions with DB",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

src/service/connetion.ts

@@ -0,0 +1,62 @@
+import mongoose, { type Connection } from 'mongoose';
+
+/**
+ * Manages the MongoDB connection using Mongoose.
+ * Implements the singleton pattern to ensure a single connection throughout the application.
+ */
+export default class MongoConnectionManager {
+  private static instance: MongoConnectionManager;
+  private mongoConnection: Connection | null = null;
+
+  // Private constructor to prevent direct instantiation
+  private constructor() {}
+
+  public static getInstance(): MongoConnectionManager {
+    if (typeof MongoConnectionManager.instance === 'undefined') {
+      MongoConnectionManager.instance = new MongoConnectionManager();
+    }
+    return MongoConnectionManager.instance;
+  }
+
+  public async connect(url: string): Promise<void> {
+    try {
+      if (this.mongoConnection !== null) {
+        throw new Error('MongoDB connection already exists');
+      }
+      await mongoose.connect(url);
+      this.mongoConnection = mongoose.connection;
+      console.log('Connected to MongoDB!');
+    } catch (error: unknown) {
+      const errorMessage = error instanceof Error ? error.message : 'Unknown error';
+      console.error(`Failed to connect to MongoDB: ${errorMessage}`);
+      throw new Error(`Failed to connect to MongoDB: ${errorMessage}`);
+    }
+  }
+
+  public async disconnect(): Promise<void> {
+    try {
+      if (this.mongoConnection !== null) {
+        await mongoose.disconnect();
+        this.mongoConnection = null;
+        console.log('Disconnected from MongoDB');
+      } else {
+        console.warn('No active MongoDB connection to disconnect.');
+      }
+    } catch (error: unknown) {
+      const errorMessage = error instanceof Error ? error.message : 'Unknown error';
+      console.error(`Failed to disconnect from MongoDB: ${errorMessage}`);
+      throw new Error(`Failed to disconnect from MongoDB: ${errorMessage}`);
+    }
+  }
+
+  public ensureConnected(): void {
+    if (this.mongoConnection === null) {
+      throw new Error('No active MongoDB connection. Please connect before performing database operations.');
+    }
+  }
+
+  public getConnection(): Connection | null {
+    this.ensureConnected();
+    return this.mongoConnection;
+  }
+}

src/service/databaseManager.ts

@@ -21,33 +21,6 @@ import { type Snowflake } from 'discord.js';
 export default class DatabaseManager {
   private static instance: DatabaseManager;
   private modelCache: Record<string, boolean> = {};
-  private mongoConnection: mongoose.Connection | null = null;
-
-  public async connectToMongoDB(url: string): Promise<void> {
-    try {
-      if (this.mongoConnection !== null) {
-        throw new Error('MongoDB connection already exists');
-      }
-      await mongoose.connect(url);
-      this.mongoConnection = mongoose.connection;
-      console.log('Connected to MongoDB!');
-    } catch (error) {
-      console.log({ error }, 'Failed to connect to MongoDB!');
-      throw new Error(`Failed to connect to MongoDB: ${error instanceof Error ? error.message : 'Unknown error'}`);
-    }
-  }
-
-  public async disconnectFromMongoDB(): Promise<void> {
-    try {
-      if (this.mongoConnection !== null) {
-        await this.mongoConnection.close();
-        this.mongoConnection = null;
-        console.log('Disconnected from MongoDB');
-      }
-    } catch (error) {
-      throw new Error(`Failed to disconnect from MongoDB: ${error instanceof Error ? error.message : 'Unknown error'}`);
-    }
-  }
 
   public static getInstance(): DatabaseManager {
     if (typeof DatabaseManager.instance === 'undefined') {

src/service/index.ts

@@ -1,3 +1,3 @@
 import DatabaseManager from './databaseManager';
-
-export { DatabaseManager };
+import Connetion from './connetion';
+export { DatabaseManager, Connetion };