listen socket end events

Author: sidorares

lib/Connection.js

@@ -64,13 +64,6 @@ Connection.prototype.connect = function(cb) {
       ? Net.createConnection(this.config.socketPath)
       : Net.createConnection(this.config);
 
-    // Node v0.10+ Switch socket into "old mode" (Streams2)
-    //this._socket.on("data",function() {});
-
-    //this._socket.pipe(this._protocol);
-    //this._protocol.pipe(this._socket);
-
-    
     var connection = this;
     this._protocol.on('data', function(data) {
        connection._socket.write(data);
@@ -81,6 +74,9 @@ Connection.prototype.connect = function(cb) {
     this._protocol.on('end', function() {
        connection._socket.end()
     });
+    this._socket.on('end', function(err) {
+      connection._protocol.end();
+    });
 
     this._socket.on('error', this._handleNetworkError.bind(this));
     this._socket.on('connect', this._handleProtocolConnect.bind(this));