File size: 1,107 Bytes
bc20498 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
const { Readable } = require('streamx')
module.exports = function (s, forks = 2) {
const streams = new Array(forks)
const status = new Array(forks).fill(true)
let ended = false
for (let i = 0; i < forks; i++) {
streams[i] = new Readable({
read (cb) {
const check = !status[i]
status[i] = true
if (check && allReadable()) s.resume()
cb(null)
}
})
}
s.on('end', function () {
ended = true
for (const stream of streams) stream.push(null)
})
s.on('error', function (err) {
for (const stream of streams) stream.destroy(err)
})
s.on('close', function () {
if (ended) return
for (const stream of streams) stream.destroy()
})
s.on('data', function (data) {
let needsPause = false
for (let i = 0; i < streams.length; i++) {
if (!(status[i] = streams[i].push(data))) {
needsPause = true
}
}
if (needsPause) s.pause()
})
return streams
function allReadable () {
for (let j = 0; j < status.length; j++) {
if (!status[j]) return false
}
return true
}
}
|