diff --git a/database/src/main/java/com/georgev22/library/database/DatabaseWrapper.java b/database/src/main/java/com/georgev22/library/database/DatabaseWrapper.java index 1839d2d..fc6d394 100644 --- 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. *