Clean up closed connections. Closes #282

Author: curran

lib/agent.js

@@ -258,12 +258,15 @@ Agent.prototype._open = function() {
       agent._handleMessage(request.data, callback);
     });
   });
-
-  this.stream.on('end', function() {
+  
+  function cleanup() {
     agent.backend.agentsCount--;
     if (!agent.stream.isServer) agent.backend.remoteAgentsCount--;
     agent._cleanup();
-  });
+  }
+  
+  this.stream.on('end', cleanup);
+  this.stream.on('close', cleanup);
 };
 
 // Check a request to see if its valid. Returns an error if there's a problem.