Ws drain fix

lib/transport.js

@@ -90,9 +90,9 @@ Transport.prototype.close = function () {
  * @api private
  */
 
-Transport.prototype.send = function(packets){
+Transport.prototype.send = function(packets, fn){
   if ('open' == this.readyState) {
-    this.write(packets);
+    this.write(packets, fn);
   } else {
     throw new Error('Transport not open');
   }
@@ -137,5 +137,6 @@ Transport.prototype.onPacket = function (packet) {
 
 Transport.prototype.onClose = function () {
   this.readyState = 'closed';
+  this.intervalCleanup && this.intervalCleanup.call(this);
   this.emit('close');
 };

lib/transports/websocket.js

@@ -29,6 +29,7 @@ var global = util.global();
 
 function WS(opts){
   Transport.call(this, opts);
+  this.bufferedAmountId;
 };
 
 /**
@@ -82,11 +83,32 @@ WS.prototype.doOpen = function(){
  */
 
 WS.prototype.write = function(packets){
-  for (var i = 0, l = packets.length; i < l; i++) {
-    this.socket.send(parser.encodePacket(packets[i]));
+  var self = this;
+  // encodePayload instead of encodePacket
+  this.writable = false;
+  this.socket.send(parser.encodePayload(packets));
+  // check periodically if we're done sending
+  if ('bufferedAmount' in this.socket) {
+    this.bufferedAmountId = this.setInterval(function() {
+      if (this.socket.bufferedAmount == 0) {
+        clearInterval(self.bufferedAmountId);
+        this.writable = true;
+        this.emit('drain');
+      }
+    }, 50);
+  } else {
+    // fake drain
+    this.writable = true;
+    this.emit('drain');
   }
 };
 
+WS.prototype.intervalCleanup = function(){
+  if (this.bufferedAmountId) {
+    clearInterval(this.bufferedAmountId);
+  }
+}
+
 /**
  * Closes socket.
  *