websocket: do error handling for each ws write

socketio · Aug 1, 2012 · a80c76f · a80c76f
1 parent 25b48dd
commit a80c76f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/transports/websocket.js b/lib/transports/websocket.js
@@ -70,7 +70,13 @@ WebSocket.prototype.send = function (packets) {
   for (var i = 0, l = packets.length; i < l; i++) {
     var data = parser.encodePacket(packets[i]);
     debug('writing', data);
-    this.socket.send(data);
+    this.writable = false;
+    var self = this;
+    this.socket.send(data, function (err){
+      if (err) return self.onError('write error', err.stack);
+      self.writable = true;
+      self.emit('drain');
+    });
   }
 };