Made JSON output type simple JSON, Added EJSON and Fixed XML when goi…

…ng to a file.
johnlpage · Dec 21, 2024 · 468a410 · 468a410
1 parent c8cf9d0
commit 468a410
Showing 1 changed file with 26 additions and 4 deletions.
diff --git a/src/main/java/com/johnlpage/mongosyphon/DocumentGenerator.java b/src/main/java/com/johnlpage/mongosyphon/DocumentGenerator.java
@@ -3,11 +3,15 @@
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.PrintStream;
+import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 
 import org.bson.Document;
+import org.bson.json.Converter;
+import org.bson.json.JsonMode;
+import org.bson.json.JsonWriterSettings;
 import org.json.JSONObject;
 import org.json.XML;
 import org.slf4j.Logger;
@@ -156,6 +160,18 @@ public void runConversion() {
 
 		long lasttime = System.currentTimeMillis();
 
+		   // Define custom date converter
+        Converter<Long> dateToStringConverter = (value, writer) -> {
+            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'");
+            writer.writeString(sdf.format(value));
+        };
+
+ 		JsonWriterSettings simpleJSONSettings = JsonWriterSettings.builder()
+                .outputMode(JsonMode.RELAXED)
+                .dateTimeConverter(dateToStringConverter)
+                .build();
+
+
 		Document doc = getNext();
 
 		int lastcount = 0;
@@ -179,21 +195,27 @@ public void runConversion() {
 					docGens.put(subsectionName, subgen);
 				}
 				subgen.runConversion();
-			} else if (targetMode.equalsIgnoreCase("JSON")) {
-				// TODO: This can be way nicer JSON
+			} else if (targetMode.equalsIgnoreCase("EJSON")) {
+				// Defaults to EJSON
 				if(outputStream== null){
 				System.out.println(doc.toJson());
 				} else {
 					outputStream.println(doc.toJson());
 				}
-			} else if (targetMode.equalsIgnoreCase("XML")) {
+			} else if (targetMode.equalsIgnoreCase("JSON")) {
+				if(outputStream== null){
+				System.out.println(doc.toJson(simpleJSONSettings));
+				} else {
+					outputStream.println(doc.toJson(simpleJSONSettings));
+				}
+			}  else if (targetMode.equalsIgnoreCase("XML")) {
 				// TODO: Add seom XML conversion options
 				// As this is poor
 				String xml = XML.toString(new JSONObject(doc));
 				if(outputStream == null) {
 				System.out.println(xml); }
 				else {
-					outputStream.println(doc.toJson());
+					outputStream.println(xml);
 				}
 			} else {
 				if (targetMode.equalsIgnoreCase("insert")) {