@Test public void updateDocuments() throws IOException { // System.out.println("\t\tAdd address and status to jdoe"); // System.out.println("before :\t" + mainColl.findById(new HValue("jdoe"))); Document doc = mainColl.findById(new HValue("jdoe")); assertNull(doc.getMap("address")); // create a mutation DocumentMutation mutation = new HDocumentMutation() .set("active", true) .set("address.line", "1015 15th Avenue") .set("address.city", "Redwood City") .set("address.zip", 94065); mainColl.update(new HValue("jdoe"), mutation); mainColl.flush(); // System.out.println("after :\t\t" + mainColl.findById(new HValue("jdoe"))); doc = mainColl.findById(new HValue("jdoe")); assertEquals(3, doc.getMap("address").size()); // System.out.println("\n\n\t\tAppend new interests to users"); doc = mainColl.findById(new HValue("jdoe")); assertNull(doc.getList("interests")); doc = mainColl.findById(new HValue("mdupont")); assertEquals(3, doc.getList("interests").size()); // create a mutation mutation = new HDocumentMutation().append("interests", Arrays.asList(new String[] {"development"})); mainColl.update(new HValue("jdoe"), mutation); mainColl.update(new HValue("mdupont"), mutation); mainColl.flush(); // System.out.println("after :\t\t" + mainColl.findById(new HValue("jdoe"), "first_name", // "last_name", "interests")); // System.out.println("after :\t\t" + mainColl.findById(new HValue("mdupont"), "first_name", // "last_name", "interests")); doc = mainColl.findById(new HValue("jdoe")); assertEquals(1, doc.getList("interests").size()); doc = mainColl.findById(new HValue("mdupont")); assertEquals(4, doc.getList("interests").size()); // System.out.println("\n\n\t\tRemove attributes (dob)"); // System.out.println("before :\t" + mainColl.findById(new HValue("jdoe"))); doc = mainColl.findById(new HValue("jdoe")); assertTrue(doc.getDate("dob").compareTo(ODate.parse("1970-06-23")) == 0); // create a mutation mutation = new HDocumentMutation().delete("dob"); mainColl.update(new HValue("jdoe"), mutation); mainColl.flush(); // System.out.println("after :\t\t" + mainColl.findById(new HValue("jdoe"))); doc = mainColl.findById(new HValue("jdoe")); assertNull(doc.getDate("dob")); }
@Test public void queryDocuments() throws Exception { // get a single document Document record = mainColl.findById(new HValue("mdupont")); assertEquals("mdupont", record.getIdString()); assertEquals("Maxime", record.getString("first_name")); assertEquals("Dupont", record.getString("last_name")); // get a single document with project record = mainColl.findById(new HValue("mdupont"), "last_name"); assertEquals("mdupont", record.getIdString()); assertEquals(null, record.getString("first_name")); assertEquals("Dupont", record.getString("last_name")); // get single document and map it to the bean User user = mainColl.findById(new HValue("alehmann")).toJavaBean(User.class); assertEquals("alehmann", user.getId()); assertEquals("Andrew", user.getFirstName()); assertEquals("Lehmann", user.getLastName()); // all records in the table DocumentStream rs = mainColl.find(); Iterator<? extends Document> itrs = rs.iterator(); Document readRecord; Set<Document> readRecords = Sets.newHashSet(); while (itrs.hasNext()) { readRecord = itrs.next(); readRecords.add(readRecord); // System.out.println("\t" + readRecord); } assertEquals(5, readRecords.size()); rs.close(); // all records in the table with projection int cnt = 0; try (DocumentStream documentStream = mainColl.find("first_name", "last_name")) { for (Document doc : documentStream) { cnt++; assertEquals(3, doc.size()); // System.out.println("\t" + doc); } } assertEquals(5, cnt); // all records and use a POJO // it is interesting to see how you can ignore unknown attributes with the JSON Annotations cnt = 0; try (DocumentStream documentStream = mainColl.find()) { for (Document doc : documentStream) { cnt++; User u = doc.toJavaBean(User.class); assertTrue( ImmutableList.<String>of("Doe", "Simon", "Smith", "Dupont", "Lehmann") .contains(u.getLastName())); // System.out.println("\t" + doc.toJavaBean(User.class)); } } assertEquals(5, cnt); // find with condition // Condition equals a string QueryCondition condition = new HQueryCondition().is("last_name", QueryCondition.Op.EQUAL, "Doe").build(); // System.out.println("\n\nCondition: " + condition); cnt = 0; try (DocumentStream documentStream = mainColl.find(condition)) { ((HDocumentStream) documentStream).explain(); for (Document doc : documentStream) { cnt++; assertEquals("Doe", doc.getString("last_name")); // System.out.println("\t" + doc); } } assertEquals(1, cnt); // find with condition and projection // Condition equals a string condition = new HQueryCondition().is("last_name", QueryCondition.Op.EQUAL, "Doe").build(); // System.out.println("\n\nCondition: " + condition); cnt = 0; try (DocumentStream documentStream = mainColl.find(condition, "last_name")) { for (Document doc : documentStream) { cnt++; assertEquals("Doe", doc.getString("last_name")); // System.out.println("\t" + doc); } } assertEquals(1, cnt); // Condition as date range condition = new HQueryCondition() .and() .is("dob", QueryCondition.Op.GREATER_OR_EQUAL, ODate.parse("1980-01-01")) .is("dob", QueryCondition.Op.LESS, ODate.parse("1981-01-01")) .close() .build(); // System.out.println("\n\nCondition: " + condition); cnt = 0; try (DocumentStream documentStream = mainColl.find(condition)) { for (Document doc : documentStream) { cnt++; assertTrue(doc.getDate("dob").compareTo(ODate.parse("1980-01-01")) > 0); assertTrue(doc.getDate("dob").compareTo(ODate.parse("1981-01-01")) < 0); // System.out.println("\t" + doc); } } assertEquals(2, cnt); // Condition in sub document condition = new HQueryCondition().is("address.zip", QueryCondition.Op.EQUAL, 95109).build(); // System.out.println("\n\nCondition: " + condition); cnt = 0; try (DocumentStream documentStream = mainColl.find(condition)) { for (Document doc : documentStream) { cnt++; assertEquals(95109, doc.getInt("address.zip")); // System.out.println("\t" + doc); } } assertEquals(1, cnt); // Contains a specific value in an array condition = new HQueryCondition().is("interests[]", QueryCondition.Op.EQUAL, "sports").build(); // System.out.println("\n\nCondition: " + condition); cnt = 0; try (DocumentStream documentStream = mainColl.find(condition)) { for (Document doc : documentStream) { cnt++; assertTrue(doc.getList("interests").contains("sports")); // System.out.println("\t" + doc); } } assertEquals(2, cnt); // Contains a value at a specific index condition = new HQueryCondition().is("interests[0]", QueryCondition.Op.EQUAL, "sports").build(); // System.out.println("\n\nCondition: " + condition); cnt = 0; try (DocumentStream documentStream = mainColl.find(condition, "first_name", "last_name", "interests")) { for (Document doc : documentStream) { cnt++; assertEquals("sports", doc.getString("interests[0]")); // System.out.println("\t" + doc); } } assertEquals(1, cnt); }