ldiamond/strom
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add buffer support to split, join, replace and parse streams

Browse Source
This commit is contained in:
Sami Turcotte
2018-12-03 09:18:49 -05:00
parent 407bb79260
commit 63c3681211
4 changed files with 53 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/index.ts
View File

@@ -208,8 +208,10 @@ export function split(
return new Transform({
readableObjectMode: true,
writableObjectMode: true,
async transform(chunk: string, encoding, callback) {
const splitted = chunk.split(separator);
transform(chunk: string | Buffer, encoding, callback) {
const asString =
chunk instanceof Buffer ? chunk.toString(encoding) : chunk;
const splitted = asString.split(separator);
if (buffered.length > 0 && splitted.length > 1) {
splitted[0] = buffered.concat(splitted[0]);
buffered = "";
@@ -233,11 +235,13 @@ export function join(separator: string): NodeJS.ReadWriteStream {
return new Transform({
readableObjectMode: true,
writableObjectMode: true,
async transform(chunk: string, encoding, callback) {
async transform(chunk: string | Buffer, encoding, callback) {
const asString =
chunk instanceof Buffer ? chunk.toString(encoding) : chunk;
if (!isFirstChunk) {
this.push(separator);
}
this.push(chunk);
this.push(asString);
isFirstChunk = false;
callback();
},
@@ -257,8 +261,10 @@ export function replace(
return new Transform({
readableObjectMode: true,
writableObjectMode: true,
transform(chunk: string, encoding, callback) {
callback(undefined, chunk.replace(searchValue, replaceValue));
transform(chunk: string | Buffer, encoding, callback) {
const asString =
chunk instanceof Buffer ? chunk.toString(encoding) : chunk;
callback(undefined, asString.replace(searchValue, replaceValue));
},
});
}
@@ -270,10 +276,12 @@ export function parse(): NodeJS.ReadWriteStream {
return new Transform({
readableObjectMode: true,
writableObjectMode: true,
async transform(chunk: string, encoding, callback) {
async transform(chunk: string | Buffer, encoding, callback) {
try {
const asString =
chunk instanceof Buffer ? chunk.toString(encoding) : chunk;
// Using await causes parsing errors to be emitted
callback(undefined, await JSON.parse(chunk));
callback(undefined, await JSON.parse(asString));
} catch (err) {
callback(err);
}