app/scripts/lib/obj-multiplex.js


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

const through = require('through2')

module.exports = ObjectMultiplex

function ObjectMultiplex (opts) {
  opts = opts || {}
  // create multiplexer
  var mx = through.obj(function (chunk, enc, cb) {
    var name = chunk.name
    var data = chunk.data
    var substream = mx.streams[name]
    if (!substream) {
      console.warn('orphaned data for stream ' + name)
    } else {
      substream.push(data)
    }
    return cb()
  })
  mx.streams = {}
  // create substreams
  mx.createStream = function (name) {
    var substream = mx.streams[name] = through.obj(function (chunk, enc, cb) {
      mx.push({
        name: name,
        data: chunk,
      })
      return cb()
    })
    mx.on('end', function () {
      return substream.emit('end')
    })
    if (opts.error) {
      mx.on('error', function () {
        return substream.emit('error')
      })
    }
    return substream
  }
  return mx
}