@vbarbarosh/node-helpers 3.47.1 → 3.47.2

package/package.json

@@ -2,7 +2,7 @@
     "author": "Vladimir Barbarosh <vladimir.barbarosh@gmail.com>",
     "license": "MIT",
     "name": "@vbarbarosh/node-helpers",
-    "version": "3.47.1",
+    "version": "3.47.2",
     "description": "A set of helpers for JavaScript/Node.js",
     "files": [
         "src"

package/src/stream_discard.js

@@ -1,8 +1,9 @@
 const stream = require('stream');
 
-function stream_discard()
+function stream_discard({objectMode = false} = {})
 {
     return new stream.Writable({
+        objectMode,
         write: function (buffer, encoding, callback) {
             callback();
         },

package/src/stream_map_parallel.js

@@ -53,7 +53,7 @@ function stream_map_parallel({handler, concurrency})
             callback_next = null;
             tmp();
         }
-        if (callback_flush && !buf.length) {
+        if (callback_flush && !running) {
             callback_flush();
         }
     }

package/src/stream_map_parallel.test.js

@@ -0,0 +1,26 @@
+const Promise = require('bluebird');
+const assert = require('assert');
+const stream = require('stream');
+const stream_each = require('./stream_each');
+const stream_map_parallel = require('./stream_map_parallel');
+
+describe('stream_map_parallel', function () {
+    it('basic', async function () {
+        const r = Array(100).fill().map((v,i) => i);
+        const out = [];
+        await stream.promises.pipeline(
+            stream.Readable.from(r),
+            stream_map_parallel({
+                concurrency: 5,
+                handler: async function (num) {
+                    await Promise.delay(0);
+                    return num;
+                },
+            }),
+            stream_each(function (item) {
+                out.push(item);
+            }),
+        );
+        assert.deepStrictEqual(out, r);
+    });
+});