Commit bf060488 authored by Grzegorz Pietrusza's avatar Grzegorz Pietrusza

implelemtn usersRepostiroy cleanup

parent 27cd9e21
...@@ -41,6 +41,7 @@ public class CTFApplication extends Application<ApplicationConfiguration> { ...@@ -41,6 +41,7 @@ public class CTFApplication extends Application<ApplicationConfiguration> {
mongoDBConnector.connect(); mongoDBConnector.connect();
UsersRepository usersRepository = new UsersRepository(mongoDBConnector);//todo: use injections UsersRepository usersRepository = new UsersRepository(mongoDBConnector);//todo: use injections
usersRepository.clean(); //todo
int numberOfNewUsers = new Random().nextInt(50); int numberOfNewUsers = new Random().nextInt(50);
for(int i = 0; i < numberOfNewUsers; i++) { for(int i = 0; i < numberOfNewUsers; i++) {
......
...@@ -38,4 +38,8 @@ public class UsersRepository implements Repository { ...@@ -38,4 +38,8 @@ public class UsersRepository implements Repository {
public void add(User user) { public void add(User user) {
mongoDBConnector.addDocument("users", new Document(user.toMap())); mongoDBConnector.addDocument("users", new Document(user.toMap()));
} }
public void clean() {
mongoDBConnector.removeCollection("users");
}
} }
...@@ -47,4 +47,8 @@ public class MongoDBConnector implements DBConnector { ...@@ -47,4 +47,8 @@ public class MongoDBConnector implements DBConnector {
database.getCollection(collectionName).insertOne(document); database.getCollection(collectionName).insertOne(document);
} }
public void removeCollection(String collectionName) {
database.getCollection(collectionName).drop();
}
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment