feat(Database): Added getMongoDB

GeorgeV220 · Feb 13, 2023 · b632ef9 · b632ef9
1 parent 6be18a4
commit b632ef9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/database/src/main/java/com/georgev22/library/database/DatabaseWrapper.java b/database/src/main/java/com/georgev22/library/database/DatabaseWrapper.java
@@ -36,6 +36,8 @@ public class DatabaseWrapper {
 
     private Database database;
 
+    private MongoDB mongoDB;
+
     private @Nullable MongoClient mongoClient = null;
     private @Nullable MongoDatabase mongoDatabase = null;
 
@@ -85,7 +87,7 @@ public DatabaseWrapper connect() {
             }
 
             case MONGO -> {
-                MongoDB mongoDB = new MongoDB(data[0], Integer.parseInt(data[1]), data[2], data[3], data[4]);
+                mongoDB = new MongoDB(data[0], Integer.parseInt(data[1]), data[2], data[3], data[4]);
                 mongoClient = mongoDB.getMongoClient();
                 mongoDatabase = mongoDB.getMongoDatabase();
             }
@@ -343,6 +345,10 @@ public Database getSQLDatabase() {
         return database;
     }
 
+    public MongoDB getMongoDB() {
+        return mongoDB;
+    }
+
     /**
      * Returns a MongoDB collection.
      *