private static void printDocument(String id) {
// get a single document
Document record = mainColl.findById(new HValue(id));
System.out.print("Single record\n\t");
System.out.println(record);
// print individual fields
System.out.println(
"Id : " + record.getIdString() + " - first name : " + record.getString("first_name"));
}
@Test
public void testMultiLevelJavaBeans() {
String jsonString =
"{"
+ " \"_id\" : \"1001\","
+ " \"first_name\" : \"John\","
+ " \"last_name\" : \"Doe\","
+ " \"active\" : false,"
+ " \"age\" : 35,"
+ " \"interests\" : [\"sports\",\"computers\"],"
+ " \"account\" : 1234.4321,"
+ " \"address\" : {"
+ " \"street\" : \"1015 15th av\","
+ " \"city\" : \"SFO\","
+ " \"zip\" : 94065"
+ " }"
+ "}";
Document doc = Json.newDocument(jsonString);
User user = doc.toJavaBean(User.class);
Document userDoc = Json.newDocument(user);
assertNotEquals(doc, userDoc);
assertEquals(doc.getString("first_name"), userDoc.getString("first_name"));
assertEquals(user.getFirstName(), userDoc.getString("first_name"));
assertEquals(doc.getString("last_name"), userDoc.getString("last_name"));
assertEquals(user.getLastName(), userDoc.getString("last_name"));
assertEquals(doc.getDouble("age"), userDoc.getInt("age"), 0.0);
assertEquals(user.getAge(), userDoc.getInt("age"), 0.0);
assertEquals(doc.getString("address.city"), userDoc.getString("address.city"));
assertEquals(user.getAddress().getTown(), userDoc.getString("address.city"));
Document addressDoc = Json.newDocument(user.getAddress());
assertEquals(addressDoc, userDoc.getMap("address"));
}
@Test
public void testJavaBeansWithAllTypes() {
BeanWithAllTypes bean = new BeanWithAllTypes().init();
Document doc = Json.newDocument(bean);
BeanWithAllTypes newBean = doc.toJavaBean(BeanWithAllTypes.class);
assertEquals(Type.NULL, doc.getValue("nullT").getType());
assertEquals(bean.getNullT(), newBean.getNullT());
assertEquals(bean.isBooleanT(), doc.getBoolean("booleanT"));
assertEquals(bean.isBooleanT(), newBean.isBooleanT());
assertEquals(bean.getStringT(), doc.getString("stringT"));
assertEquals(bean.getStringT(), newBean.getStringT());
assertEquals(bean.getByteT(), doc.getByte("byteT"));
assertEquals(bean.getByteT(), newBean.getByteT());
assertEquals(bean.getShortT(), doc.getShort("shortT"));
assertEquals(bean.getShortT(), newBean.getShortT());
assertEquals(bean.getIntT(), doc.getInt("intT"));
assertEquals(bean.getIntT(), newBean.getIntT());
assertEquals(bean.getLongT(), doc.getLong("longT"));
assertEquals(bean.getLongT(), newBean.getLongT());
assertEquals(bean.getFloatT(), doc.getDouble("floatT"), 0);
assertEquals(bean.getFloatT(), newBean.getFloatT(), 0);
assertEquals(bean.getDoubleT(), doc.getDouble("doubleT"), 0);
assertEquals(bean.getDoubleT(), newBean.getDoubleT(), 0);
assertEquals(bean.getBigDecimalT(), doc.getDecimal("bigDecimalT"));
assertEquals(bean.getBigDecimalT(), newBean.getBigDecimalT());
// disabled until MAPR-18949 is fixed
// assertEquals(bean.getDateT(), doc.getDate("dateT"));
// assertEquals(bean.getDateT(), newBean.getDateT());
assertEquals(bean.getTimeT().toString(), doc.getTime("timeT").toString());
assertEquals(bean.getTimeT().toString(), newBean.getTimeT().toString());
assertEquals(bean.getTimestampT(), doc.getTimestamp("timestampT"));
assertEquals(bean.getTimestampT(), newBean.getTimestampT());
// disabled until Interval.equals() is implemented
// assertEquals(bean.getIntervalT(), doc.getInterval("intervalT"));
// assertEquals(bean.getIntervalT(), newBean.getIntervalT());
assertEquals(bean.getByteBufferT(), doc.getBinary("byteBufferT"));
assertEquals(bean.getByteBufferT(), newBean.getByteBufferT());
assertArrayEquals(bean.getIntArrayT(), newBean.getIntArrayT());
assertEquals(bean.getChildObjectT(), newBean.getChildObjectT());
assertEquals(bean.getBigDecimalT(), doc.getDecimal("bigDecimalT"));
assertEquals(bean.getBigDecimalT(), newBean.getBigDecimalT());
}
private static void createDocuments() throws IOException {
Set<Document> documents = Sets.newHashSet();
// Create a new document (simple format)
Document document =
new HDocument()
.set("_id", "jdoe")
.set("first_name", "John")
.set("last_name", "Doe")
.set("dob", ODate.parse("1970-06-23"));
// save document into the table
mainColl.insertOrReplace(document);
documents.add(document);
printDocument("jdoe");
// create a new document without _id
document =
new HDocument()
.set("first_name", "David")
.set("last_name", "Simon")
.set("dob", ODate.parse("1980-10-13"));
document.setId(new HValue("dsimon"));
mainColl.insert(new HValue("dsimon"), document);
documents.add(((HDocument) document).shallowCopy().setId(new HValue("dsimon")));
printDocument("dsimon");
// create a new document from a simple bean
// look at the User class to see how you can use JSON Annotation to drive the format of the
// document
User user = new User();
user.setId("alehmann");
user.setFirstName("Andrew");
user.setLastName("Lehmann");
user.setDob(ODate.parse("1980-10-13"));
user.addInterest("html");
user.addInterest("css");
user.addInterest("js");
document = Json.newDocument(user);
// save document into the table
mainColl.insertOrReplace(document);
documents.add(HValue.initFromDocument(document));
printDocument("alehmann");
// try to insert the same document ID
try {
mainColl.insert(new HValue("dsimon"), document);
} catch (DocumentExistsException dee) {
System.out.println("Exception during insert : " + dee.getMessage());
}
// Create more complex Record
document =
new HDocument()
.set("_id", "mdupont")
.set("first_name", "Maxime")
.set("last_name", "Dupont")
.set("dob", ODate.parse("1982-02-03"))
.set("interests", Arrays.asList("sports", "movies", "electronics"))
.set("address.line", "1223 Broadway")
.set("address.city", "San Jose")
.set("address.zip", 95109);
mainColl.insert(document);
documents.add(document);
printDocument("mdupont");
// Another way to create sub document
// Create the sub document as document and use it to set the value
Document addressRecord =
new HDocument()
.set("line", "100 Main Street")
.set("city", "San Francisco")
.set("zip", 94105);
document =
new HDocument()
.set("_id", "rsmith")
.set("first_name", "Robert")
.set("last_name", "Smith")
.set("dob", ODate.parse("1982-02-03"))
.set("interests", Arrays.asList("electronics", "music", "sports"))
.set("address", addressRecord);
mainColl.insert(document);
documents.add(document);
printDocument("rsmith");
mainColl.flush(); // flush to the server
}
@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);
}