STREAMS-1991: Add sortedKey param to tojson function

Author: LijieZhang1998

snippets/mongocompat/mongotypes.js

@@ -256,7 +256,7 @@ Array.tojson = function(a, indent, nolint, depth) {
         indent += "\t";
 
     for (var i = 0; i < a.length; i++) {
-        s += indent + tojson(a[i], indent, nolint, depth + 1);
+        s += indent + tojson(a[i], indent, nolint, depth + 1, false);
         if (i < a.length - 1) {
             s += "," + elementSeparator;
         }
@@ -713,7 +713,7 @@ tojsononeline = function(x) {
     return tojson(x, " ", true);
 };
 
-tojson = function(x, indent, nolint, depth) {
+tojson = function(x, indent, nolint, depth, sortKeys) {
     if (x === null)
         return "null";
 
@@ -734,7 +734,7 @@ tojson = function(x, indent, nolint, depth) {
         case "boolean":
             return "" + x;
         case "object": {
-            var s = tojsonObject(x, indent, nolint, depth);
+            var s = tojsonObject(x, indent, nolint, depth, sortKeys);
             if ((nolint == null || nolint == true) && s.length < 80 &&
                 (indent == null || indent.length == 0)) {
                 s = s.replace(/[\t\r\n]+/gm, " ");
@@ -751,10 +751,13 @@ tojson = function(x, indent, nolint, depth) {
 };
 tojson.MAX_DEPTH = 100;
 
-tojsonObject = function(x, indent, nolint, depth) {
+tojsonObject = function(x, indent, nolint, depth, sortKeys) {
     if (typeof depth !== 'number') {
         depth = 0;
     }
+    if (typeof sortKeys !== 'boolean') {
+        sortKeys = false;
+    }
     var lineEnding = nolint ? " " : "\n";
     var tabSpace = nolint ? "" : "\t";
     if (typeof x !== "object") {
@@ -765,12 +768,12 @@ tojsonObject = function(x, indent, nolint, depth) {
         indent = "";
 
     if (typeof (x.tojson) == "function" && x.tojson != tojson) {
-        return x.tojson(indent, nolint, depth);
+        return x.tojson(indent, nolint, depth, sortKeys);
     }
 
     if (x.constructor && typeof (x.constructor.tojson) == "function" &&
         x.constructor.tojson != tojson) {
-        return x.constructor.tojson(x, indent, nolint, depth);
+        return x.constructor.tojson(x, indent, nolint, depth, sortKeys);
     }
 
     if (x instanceof Error) {
@@ -796,8 +799,14 @@ tojsonObject = function(x, indent, nolint, depth) {
     var keys = x;
     if (typeof (x._simpleKeys) == "function")
         keys = x._simpleKeys();
-    var fieldStrings = [];
+    var keyNames = [];
     for (var k in keys) {
+        keyNames.push(k);
+    }
+    if (sortKeys) keyNames.sort();
+
+    var fieldStrings = [];
+    for (var k of keyNames) {
         var val = x[k];
 
         // skip internal DB types to avoid issues with interceptors
@@ -806,7 +815,7 @@ tojsonObject = function(x, indent, nolint, depth) {
         if (typeof DBCollection != 'undefined' && val == DBCollection.prototype)
             continue;
 
-        fieldStrings.push(indent + "\"" + k + "\" : " + tojson(val, indent, nolint, depth + 1));
+        fieldStrings.push(indent + "\"" + k + "\" : " + tojson(val, indent, nolint, depth + 1, sortKeys));
     }
 
     if (fieldStrings.length > 0) {

snippets/mongocompat/test.js

@@ -113,3 +113,25 @@ try {
 }
 assert.strictEqual(typeof tojsonObject({ key: "value" }), 'string');
 assert.strictEqual(typeof tojsonObject([1, 2, 3]), 'string');
+
+// Test sortedkey parameter
+const unsortedObj = { z: 1, a: 2, m: 3 };
+const sortedJson = tojson(unsortedObj, "", true, 0, true);
+const unsortedJson = tojson(unsortedObj, "", true, 0, false);
+const defaultJson = tojson(unsortedObj);
+assert(sortedJson.indexOf('"a"') < sortedJson.indexOf('"m"'), 'sortedJson should be sorted alphabetically');
+assert(sortedJson.indexOf('"m"') < sortedJson.indexOf('"z"'), 'sortedJson should be sorted alphabetically');
+assert(unsortedJson.indexOf('"z"') < unsortedJson.indexOf('"a"'), 'unsortedJson should not be sorted alphabetically');
+assert(defaultJson.indexOf('"z"') < defaultJson.indexOf('"a"'), 'tojson without sortedkey should not sort keys');
+const nestedObj = { b: { y: 1, x: 2 }, a: { z: 1, a: 2 } };
+const sortedNestedJson = tojson(nestedObj, "", true, 0, true);
+assert(sortedNestedJson.indexOf('"a"') < sortedNestedJson.indexOf('"b"'), 'sortedkey=true should sort top-level keys');
+assert(sortedNestedJson.indexOf('"a" :') < sortedNestedJson.indexOf('"z" :'), 'sortedkey=true should sort nested keys');
+const objWithBson = {
+    c: NumberLong(123),
+    b: ObjectId('0123456789abcdef01234567'),
+    a: NumberDecimal("1.1")
+};
+const sortedBsonJson = tojson(objWithBson, "", true, 0, true);
+assert(sortedBsonJson.indexOf('"a"') < sortedBsonJson.indexOf('"b"'), 'sortedkey=true should sort keys with BSON types');
+assert(sortedBsonJson.indexOf('"b"') < sortedBsonJson.indexOf('"c"'), 'sortedkey=true should sort keys with BSON types');