Modified Client.close to handle streams which have already ended.

moscajs · Dec 10, 2013 · d20bdc9 · d20bdc9
1 parent 2dbfab8
commit d20bdc9
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lib/client.js b/lib/client.js
@@ -510,13 +510,16 @@ Client.prototype.close = function(callback) {
     }
   };
 
+  that.connection.stream.on('end' , function(){
+    that._streamClosedRequiresCleanup = true;
+  });
+
   that._closing = true;
+
   async.parallel(Object.keys(that.subscriptions).map(that.unsubscribeMapTo.bind(that)), function() {
     that.server.persistClient(that);
-    if(that.connection.stream.destroyed){
-      if(callback) {
-        callback();
-      }
+    if(that._streamClosedRequiresCleanup){
+      cleanup();
     } else {
       that.connection.stream.on('end', cleanup);
       that.connection.stream.end();