discard transport on server close

socketio · Jan 21, 2016 · 941a6eb · 941a6eb
1 parent 7a64e67
commit 941a6eb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/lib/server.js b/lib/server.js
@@ -179,7 +179,7 @@ Server.prototype.close = function(){
   debug('closing all open clients');
   for (var i in this.clients) {
     if (this.clients.hasOwnProperty(i)) {
-      this.clients[i].close();
+      this.clients[i].close(true);
     }
   }
   return this;

diff --git a/lib/socket.js b/lib/socket.js
@@ -440,29 +440,32 @@ Socket.prototype.getAvailableUpgrades = function () {
 /**
  * Closes the socket and underlying transport.
  *
+ * @param {Boolean} optional, discard
  * @return {Socket} for chaining
  * @api public
  */
 
-Socket.prototype.close = function () {
+Socket.prototype.close = function (discard) {
   if ('open' != this.readyState) return;
 
   this.readyState = 'closing';
 
   if (this.writeBuffer.length) {
-    this.once('drain', this.closeTransport.bind(this));
+    this.once('drain', this.closeTransport.bind(this, discard));
     return;
   }
 
-  this.closeTransport();
+  this.closeTransport(discard);
 };
 
 /**
  * Closes the underlying transport.
  *
+ * @param {Boolean} discard
  * @api private
  */
 
-Socket.prototype.closeTransport = function () {
+Socket.prototype.closeTransport = function (discard) {
+  if (discard) this.transport.discard();
   this.transport.close(this.onClose.bind(this, 'forced close'));
 };