Compare commits

..

8 Commits

Author SHA1 Message Date
Lewis Diamond
2fb7942d9c Update package to use itself in dev, update samples 2020-08-05 19:08:13 -04:00
Lewis Diamond
fc5b96f725 Merge branch 'dev' into master 2020-08-05 10:42:33 -04:00
Lewis Diamond
491ec146a6 Update readme 2020-08-05 10:41:23 -04:00
Lewis Diamond
38bb853181 Set version to 0.5.0, diverge from Mhysa's versioning 2020-08-03 18:56:44 -04:00
Lewis Diamond
87c44de799 Default to an objectMode: true instance but allow creating other instances with other defaults 2020-08-03 18:54:40 -04:00
Lewis Diamond
7f1309f45f update pkg name 2020-08-01 22:09:00 -04:00
Lewis Diamond
4aac05c9c0 Adding some partial documentation. Fixing batching timeout 2020-07-31 23:32:26 -04:00
Lewis Diamond
caaabf4427 Make the package name stromjs because strom is taken 2020-07-31 22:17:57 -04:00
50 changed files with 370 additions and 136 deletions

140
README.md
View File

@ -2,10 +2,13 @@
**Dependency-free stream utils for Node.js** **Dependency-free stream utils for Node.js**
<sub>Released under the [MIT](https://git.lewis.id/strom/blob/master/LICENSE) license.</sub> <sub>Released under the [MIT](LICENSE) license.</sub>
```sh ```sh
yarn add strom // Name TBD yarn add stromjs
```
```sh
npm add stromjs
``` ```
## fromArray(array) ## fromArray(array)
@ -23,7 +26,7 @@ strom.fromArray(["a", "b"])
## map(mapper, options) ## map(mapper, options)
Return a `ReadWrite` stream that maps streamed chunks Returns a `ReadWrite` stream that maps streamed chunks
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -41,7 +44,7 @@ strom.fromArray(["a", "b"])
## flatMap(mapper, options) ## flatMap(mapper, options)
Return a `ReadWrite` stream that flat maps streamed chunks Returns a `ReadWrite` stream that flat maps streamed chunks
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -59,7 +62,7 @@ strom.fromArray(["a", "AA"])
## filter(predicate, options) ## filter(predicate, options)
Return a `ReadWrite` stream that filters out streamed chunks for which the predicate does not hold Returns a `ReadWrite` stream that filters out streamed chunks for which the predicate does not hold
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -76,7 +79,7 @@ strom.fromArray(["a", "b", "c"])
## reduce(iteratee, initialValue, options) ## reduce(iteratee, initialValue, options)
Return a `ReadWrite` stream that reduces streamed chunks down to a single value and yield that Returns a `ReadWrite` stream that reduces streamed chunks down to a single value and yield that
value value
| Param | Type | Description | | Param | Type | Description |
@ -97,7 +100,7 @@ strom.fromArray(["a", "b", "cc"])
## split(separator) ## split(separator)
Return a `ReadWrite` stream that splits streamed chunks using the given separator Returns a `ReadWrite` stream that splits streamed chunks using the given separator
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -115,7 +118,7 @@ strom.fromArray(["a,b", "c,d"])
## join(separator) ## join(separator)
Return a `ReadWrite` stream that joins streamed chunks using the given separator Returns a `ReadWrite` stream that joins streamed chunks using the given separator
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -132,7 +135,7 @@ strom.fromArray(["a", "b", "c"])
## replace(searchValue, replaceValue) ## replace(searchValue, replaceValue)
Return a `ReadWrite` stream that replaces occurrences of the given string or regular expression in Returns a `ReadWrite` stream that replaces occurrences of the given string or regular expression in
the streamed chunks with the specified replacement string the streamed chunks with the specified replacement string
| Param | Type | Description | | Param | Type | Description |
@ -151,7 +154,7 @@ strom.fromArray(["a1", "b22", "c333"])
## parse() ## parse()
Return a `ReadWrite` stream that parses the streamed chunks as JSON Returns a `ReadWrite` stream that parses the streamed chunks as JSON
```js ```js
strom.fromArray(['{ "a": "b" }']) strom.fromArray(['{ "a": "b" }'])
@ -162,7 +165,7 @@ strom.fromArray(['{ "a": "b" }'])
## stringify() ## stringify()
Return a `ReadWrite` stream that stringifies the streamed chunks to JSON Returns a `ReadWrite` stream that stringifies the streamed chunks to JSON
```js ```js
strom.fromArray([{ a: "b" }]) strom.fromArray([{ a: "b" }])
@ -173,7 +176,7 @@ strom.fromArray([{ a: "b" }])
## collect(options) ## collect(options)
Return a `ReadWrite` stream that collects streamed chunks into an array or buffer Returns a `ReadWrite` stream that collects streamed chunks into an array or buffer
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -189,7 +192,7 @@ strom.fromArray(["a", "b", "c"])
## concat(streams) ## concat(streams)
Return a `Readable` stream of readable streams concatenated together Returns a `Readable` stream of readable streams concatenated together
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -210,7 +213,7 @@ source2.push(null);
## merge(streams) ## merge(streams)
Return a `Readable` stream of readable streams merged together in chunk arrival order Returns a `Readable` stream of readable streams merged together in chunk arrival order
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -231,7 +234,7 @@ setTimeout(() => source2.push(null), 50);
## duplex(writable, readable) ## duplex(writable, readable)
Return a `Duplex` stream from a writable stream that is assumed to somehow, when written to, Returns a `Duplex` stream from a writable stream that is assumed to somehow, when written to,
cause the given readable stream to yield chunks cause the given readable stream to yield chunks
| Param | Type | Description | | Param | Type | Description |
@ -249,7 +252,7 @@ strom.fromArray(["a", "b", "c"])
## child(childProcess) ## child(childProcess)
Return a `Duplex` stream from a child process' stdin and stdout Returns a `Duplex` stream from a child process' stdin and stdout
| Param | Type | Description | | Param | Type | Description |
| --- | --- | --- | | --- | --- | --- |
@ -265,7 +268,7 @@ strom.fromArray(["a", "b", "c"])
## last(readable) ## last(readable)
Return a `Promise` resolving to the last streamed chunk of the given readable stream, after it has Returns a `Promise` resolving to the last streamed chunk of the given readable stream, after it has
ended ended
| Param | Type | Description | | Param | Type | Description |
@ -280,3 +283,106 @@ let f = async () => {
f(); f();
// c is printed out // c is printed out
``` ```
## accumulator(flushStrategy, iteratee, options)
TO BE DOCUMENTED
## batch(batchSize, maxBatchAge, options)
Returns a `Transform` stream which produces all incoming data in batches of size `batchSize`.
| Param | Type | Description |
| --- | --- | --- |
| `batchSize` | `number` | Size of the batches to be produced |
| `maxBatchAge` | `number` | Maximum number of milliseconds a message will be queued for. E.g. a batch will be produced before reaching `batchSize` if the first message queued is `maxBatchAge` ms old or more |
| `options` | `TransformOptions` | Options passed down to the Transform object |
```js
strom.fromArray(["a", "b", "c", "d"])
.pipe(strom.batch(3, 500))
.pipe(process.stdout);
// ["a","b","c"]
// ["d"] //After 500ms
```
## compose(streams, errorCb, options)
Returns a `Transform` stream which consists of all `streams` but behaves as a single stream. The returned stream can be piped into and from transparently.
| Param | Type | Description |
| --- | --- | --- |
| `streams` | `Array` | Streams to be composed |
| `errorCb` | `(err: Error) => void` | Function called when an error occurs in any of the streams |
| `options` | `TransformOptions` | Options passed down to the Transform object |
```js
const composed = strom.compose([
strom.split(),
strom.map(data => data.trim()),
strom.filter(str => !!str),
strom.parse(),
strom.flatMap(data => data),
strom.stringify(),
]);
const data = ["[1,2,3] \n [4,5,6] ", "\n [7,8,9] \n\n"];
strom.fromArray(data).pipe(composed).pipe(process.stdout);
// 123456789
```
## demux(pipelineConstructor, demuxBy, options)
TO BE DOCUMENTED
## parallelMap(mapper, parallel, sleepTime, options)
Returns a `Transform` stream which maps incoming data through the async mapper with the given parallelism.
| Param | Type | Description | Default |
| --- | --- | --- | --- |
| `mapper` | `async (chunk: T, encoding: string) => R` | Mapper function, mapping each (chunk, encoding) to a new chunk (non-async will not be parallelized) | -- |
| `parallel` | `number` | Number of concurrent executions of the mapper allowed | 10 |
| `sleepTime` | `number` | Number of milliseconds to wait before testing if more messages can be processed | 1 |
```js
function sleep(time) {
return time > 0 ? new Promise(resolve => setTimeout(resolve, time)) : null;
}
strom
.fromArray([1, 2, 3, 4, 6, 8])
.pipe(
strom.parallelMap(async d => {
await sleep(10000 - d * 1000);
return `${d}`;
}, 3),
)
.pipe(process.stdout);
// 321864
```
## rate()
```js
const strom = require("stromjs").strom();
function sleep(time) {
return time > 0 ? new Promise(resolve => setTimeout(resolve, time)) : null;
}
const rate = strom.rate(2, 1, { behavior: 1 });
rate.pipe(strom.map(x => console.log(x)));
async function produce() {
rate.write(1);
await sleep(500);
rate.write(2);
await sleep(500);
rate.write(3);
rate.write(4);
rate.write(5);
await sleep(500);
rate.write(6);
}
produce();
```

View File

@ -1,7 +1,7 @@
{ {
"name": "strom", "name": "stromjs",
"version": "2.0.0", "version": "0.5.1",
"description": "Streams utils for Node.js", "description": "Dependency-free streams utils for Node.js",
"keywords": [ "keywords": [
"promise", "promise",
"stream", "stream",
@ -9,15 +9,18 @@
], ],
"contributors": [ "contributors": [
{ {
"name": "Wenzil" "name": "Sami Turcotte",
"url": "https://github.com/Wenzil"
}, },
{ {
"name": "jerry", "name": "Jerry Kurian",
"email": "jerry@jogogo.co" "email": "jerrykurian@protonmail.com",
"url": "https://github.com/jkurian"
}, },
{ {
"name": "lewis", "name": "Lewis Diamond",
"email": "lewis@jogogo.co" "email": "stromjs@lewisdiamond.com",
"url": "https://github.com/lewisdiamond"
} }
], ],
"license": "MIT", "license": "MIT",
@ -27,7 +30,7 @@
"dist" "dist"
], ],
"repository": { "repository": {
"url": "git@git.lewis.id:ldiamond/strom.git", "url": "git@git.lewis.id:strom/strom.git",
"type": "git" "type": "git"
}, },
"scripts": { "scripts": {
@ -44,7 +47,7 @@
"@types/sinon": "^7.0.13", "@types/sinon": "^7.0.13",
"ava": "^2.4.0", "ava": "^2.4.0",
"chai": "^4.2.0", "chai": "^4.2.0",
"mhysa": "./", "stromjs": "./",
"prettier": "^1.14.3", "prettier": "^1.14.3",
"sinon": "^7.4.2", "sinon": "^7.4.2",
"ts-node": "^8.3.0", "ts-node": "^8.3.0",

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "b", "c"]) strom
.fromArray(["a", "b", "c"])
.pipe(strom.map(s => Promise.resolve(s + s))) .pipe(strom.map(s => Promise.resolve(s + s)))
.pipe(strom.flatMap(s => Promise.resolve([s, s.toUpperCase()]))) .pipe(strom.flatMap(s => Promise.resolve([s, s.toUpperCase()])))
.pipe(strom.filter(s => Promise.resolve(s !== "bb"))) .pipe(strom.filter(s => Promise.resolve(s !== "bb")))

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
const catProcess = require("child_process").exec("grep -o ab"); const catProcess = require("child_process").exec("grep -o ab");
strom.fromArray(["a", "b", "c"]) strom
.fromArray(["a", "b", "c"])
.pipe(strom.child(catProcess)) .pipe(strom.child(catProcess))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,5 +1,6 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "b", "c"]) strom
.fromArray(["a", "b", "c"])
.pipe(strom.collect({ objectMode: true })) .pipe(strom.collect({ objectMode: true }))
.on("data", object => console.log(object)); .on("data", object => console.log(object));

View File

@ -1,5 +1,5 @@
const { Readable } = require("stream"); const { Readable } = require("stream");
const strom = require("strom").strom(); const strom = require("stromjs");
const source1 = new Readable(); const source1 = new Readable();
const source2 = new Readable(); const source2 = new Readable();

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
const catProcess = require("child_process").exec("grep -o ab"); const catProcess = require("child_process").exec("grep -o ab");
strom.fromArray(["a", "b", "c"]) strom
.fromArray(["a", "b", "c"])
.pipe(strom.duplex(catProcess.stdin, catProcess.stdout)) .pipe(strom.duplex(catProcess.stdin, catProcess.stdout))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "b", "c"]) strom
.fromArray(["a", "b", "c"])
.pipe(strom.filter(s => s !== "b")) .pipe(strom.filter(s => s !== "b"))
.pipe(strom.join(",")) .pipe(strom.join(","))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "AA"]) strom
.fromArray(["a", "AA"])
.pipe(strom.flatMap(s => new Array(s.length).fill(s))) .pipe(strom.flatMap(s => new Array(s.length).fill(s)))
.pipe(strom.join(",")) .pipe(strom.join(","))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,5 +1,6 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "b", "c"]) strom
.fromArray(["a", "b", "c"])
.pipe(strom.join(",")) .pipe(strom.join(","))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,4 +1,4 @@
const strom = require("strom").strom(); const strom = require("stromjs");
let f = async () => { let f = async () => {
const source = strom.fromArray(["a", "b", "c"]); const source = strom.fromArray(["a", "b", "c"]);

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "b"]) strom
.fromArray(["a", "b"])
.pipe(strom.map(s => s.toUpperCase())) .pipe(strom.map(s => s.toUpperCase()))
.pipe(strom.join(",")) .pipe(strom.join(","))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,5 +1,5 @@
const { Readable } = require("stream"); const { Readable } = require("stream");
const strom = require("strom").strom(); const strom = require("stromjs");
const source1 = new Readable({ read() {} }); const source1 = new Readable({ read() {} });
const source2 = new Readable({ read() {} }); const source2 = new Readable({ read() {} });

15
samples/parallelMap.js Normal file
View File

@ -0,0 +1,15 @@
const strom = require("stromjs");
function sleep(time) {
return time > 0 ? new Promise(resolve => setTimeout(resolve, time)) : null;
}
strom
.fromArray([1, 2, 3, 4, 6, 8])
.pipe(
strom.parallelMap(async d => {
await sleep(10000 - d * 1000);
return `${d}`;
}, 3),
)
.pipe(process.stdout);

View File

@ -1,5 +1,6 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(['{ "a": "b" }']) strom
.fromArray(['{ "a": "b" }'])
.pipe(strom.parse()) .pipe(strom.parse())
.on("data", object => console.log(object)); .on("data", object => console.log(object));

21
samples/rate.js Normal file
View File

@ -0,0 +1,21 @@
const strom = require("stromjs");
function sleep(time) {
return time > 0 ? new Promise(resolve => setTimeout(resolve, time)) : null;
}
const rate = strom.rate(2, 1, { behavior: 1 });
rate.pipe(strom.map(x => console.log(x)));
async function produce() {
rate.write(1);
await sleep(500);
rate.write(2);
await sleep(500);
rate.write(3);
rate.write(4);
rate.write(5);
await sleep(500);
rate.write(6);
}
produce();

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a", "b", "cc"]) strom
.fromArray(["a", "b", "cc"])
.pipe(strom.reduce((acc, s) => ({ ...acc, [s]: s.length }), {})) .pipe(strom.reduce((acc, s) => ({ ...acc, [s]: s.length }), {}))
.pipe(strom.stringify()) .pipe(strom.stringify())
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,5 +1,6 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a1", "b22", "c333"]) strom
.fromArray(["a1", "b22", "c333"])
.pipe(strom.replace(/b\d+/, "B")) .pipe(strom.replace(/b\d+/, "B"))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,6 +1,7 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray(["a,b", "c,d"]) strom
.fromArray(["a,b", "c,d"])
.pipe(strom.split(",")) .pipe(strom.split(","))
.pipe(strom.join("|")) .pipe(strom.join("|"))
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -1,5 +1,6 @@
const strom = require("strom").strom(); const strom = require("stromjs");
strom.fromArray([{ a: "b" }]) strom
.fromArray([{ a: "b" }])
.pipe(strom.stringify()) .pipe(strom.stringify())
.pipe(process.stdout); .pipe(process.stdout);

View File

@ -2,7 +2,7 @@ import { Transform, TransformOptions } from "stream";
export function batch( export function batch(
batchSize: number = 1000, batchSize: number = 1000,
maxBatchAge: number = 500, maxBatchAge: number = 0,
options: TransformOptions = {}, options: TransformOptions = {},
): Transform { ): Transform {
let buffer: any[] = []; let buffer: any[] = [];
@ -23,7 +23,7 @@ export function batch(
buffer.push(chunk); buffer.push(chunk);
if (buffer.length === batchSize) { if (buffer.length === batchSize) {
sendChunk(this); sendChunk(this);
} else { } else if (maxBatchAge) {
if (timer === null) { if (timer === null) {
timer = setInterval(() => { timer = setInterval(() => {
sendChunk(this); sendChunk(this);

View File

@ -263,5 +263,10 @@ export function strom(defaultOptions: TransformOptions = { objectMode: true }) {
* @param options Writable stream options * @param options Writable stream options
*/ */
demux: withDefaultOptions(2, demux), demux: withDefaultOptions(2, demux),
/**
* Create a new strom instance overriding the defaults
*/
instance: strom,
}; };
} }

View File

@ -2,20 +2,41 @@ import { Transform, TransformOptions } from "stream";
import { performance } from "perf_hooks"; import { performance } from "perf_hooks";
import { sleep } from "../helpers"; import { sleep } from "../helpers";
export enum Behavior {
BUFFER = 0,
DROP = 1,
}
export interface RateOptions {
window?: number;
behavior?: Behavior;
}
export function rate( export function rate(
targetRate: number = 50, targetRate: number = 50,
period: number = 1, period: number = 1,
options?: TransformOptions, options?: TransformOptions & RateOptions,
): Transform { ): Transform {
const deltaMS = ((1 / targetRate) * 1000) / period; // Skip a full period const deltaMS = ((1 / targetRate) * 1000) / period; // Skip a full period
let total = 0; let total = 0;
const start = performance.now(); const window = options?.window || Infinity;
const behavior = options?.behavior || Behavior.BUFFER;
let start = performance.now();
return new Transform({ return new Transform({
...options, ...options,
async transform(data, encoding, callback) { async transform(data, encoding, callback) {
const currentRate = (total / (performance.now() - start)) * 1000; const now = performance.now();
if (now - start >= window) {
start = now - window;
}
const currentRate = (total / (now - start)) * 1000;
if (targetRate && currentRate > targetRate) { if (targetRate && currentRate > targetRate) {
await sleep(deltaMS); if (behavior === Behavior.DROP) {
callback(undefined);
return;
} else {
await sleep(deltaMS);
}
} }
total += 1; total += 1;
callback(undefined, data); callback(undefined, data);

View File

@ -1,2 +1,29 @@
export { strom } from "./functions"; import { strom } from "./functions";
export * from "./utils"; export * from "./utils";
export const {
fromArray,
map,
flatMap,
filter,
reduce,
split,
join,
replace,
parse,
stringify,
collect,
concat,
merge,
duplex,
child,
last,
batch,
unbatch,
rate,
parallelMap,
accumulator,
accumulatorBy,
compose,
demux,
instance,
} = strom();

View File

@ -1,10 +1,9 @@
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { Readable } from "stream"; import { Readable } from "stream";
import { strom } from "../src"; import { accumulator, accumulatorBy } from "../src";
import { FlushStrategy } from "../src/functions/accumulator"; import { FlushStrategy } from "../src/functions/accumulator";
import { performance } from "perf_hooks"; import { performance } from "perf_hooks";
const { accumulator, accumulatorBy } = strom({ objectMode: true });
test.cb("accumulator() rolling", t => { test.cb("accumulator() rolling", t => {
t.plan(3); t.plan(3);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { batch, map, fromArray } from "../src";
const { batch, map, fromArray } = strom({ objectMode: true });
test.cb("batch() batches chunks together", t => { test.cb("batch() batches chunks together", t => {
t.plan(3); t.plan(3);
@ -39,7 +38,7 @@ test.cb("batch() yields a batch after the timeout", t => {
const expectedElements = [["a", "b"], ["c"], ["d"]]; const expectedElements = [["a", "b"], ["c"], ["d"]];
let i = 0; let i = 0;
source source
.pipe(batch(3)) .pipe(batch(3, 500))
.on("data", (element: string[]) => { .on("data", (element: string[]) => {
t.deepEqual(element, expectedElements[i]); t.deepEqual(element, expectedElements[i]);
i++; i++;

View File

@ -2,8 +2,7 @@ import * as cp from "child_process";
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { child } from "../src";
const { child } = strom();
test.cb( test.cb(
"child() allows easily writing to child process stdin and reading from its stdout", "child() allows easily writing to child process stdin and reading from its stdout",

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { collect } from "../src";
const { collect } = strom();
test.cb( test.cb(
"collect() collects streamed elements into an array (object, flowing mode)", "collect() collects streamed elements into an array (object, flowing mode)",

View File

@ -2,9 +2,8 @@ import * as test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { sleep } from "../src/helpers"; import { sleep } from "../src/helpers";
import { Readable, Writable } from "stream"; import { Readable, Writable } from "stream";
import { strom } from "../src"; import { compose, map, fromArray } from "../src";
import { performance } from "perf_hooks"; import { performance } from "perf_hooks";
const { compose, map, fromArray } = strom({ objectMode: true });
test.cb("compose() chains two streams together in the correct order", t => { test.cb("compose() chains two streams together in the correct order", t => {
t.plan(3); t.plan(3);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { concat, collect } from "../src";
const { concat, collect } = strom();
test.cb( test.cb(
"concat() concatenates multiple readable streams (object, flowing mode)", "concat() concatenates multiple readable streams (object, flowing mode)",

View File

@ -1,21 +1,30 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { strom } from "../src"; import { batch as _batch, instance as strom } from "../src";
const withDefaultOptions = strom({ objectMode: true }); const withDefaultOptions = strom({ objectMode: false });
const withoutOptions = strom();
test("strom instances can have default options", t => { test("strom instances can have default options", t => {
let batch = withDefaultOptions.batch(); let batch = withDefaultOptions.batch();
t.true(batch._readableState.objectMode); t.false(batch._readableState.objectMode);
t.true(batch._writableState.objectMode); t.false(batch._writableState.objectMode);
batch = withDefaultOptions.batch(3); batch = withDefaultOptions.batch(3);
t.true(batch._readableState.objectMode); t.false(batch._readableState.objectMode);
t.true(batch._writableState.objectMode); t.false(batch._writableState.objectMode);
batch = withDefaultOptions.batch(3, 1); batch = withDefaultOptions.batch(3, 1);
t.false(batch._readableState.objectMode);
t.false(batch._writableState.objectMode);
batch = withDefaultOptions.batch(3, 1, { objectMode: true });
t.true(batch._readableState.objectMode); t.true(batch._readableState.objectMode);
t.true(batch._writableState.objectMode); t.true(batch._writableState.objectMode);
batch = withDefaultOptions.batch(3, 1, { objectMode: false });
batch = _batch(3);
t.true(batch._readableState.objectMode);
t.true(batch._writableState.objectMode);
batch = _batch(3, 1);
t.true(batch._readableState.objectMode);
t.true(batch._writableState.objectMode);
batch = _batch(3, 1, { objectMode: false });
t.false(batch._readableState.objectMode); t.false(batch._readableState.objectMode);
t.false(batch._writableState.objectMode); t.false(batch._writableState.objectMode);
}); });

View File

@ -1,11 +1,10 @@
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { demux, map, fromArray } from "../src";
import { Writable, Readable } from "stream"; import { Writable, Readable } from "stream";
import * as sinon from "sinon"; import * as sinon from "sinon";
import { sleep } from "../src/helpers"; import { sleep } from "../src/helpers";
import { performance } from "perf_hooks"; import { performance } from "perf_hooks";
const { demux, map, fromArray } = strom({ objectMode: true });
interface Test { interface Test {
key: string; key: string;

View File

@ -2,8 +2,7 @@ import * as cp from "child_process";
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { duplex } from "../src";
const { duplex } = strom();
test.cb( test.cb(
"duplex() combines a writable and readable stream into a ReadWrite stream", "duplex() combines a writable and readable stream into a ReadWrite stream",

View File

@ -1,8 +1,7 @@
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { Readable } from "stream"; import { Readable } from "stream";
import { strom } from "../src"; import { filter } from "../src";
const { filter } = strom();
test.cb("filter() filters elements synchronously", t => { test.cb("filter() filters elements synchronously", t => {
t.plan(2); t.plan(2);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { flatMap } from "../src";
const { flatMap } = strom({ objectMode: true });
test.cb("flatMap() maps elements synchronously", t => { test.cb("flatMap() maps elements synchronously", t => {
t.plan(6); t.plan(6);

View File

@ -1,7 +1,6 @@
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { fromArray } from "../src";
const { fromArray } = strom();
test.cb("fromArray() streams array elements in flowing mode", t => { test.cb("fromArray() streams array elements in flowing mode", t => {
t.plan(3); t.plan(3);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { join } from "../src";
const { join } = strom();
test.cb("join() joins chunks using the specified separator", t => { test.cb("join() joins chunks using the specified separator", t => {
t.plan(9); t.plan(9);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { last } from "../src";
const { last } = strom();
test("last() resolves to the last chunk streamed by the given readable stream", async t => { test("last() resolves to the last chunk streamed by the given readable stream", async t => {
const source = new Readable({ objectMode: true }); const source = new Readable({ objectMode: true });

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { map } from "../src";
const { map } = strom();
test.cb("map() maps elements synchronously", t => { test.cb("map() maps elements synchronously", t => {
t.plan(3); t.plan(3);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { merge } from "../src";
const { merge } = strom();
test.cb( test.cb(
"merge() merges multiple readable streams in chunk arrival order", "merge() merges multiple readable streams in chunk arrival order",

View File

@ -2,9 +2,8 @@ import { Readable } from "stream";
import { performance } from "perf_hooks"; import { performance } from "perf_hooks";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { parallelMap } from "../src";
import { sleep } from "../src/helpers"; import { sleep } from "../src/helpers";
const { parallelMap } = strom({ objectMode: true });
test.cb("parallelMap() parallel mapping", t => { test.cb("parallelMap() parallel mapping", t => {
t.plan(6); t.plan(6);

View File

@ -1,8 +1,7 @@
import { Readable, finished } from "stream"; import { Readable, finished } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { parse } from "../src";
const { parse } = strom();
test.cb("parse() parses the streamed elements as JSON", t => { test.cb("parse() parses the streamed elements as JSON", t => {
t.plan(3); t.plan(3);

View File

@ -1,12 +1,11 @@
import { Readable } from "stream"; import { Readable } from "stream";
import { performance } from "perf_hooks"; import { performance } from "perf_hooks";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { rate } from "../src";
import { strom } from "../src"; import { sleep } from "../src/helpers";
const { rate } = strom({ objectMode: true });
test.cb("rate() sends data at a rate of 150", t => { test.cb("rate() sends data at a rate of 150", t => {
t.plan(5); t.plan(15);
const targetRate = 150; const targetRate = 150;
const source = new Readable({ objectMode: true }); const source = new Readable({ objectMode: true });
const expectedElements = ["a", "b", "c", "d", "e"]; const expectedElements = ["a", "b", "c", "d", "e"];
@ -15,10 +14,10 @@ test.cb("rate() sends data at a rate of 150", t => {
source source
.pipe(rate(targetRate)) .pipe(rate(targetRate))
.on("data", (element: string[]) => { .on("data", (element: string) => {
const currentRate = (i / (performance.now() - start)) * 1000; const currentRate = (i / (performance.now() - start)) * 1000;
expect(element).to.deep.equal(expectedElements[i]); t.is(element, expectedElements[i]);
expect(currentRate).lessThan(targetRate); t.true(currentRate <= targetRate);
t.pass(); t.pass();
i++; i++;
}) })
@ -34,7 +33,7 @@ test.cb("rate() sends data at a rate of 150", t => {
}); });
test.cb("rate() sends data at a rate of 50", t => { test.cb("rate() sends data at a rate of 50", t => {
t.plan(5); t.plan(15);
const targetRate = 50; const targetRate = 50;
const source = new Readable({ objectMode: true }); const source = new Readable({ objectMode: true });
const expectedElements = ["a", "b", "c", "d", "e"]; const expectedElements = ["a", "b", "c", "d", "e"];
@ -43,10 +42,10 @@ test.cb("rate() sends data at a rate of 50", t => {
source source
.pipe(rate(targetRate)) .pipe(rate(targetRate))
.on("data", (element: string[]) => { .on("data", (element: string) => {
const currentRate = (i / (performance.now() - start)) * 1000; const currentRate = (i / (performance.now() - start)) * 1000;
expect(element).to.deep.equal(expectedElements[i]); t.is(element, expectedElements[i]);
expect(currentRate).lessThan(targetRate); t.true(currentRate <= targetRate);
t.pass(); t.pass();
i++; i++;
}) })
@ -62,7 +61,7 @@ test.cb("rate() sends data at a rate of 50", t => {
}); });
test.cb("rate() sends data at a rate of 1", t => { test.cb("rate() sends data at a rate of 1", t => {
t.plan(5); t.plan(15);
const targetRate = 1; const targetRate = 1;
const source = new Readable({ objectMode: true }); const source = new Readable({ objectMode: true });
const expectedElements = ["a", "b", "c", "d", "e"]; const expectedElements = ["a", "b", "c", "d", "e"];
@ -71,10 +70,10 @@ test.cb("rate() sends data at a rate of 1", t => {
source source
.pipe(rate(targetRate)) .pipe(rate(targetRate))
.on("data", (element: string[]) => { .on("data", (element: string) => {
const currentRate = (i / (performance.now() - start)) * 1000; const currentRate = (i / (performance.now() - start)) * 1000;
expect(element).to.deep.equal(expectedElements[i]); t.is(element, expectedElements[i]);
expect(currentRate).lessThan(targetRate); t.true(currentRate <= targetRate);
t.pass(); t.pass();
i++; i++;
}) })
@ -88,3 +87,41 @@ test.cb("rate() sends data at a rate of 1", t => {
source.push("e"); source.push("e");
source.push(null); source.push(null);
}); });
test("rate() sends data at a rate of 1 and drops extra messages", async t => {
t.plan(9);
const targetRate = 1;
const source = new Readable({
objectMode: true,
read: () => {
return;
},
});
const expectedElements = ["a", "b", "e"];
const start = performance.now();
let i = 0;
let plan = 0;
source
.pipe(rate(targetRate, 1, { behavior: 1 }))
.on("data", (element: string) => {
const currentRate = (i / (performance.now() - start)) * 1000;
t.is(element, expectedElements[i]);
t.true(currentRate <= targetRate);
plan++;
t.pass();
i++;
})
.on("error", t.fail)
.on("end", t.fail);
source.push("a");
await sleep(1000);
source.push("b");
source.push("c");
source.push("d");
await sleep(1000);
source.push("e");
await sleep(1000);
source.push(null);
});

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { reduce } from "../src";
const { reduce } = strom({ objectMode: true });
test.cb("reduce() reduces elements synchronously", t => { test.cb("reduce() reduces elements synchronously", t => {
t.plan(1); t.plan(1);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { replace } from "../src";
const { replace } = strom();
test.cb( test.cb(
"replace() replaces occurrences of the given string in the streamed elements with the specified " + "replace() replaces occurrences of the given string in the streamed elements with the specified " +

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { split } from "../src";
const { split } = strom();
test.cb("split() splits chunks using the default separator (\\n)", t => { test.cb("split() splits chunks using the default separator (\\n)", t => {
t.plan(5); t.plan(5);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { stringify } from "../src";
const { stringify } = strom();
test.cb("stringify() stringifies the streamed elements as JSON", t => { test.cb("stringify() stringifies the streamed elements as JSON", t => {
t.plan(4); t.plan(4);

View File

@ -1,8 +1,7 @@
import { Readable } from "stream"; import { Readable } from "stream";
import test from "ava"; import test from "ava";
import { expect } from "chai"; import { expect } from "chai";
import { strom } from "../src"; import { unbatch, batch } from "../src";
const { unbatch, batch } = strom({ objectMode: true });
test.cb("unbatch() unbatches", t => { test.cb("unbatch() unbatches", t => {
t.plan(3); t.plan(3);

View File

@ -1,7 +1,6 @@
import test from "ava"; import test from "ava";
import { collected } from "../../src/utils"; import { collected } from "../../src/utils";
import { strom } from "../../src"; import { fromArray, collect } from "../../src";
const { fromArray, collect } = strom({ objectMode: true });
test("collected returns a promise for the first data point", async t => { test("collected returns a promise for the first data point", async t => {
const data = collected(fromArray([1, 2, 3, 4]).pipe(collect())); const data = collected(fromArray([1, 2, 3, 4]).pipe(collect()));

View File

@ -1934,9 +1934,6 @@ merge2@^1.2.3, merge2@^1.3.0:
resolved "https://npm.dev.jogogo.co/merge2/-/merge2-1.3.0.tgz#5b366ee83b2f1582c48f87e47cf1a9352103ca81" resolved "https://npm.dev.jogogo.co/merge2/-/merge2-1.3.0.tgz#5b366ee83b2f1582c48f87e47cf1a9352103ca81"
integrity sha512-2j4DAdlBOkiSZIsaXk4mTE3sRS02yBHAtfy127xRV3bQUFqXkjHCHLW6Scv7DwNRbIWNHH8zpnz9zMaKXIdvYw== integrity sha512-2j4DAdlBOkiSZIsaXk4mTE3sRS02yBHAtfy127xRV3bQUFqXkjHCHLW6Scv7DwNRbIWNHH8zpnz9zMaKXIdvYw==
mhysa@./:
version "2.0.0-alpha.1"
micromatch@^4.0.2: micromatch@^4.0.2:
version "4.0.2" version "4.0.2"
resolved "https://npm.dev.jogogo.co/micromatch/-/micromatch-4.0.2.tgz#4fcb0999bf9fbc2fcbdd212f6d629b9a56c39259" resolved "https://npm.dev.jogogo.co/micromatch/-/micromatch-4.0.2.tgz#4fcb0999bf9fbc2fcbdd212f6d629b9a56c39259"
@ -2719,6 +2716,9 @@ strip-json-comments@~2.0.1:
resolved "https://registry.yarnpkg.com/strip-json-comments/-/strip-json-comments-2.0.1.tgz#3c531942e908c2697c0ec344858c286c7ca0a60a" resolved "https://registry.yarnpkg.com/strip-json-comments/-/strip-json-comments-2.0.1.tgz#3c531942e908c2697c0ec344858c286c7ca0a60a"
integrity sha1-PFMZQukIwml8DsNEhYwobHygpgo= integrity sha1-PFMZQukIwml8DsNEhYwobHygpgo=
stromjs@./:
version "0.5.1"
supertap@^1.0.0: supertap@^1.0.0:
version "1.0.0" version "1.0.0"
resolved "https://registry.yarnpkg.com/supertap/-/supertap-1.0.0.tgz#bd9751c7fafd68c68cf8222a29892206a119fa9e" resolved "https://registry.yarnpkg.com/supertap/-/supertap-1.0.0.tgz#bd9751c7fafd68c68cf8222a29892206a119fa9e"