Added mongodb flatten format.

hkazmi · Mar 23, 2017 · 6a4313a · 6a4313a
1 parent aed0ef5
commit 6a4313a
Show file tree

Hide file tree

Showing 5 changed files with 83 additions and 8 deletions.
diff --git a/src/main/java/com/github/wnameless/json/flattener/FlattenMode.java b/src/main/java/com/github/wnameless/json/flattener/FlattenMode.java
@@ -34,6 +34,11 @@ public enum FlattenMode {
  /**
  * Flattens every objects except arrays.
  */
- KEEP_ARRAYS;
+ KEEP_ARRAYS,
+
+ /**
+ * Conforms to MongoDB dot.notation to update also nested documents.
+ */
+ MONGODB;
 
 }
diff --git a/src/main/java/com/github/wnameless/json/flattener/JsonFlattener.java b/src/main/java/com/github/wnameless/json/flattener/JsonFlattener.java
@@ -142,6 +142,10 @@ public JsonFlattener(Reader jsonReader) throws IOException {
  */
  public JsonFlattener withFlattenMode(FlattenMode flattenMode) {
  this.flattenMode = notNull(flattenMode);
+ if(flattenMode.equals(FlattenMode.MONGODB)) {
+ this.separator = '.';
+ this.leftBracket = '.';
+ }
  flattenedMap = null;
  return this;
  }
@@ -181,8 +185,12 @@ public JsonFlattener withSeparator(char separator) {
  }
 
  private String illegalBracketsRegex() {
- return "[\"\\s" + "]";
- //return "[\"\\s" + Pattern.quote(this.separator.toString()) + "]";
+ if(this.flattenMode.equals(FlattenMode.MONGODB)) {
+ //allow . as separator, needed for arrays
+ return "[\"\\s" + "]";
+ } else {
+ return "[\"\\s" + Pattern.quote(this.separator.toString()) + "]";
+ }
  }
 
  /**
@@ -198,7 +206,7 @@ private String illegalBracketsRegex() {
  */
  public JsonFlattener withLeftAndRightBrackets(char leftBracket,
  char rightBracket) {
- //isTrue(leftBracket != rightBracket, "Both brackets cannot be the same");
+ isTrue(!this.flattenMode.equals(FlattenMode.MONGODB) && (leftBracket != rightBracket), "Both brackets cannot be the same");
  isTrue(!Character.toString(leftBracket).matches(illegalBracketsRegex()),
  "Left bracket contains illegal chracter(%s)",
  Character.toString(leftBracket));
@@ -358,15 +366,19 @@ private String computeKey() {
  sb.append(policy.getCharSequenceTranslator().translate(key));
  sb.append('\\');
  sb.append('"');
- //sb.append(rightBracket);
+ if(!this.flattenMode.equals(FlattenMode.MONGODB)) {
+ sb.append(rightBracket);
+ }
  } else {
  if (sb.length() != 0) sb.append(separator);
  sb.append(policy.getCharSequenceTranslator().translate(key));
  }
  } else { // JsonValue
  sb.append(leftBracket);
  sb.append(iter.getIndex());
- // sb.append(rightBracket);
+ if(!this.flattenMode.equals(FlattenMode.MONGODB)) {
+ sb.append(rightBracket);
+ }
  }
  }
 

diff --git a/src/test/java/com/github/wnameless/json/flattener/JsonFlattenerTest.java b/src/test/java/com/github/wnameless/json/flattener/JsonFlattenerTest.java
@@ -254,10 +254,8 @@ public void testWithLeftAndRightBracket() {
  }
 
  @Test
- @Ignore
  public void testWithLeftAndRightBracketsExceptions() {
  String json = "{\"abc\":{\"def\":123}}";
-
  try {
  new JsonFlattener(json).withLeftAndRightBrackets('#', '#');
  fail();
@@ -333,6 +331,7 @@ public void testRootInMap() {
  root.get(0).get(0));
  }
 
+
  @Test
  public void testPrintMode() throws IOException {
  URL url = Resources.getResource("test.json");
@@ -439,4 +438,18 @@ public void testInitByReader() throws IOException {
  assertEquals(jf, new JsonFlattener(json));
  }
 
+ @Test
+ public void testMongoFlattening() throws IOException {
+ URL url = Resources.getResource("test_mongo.json");
+ String src = Resources.toString(url, Charsets.UTF_8);
+
+ URL urlMongo = Resources.getResource("test_mongo_flattened.json");
+ String expectedJson = Resources.toString(urlMongo, Charsets.UTF_8);
+
+ String flattened = new JsonFlattener(src)
+ .withFlattenMode(FlattenMode.MONGODB).withPrintMode(PrintMode.PRETTY).flatten();
+
+ assertEquals(expectedJson, flattened);
+ }
+
 }
diff --git a/src/test/resources/test_mongo.json b/src/test/resources/test_mongo.json
@@ -0,0 +1,30 @@
+{
+ "glossary": {
+ "title": "example glossary",
+ "GlossDiv": {
+ "title": "S",
+ "GlossList": {
+ "GlossEntry": {
+ "ID": "SGML",
+ "SortAs": "SGML",
+ "GlossTerm": "Standard Generalized Markup Language",
+ "Acronym": "SGML",
+ "Abbrev": "ISO 8879:1986",
+ "GlossDef": {
+ "para": "A meta-markup language, used to create markup languages such as DocBook.",
+ "GlossSeeAlso": ["GML", "XML"]
+ },
+ "GlossSee": "markup",
+ "GlossArray" : [
+ {
+ "a" : 1
+ },
+ {
+ "a" : 2
+ }
+ ]
+ }
+ }
+ }
+ }
+}
diff --git a/src/test/resources/test_mongo_flattened.json b/src/test/resources/test_mongo_flattened.json
@@ -0,0 +1,15 @@
+{
+ "glossary.title": "example glossary",
+ "glossary.GlossDiv.title": "S",
+ "glossary.GlossDiv.GlossList.GlossEntry.ID": "SGML",
+ "glossary.GlossDiv.GlossList.GlossEntry.SortAs": "SGML",
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossTerm": "Standard Generalized Markup Language",
+ "glossary.GlossDiv.GlossList.GlossEntry.Acronym": "SGML",
+ "glossary.GlossDiv.GlossList.GlossEntry.Abbrev": "ISO 8879:1986",
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossDef.para": "A meta-markup language, used to create markup languages such as DocBook.",
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossDef.GlossSeeAlso.0": "GML",
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossDef.GlossSeeAlso.1": "XML",
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossSee": "markup",
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossArray.0.a": 1,
+ "glossary.GlossDiv.GlossList.GlossEntry.GlossArray.1.a": 2
+}