Merge pull request #7 from hermes-serverless/feature/drain-stream

Add drainStream feature

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hermes-serverless/stream-utils",
-  "version": "0.3.2",
+  "version": "0.4.0",
   "description": "",
   "main": "build/index.js",
   "types": "build/index.d.ts",

src/__tests__/drainStream.test.ts

@@ -0,0 +1,9 @@
+import { PassThrough } from 'stream'
+import { drainStream } from '../'
+
+test('drainStream', done => {
+  const stream = new PassThrough()
+  stream.end('.'.repeat(1000000))
+  stream.on('end', done)
+  drainStream(stream)
+})

src/drainStream.ts

@@ -0,0 +1,5 @@
+import { Readable } from 'stream'
+
+export const drainStream = (stream: Readable) => {
+  stream.on('readable', stream.read.bind(stream))
+}

src/index.ts

@@ -1,3 +1,4 @@
+export { drainStream } from './drainStream'
 export { flowUntilLimit, normalizeToWritableWithEnd, WritableWithEnd } from './flowUntilLimit'
 export { streamFinished } from './streamFinished'
 export { StringStream } from './StringStream'