package databases; import com.mongodb.BasicDBObject; import com.mongodb.MongoClient; import com.mongodb.client.FindIterable; import com.mongodb.client.MongoCollection; import com.mongodb.client.MongoDatabase; import org.bson.Document; import java.util.ArrayList; import java.util.Iterator; import java.util.List; public class ConnectToMongoDB { public static MongoDatabase mongoDatabase = null; public MongoDatabase connectToMongoDB() { MongoClient mongoClient = new MongoClient("localhost" , 27017); mongoDatabase = mongoClient.getDatabase("students"); System.out.println("Database Connected"); return mongoDatabase; } public String insertIntoToMongoDB(User user){ String profile = user.getStName(); MongoDatabase mongoDatabase = connectToMongoDB(); MongoCollection collection = mongoDatabase.getCollection("profile"); Document document = new Document().append("stName",user.getStName()).append("stID", user.getStID()). append("stDOB",user.getStDOB()); collection.insertOne(document); return profile + " has been registered"; } /*public List readUserProfileFromMongoDB(){ List list = new ArrayList(); User user = new User(); MongoDatabase mongoDatabase = connectToMongoDB(); MongoCollection collection = mongoDatabase.getCollection("profile"); BasicDBObject basicDBObject = new BasicDBObject(); FindIterable iterable = collection.find(basicDBObject); // FindIterable iterable = collection.find(); for(Document doc:iterable){ String stName = (String)doc.get("stName"); user.setStName(stName); String stID = (String)doc.get("stID"); user.setStID(stID); String stDOB = (String)doc.get("stDOB"); user.setStID(stDOB); user = new User(stName,stID,stDOB); //list.add(user); } return list; } */ public List readFromMongoDB(){ List list = new ArrayList(); MongoDatabase mongoDatabase = connectToMongoDB(); MongoCollection collection = mongoDatabase.getCollection("profile"); FindIterable iterDoc = collection.find(); Iterator it = iterDoc.iterator(); while (it.hasNext()) { list.add(it.next().toString()); } return list; } public static void main(String[] args){ // insertIntoToMongoDB(new User("Naomi Chan", "4493","07-1996")); //List user = readUserProfileFromMongoDB(); //for(User person:user){ //System.out.println(person.getStName()+ " "+ person.getStID()); //} } }