More fixes to the persistence sequence.

moscajs · Sep 8, 2013 · f726ad1 · f726ad1
1 parent d5178e9
commit f726ad1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/client.js b/lib/client.js
@@ -215,7 +215,7 @@ Client.prototype.unsubscribeMapTo = function(topic) {
  */
 Client.prototype.unsubAndClose = function(cb) {
   var that = this;
-  this.server.persistClient(this);
+  this._closing = true;
   async.parallel(Object.keys(that.subscriptions).map(that.unsubscribeMapTo.bind(that)), function() {
     that.close(cb);
   });
@@ -491,6 +491,7 @@ Client.prototype.close = function(callback) {
   if (this._closed) {
     cleanup();
   } else {
+    this.server.persistClient(this);
     this.connection.stream.on("end", cleanup);
     this.connection.stream.end();
   }