add command-line flag to support very big json objs. (#66)

mholt · Aug 14, 2019 · 5036fd4 · 5036fd4
1 parent f223e87
commit 5036fd4
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/json-to-go.js b/json-to-go.js
@@ -398,10 +398,21 @@ function jsonToGo(json, typename, flatten = true)
 
 if (typeof module != 'undefined') {
     if (!module.parent) {
-        process.stdin.on('data', function(buf) {
-            const json = buf.toString('utf8')
-            console.log(jsonToGo(json).go)
-        })
+        if (process.argv.length > 2 && process.argv[2] === '-big') {
+            bufs = []
+            process.stdin.on('data', function(buf) {
+                bufs.push(buf)
+            })
+            process.stdin.on('end', function() {
+                const json = Buffer.concat(bufs).toString('utf8')
+                console.log(jsonToGo(json).go)
+            })
+        } else {
+            process.stdin.on('data', function(buf) {
+                const json = buf.toString('utf8')
+                console.log(jsonToGo(json).go)
+            })
+        }
     } else {
         module.exports = jsonToGo
     }