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.
		
		
		
		
		
			
		
			
				
					
					
						
							107 lines
						
					
					
						
							2.3 KiB
						
					
					
				
			
		
		
		
			
			
			
		
		
	
	
							107 lines
						
					
					
						
							2.3 KiB
						
					
					
				| var Stream = require('stream').Stream; | |
| var util = require('util'); | |
| 
 | |
| module.exports = DelayedStream; | |
| function DelayedStream() { | |
|   this.source = null; | |
|   this.dataSize = 0; | |
|   this.maxDataSize = 1024 * 1024; | |
|   this.pauseStream = true; | |
| 
 | |
|   this._maxDataSizeExceeded = false; | |
|   this._released = false; | |
|   this._bufferedEvents = []; | |
| } | |
| util.inherits(DelayedStream, Stream); | |
| 
 | |
| DelayedStream.create = function(source, options) { | |
|   var delayedStream = new this(); | |
| 
 | |
|   options = options || {}; | |
|   for (var option in options) { | |
|     delayedStream[option] = options[option]; | |
|   } | |
| 
 | |
|   delayedStream.source = source; | |
| 
 | |
|   var realEmit = source.emit; | |
|   source.emit = function() { | |
|     delayedStream._handleEmit(arguments); | |
|     return realEmit.apply(source, arguments); | |
|   }; | |
| 
 | |
|   source.on('error', function() {}); | |
|   if (delayedStream.pauseStream) { | |
|     source.pause(); | |
|   } | |
| 
 | |
|   return delayedStream; | |
| }; | |
| 
 | |
| Object.defineProperty(DelayedStream.prototype, 'readable', { | |
|   configurable: true, | |
|   enumerable: true, | |
|   get: function() { | |
|     return this.source.readable; | |
|   } | |
| }); | |
| 
 | |
| DelayedStream.prototype.setEncoding = function() { | |
|   return this.source.setEncoding.apply(this.source, arguments); | |
| }; | |
| 
 | |
| DelayedStream.prototype.resume = function() { | |
|   if (!this._released) { | |
|     this.release(); | |
|   } | |
| 
 | |
|   this.source.resume(); | |
| }; | |
| 
 | |
| DelayedStream.prototype.pause = function() { | |
|   this.source.pause(); | |
| }; | |
| 
 | |
| DelayedStream.prototype.release = function() { | |
|   this._released = true; | |
| 
 | |
|   this._bufferedEvents.forEach(function(args) { | |
|     this.emit.apply(this, args); | |
|   }.bind(this)); | |
|   this._bufferedEvents = []; | |
| }; | |
| 
 | |
| DelayedStream.prototype.pipe = function() { | |
|   var r = Stream.prototype.pipe.apply(this, arguments); | |
|   this.resume(); | |
|   return r; | |
| }; | |
| 
 | |
| DelayedStream.prototype._handleEmit = function(args) { | |
|   if (this._released) { | |
|     this.emit.apply(this, args); | |
|     return; | |
|   } | |
| 
 | |
|   if (args[0] === 'data') { | |
|     this.dataSize += args[1].length; | |
|     this._checkIfMaxDataSizeExceeded(); | |
|   } | |
| 
 | |
|   this._bufferedEvents.push(args); | |
| }; | |
| 
 | |
| DelayedStream.prototype._checkIfMaxDataSizeExceeded = function() { | |
|   if (this._maxDataSizeExceeded) { | |
|     return; | |
|   } | |
| 
 | |
|   if (this.dataSize <= this.maxDataSize) { | |
|     return; | |
|   } | |
| 
 | |
|   this._maxDataSizeExceeded = true; | |
|   var message = | |
|     'DelayedStream#maxDataSize of ' + this.maxDataSize + ' bytes exceeded.' | |
|   this.emit('error', new Error(message)); | |
| };
 | |
| 
 |