socket: keep track of socket in connected hash

dkgaoqi · Dec 24, 2012 · 95d8e75 · 95d8e75
1 parent 2e71357
commit 95d8e75
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/socket.js b/lib/socket.js
@@ -233,6 +233,7 @@ Socket.prototype.onconnect = function(){
   debug('socket connected - writing packet');
   this.join(this.id);
   this.packet({ type: parser.CONNECT });
+  this.nsp.connected[this.id] = this;
 };
 
 /**
@@ -342,10 +343,12 @@ Socket.prototype.ondisconnect = function(){
 Socket.prototype.onclose = function(reason){
   if (!this.connected) return this;
   debug('closing socket - reason %s', reason);
+  this.leaveAll();
   this.nsp.remove(this);
   this.client.remove(this);
   this.connected = false;
   this.disconnected = true;
+  delete this.nsp.connected[this.id];
   this.emit('disconnect', reason);
 };