You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							209 lines
						
					
					
						
							4.6 KiB
						
					
					
				
			
		
		
	
	
							209 lines
						
					
					
						
							4.6 KiB
						
					
					
				| var util = require('util');
 | |
| var Stream = require('stream').Stream;
 | |
| var DelayedStream = require('delayed-stream');
 | |
| 
 | |
| module.exports = CombinedStream;
 | |
| function CombinedStream() {
 | |
|   this.writable = false;
 | |
|   this.readable = true;
 | |
|   this.dataSize = 0;
 | |
|   this.maxDataSize = 2 * 1024 * 1024;
 | |
|   this.pauseStreams = true;
 | |
| 
 | |
|   this._released = false;
 | |
|   this._streams = [];
 | |
|   this._currentStream = null;
 | |
|   this._insideLoop = false;
 | |
|   this._pendingNext = false;
 | |
| }
 | |
| util.inherits(CombinedStream, Stream);
 | |
| 
 | |
| CombinedStream.create = function(options) {
 | |
|   var combinedStream = new this();
 | |
| 
 | |
|   options = options || {};
 | |
|   for (var option in options) {
 | |
|     combinedStream[option] = options[option];
 | |
|   }
 | |
| 
 | |
|   return combinedStream;
 | |
| };
 | |
| 
 | |
| CombinedStream.isStreamLike = function(stream) {
 | |
|   return (typeof stream !== 'function')
 | |
|     && (typeof stream !== 'string')
 | |
|     && (typeof stream !== 'boolean')
 | |
|     && (typeof stream !== 'number')
 | |
|     && (!Buffer.isBuffer(stream));
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.append = function(stream) {
 | |
|   var isStreamLike = CombinedStream.isStreamLike(stream);
 | |
| 
 | |
|   if (isStreamLike) {
 | |
|     if (!(stream instanceof DelayedStream)) {
 | |
|       var newStream = DelayedStream.create(stream, {
 | |
|         maxDataSize: Infinity,
 | |
|         pauseStream: this.pauseStreams,
 | |
|       });
 | |
|       stream.on('data', this._checkDataSize.bind(this));
 | |
|       stream = newStream;
 | |
|     }
 | |
| 
 | |
|     this._handleErrors(stream);
 | |
| 
 | |
|     if (this.pauseStreams) {
 | |
|       stream.pause();
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   this._streams.push(stream);
 | |
|   return this;
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.pipe = function(dest, options) {
 | |
|   Stream.prototype.pipe.call(this, dest, options);
 | |
|   this.resume();
 | |
|   return dest;
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._getNext = function() {
 | |
|   this._currentStream = null;
 | |
| 
 | |
|   if (this._insideLoop) {
 | |
|     this._pendingNext = true;
 | |
|     return; // defer call
 | |
|   }
 | |
| 
 | |
|   this._insideLoop = true;
 | |
|   try {
 | |
|     do {
 | |
|       this._pendingNext = false;
 | |
|       this._realGetNext();
 | |
|     } while (this._pendingNext);
 | |
|   } finally {
 | |
|     this._insideLoop = false;
 | |
|   }
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._realGetNext = function() {
 | |
|   var stream = this._streams.shift();
 | |
| 
 | |
| 
 | |
|   if (typeof stream == 'undefined') {
 | |
|     this.end();
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   if (typeof stream !== 'function') {
 | |
|     this._pipeNext(stream);
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   var getStream = stream;
 | |
|   getStream(function(stream) {
 | |
|     var isStreamLike = CombinedStream.isStreamLike(stream);
 | |
|     if (isStreamLike) {
 | |
|       stream.on('data', this._checkDataSize.bind(this));
 | |
|       this._handleErrors(stream);
 | |
|     }
 | |
| 
 | |
|     this._pipeNext(stream);
 | |
|   }.bind(this));
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._pipeNext = function(stream) {
 | |
|   this._currentStream = stream;
 | |
| 
 | |
|   var isStreamLike = CombinedStream.isStreamLike(stream);
 | |
|   if (isStreamLike) {
 | |
|     stream.on('end', this._getNext.bind(this));
 | |
|     stream.pipe(this, {end: false});
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   var value = stream;
 | |
|   this.write(value);
 | |
|   this._getNext();
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._handleErrors = function(stream) {
 | |
|   var self = this;
 | |
|   stream.on('error', function(err) {
 | |
|     self._emitError(err);
 | |
|   });
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.write = function(data) {
 | |
|   this.emit('data', data);
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.pause = function() {
 | |
|   if (!this.pauseStreams) {
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   if(this.pauseStreams && this._currentStream && typeof(this._currentStream.pause) == 'function') this._currentStream.pause();
 | |
|   this.emit('pause');
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.resume = function() {
 | |
|   if (!this._released) {
 | |
|     this._released = true;
 | |
|     this.writable = true;
 | |
|     this._getNext();
 | |
|   }
 | |
| 
 | |
|   if(this.pauseStreams && this._currentStream && typeof(this._currentStream.resume) == 'function') this._currentStream.resume();
 | |
|   this.emit('resume');
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.end = function() {
 | |
|   this._reset();
 | |
|   this.emit('end');
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype.destroy = function() {
 | |
|   this._reset();
 | |
|   this.emit('close');
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._reset = function() {
 | |
|   this.writable = false;
 | |
|   this._streams = [];
 | |
|   this._currentStream = null;
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._checkDataSize = function() {
 | |
|   this._updateDataSize();
 | |
|   if (this.dataSize <= this.maxDataSize) {
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   var message =
 | |
|     'DelayedStream#maxDataSize of ' + this.maxDataSize + ' bytes exceeded.';
 | |
|   this._emitError(new Error(message));
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._updateDataSize = function() {
 | |
|   this.dataSize = 0;
 | |
| 
 | |
|   var self = this;
 | |
|   this._streams.forEach(function(stream) {
 | |
|     if (!stream.dataSize) {
 | |
|       return;
 | |
|     }
 | |
| 
 | |
|     self.dataSize += stream.dataSize;
 | |
|   });
 | |
| 
 | |
|   if (this._currentStream && this._currentStream.dataSize) {
 | |
|     this.dataSize += this._currentStream.dataSize;
 | |
|   }
 | |
| };
 | |
| 
 | |
| CombinedStream.prototype._emitError = function(err) {
 | |
|   this._reset();
 | |
|   this.emit('error', err);
 | |
| };
 |