Adding a little buffer to cope with slow connections.

mcollina · mcollina · commit cf506baa3103 · 2013-07-20T23:03:03.000+01:00
diff --git a/lib/connection.js b/lib/connection.js
@@ -2,12 +2,13 @@ var events = require('events')
   , util = require('util')
   , protocol = require('./protocol')
   , generate = require('./generate')
-  , parse = require('./parse');
-
-var Writable = require("stream").Writable
+  , parse = require('./parse')
+  , Writable = require("stream").Writable
+  , PassThrough = require("stream").PassThrough;
 
 if (!Writable) {
   Writable = require("readable-stream").Writable;
+  PassThrough = require("readable-stream").PassThrough;
 }
 
 var Connection = module.exports = 
@@ -23,17 +24,20 @@ function Connection(stream, server) {
 };
 util.inherits(Connection, events.EventEmitter);
 
-Connection.prototype._setupParser = function() {
+Connection.prototype._setup = function() {
   this.parser = this.stream.pipe(new PacketParser({
     connection: this
   }));
+
+  this.bufferer = new PassThrough();
+  this.bufferer.pipe(this.stream);
 };
 
 Connection.prototype.reconnect = function(stream) {
   var that = this;
 
   this.stream = stream;
-  this._setupParser();
+  this._setup();
 };
 
 for (var k in protocol.types) {
@@ -43,8 +47,8 @@ for (var k in protocol.types) {
   "   var p = this.generate." + v +"(arguments[0]); " +
   "   if (p instanceof Error) { " +
   "     this.emit('error', p) " +
-  "   } else { " +
-  "     this.stream.write(p); " +
+  "   } else if(!this.bufferer.write(p)) { " +
+  "     this.emit('error', 'Unable to write on the stream'); " +
   "   } "
   " } ";
 
