diff --git a/src/main/java/me/braydon/feather/database/impl/mongodb/MongoDB.java b/src/main/java/me/braydon/feather/database/impl/mongodb/MongoDB.java index ec449c3..0760021 100644 --- a/src/main/java/me/braydon/feather/database/impl/mongodb/MongoDB.java +++ b/src/main/java/me/braydon/feather/database/impl/mongodb/MongoDB.java @@ -124,7 +124,7 @@ public class MongoDB implements IDatabase { if (!isConnected()) { // Not connected throw new IllegalStateException("Not connected"); } - return new MongoRepository<>(this, entityClass, database.getCollection(collectionName)); + return new MongoRepository<>(this, entityClass, collectionName); } /** diff --git a/src/main/java/me/braydon/feather/database/impl/mongodb/MongoRepository.java b/src/main/java/me/braydon/feather/database/impl/mongodb/MongoRepository.java index 2983ac7..c10fc81 100644 --- a/src/main/java/me/braydon/feather/database/impl/mongodb/MongoRepository.java +++ b/src/main/java/me/braydon/feather/database/impl/mongodb/MongoRepository.java @@ -36,9 +36,9 @@ public class MongoRepository extends Repository { */ @NonNull private final MongoCollection collection; - public MongoRepository(@NonNull MongoDB database, @NonNull Class entityClass, @NonNull MongoCollection collection) { + public MongoRepository(@NonNull MongoDB database, @NonNull Class entityClass, @NonNull String collectionName) { super(database, entityClass); - this.collection = collection; + collection = database.getDatabase().getCollection(collectionName); // Get the collection by the name } /**