From 171892c66920c0a1227ea88d4623bd650b3859d8 Mon Sep 17 00:00:00 2001
From: TJ Holowaychuk <tj@vision-media.ca>
Date: Thu, 19 Dec 2013 08:53:48 -0800
Subject: [PATCH] fix: cleanup socker error handler on response

---
 lib/application.js |  2 +-
 lib/context.js     | 16 ++++++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/lib/application.js b/lib/application.js
index 3ba9193..8a8b35d 100644
--- a/lib/application.js
+++ b/lib/application.js
@@ -98,7 +98,7 @@ app.callback = function(){
   return function(req, res, next){
     var ctx = self.createContext(req, res);
     next = next || ctx.onerror;
-    ctx.socket.once('error', next);
+    ctx.onSocketError(next);
     co.call(ctx, gen)(next);
   }
 };
diff --git a/lib/context.js b/lib/context.js
index 050ea1a..b379754 100644
--- a/lib/context.js
+++ b/lib/context.js
@@ -488,4 +488,20 @@ module.exports = {
   attachment: function() {
     return this.response.attachment.apply(this.response, arguments);
   },
+
+  /**
+   * Invoke `fn` on socket errors and
+   * handle cleanup on response.
+   *
+   * @param {Function} fn
+   * @api private
+   */
+
+  onSocketError: function(fn){
+    var sock = this.socket;
+    sock.once('error', fn);
+    this.res.on('finish', function(){
+      sock.removeListener('error', fn);
+    });
+  }
 };