fixup

Author: ronag

examples/proxy/proxy.js

@@ -68,9 +68,8 @@ class HTTPHandler {
   }
 
   onError (err) {
-    this.res
-      .off('drain', this.resume)
-      .off('close', this.abort)
+    this.res.off('close', this.abort)
+    this.res.off('drain', this.resume)
     this.callback(err)
   }
 }
@@ -83,6 +82,7 @@ class WSHandler {
 
     this.socket = socket
     this.head = head
+    this.abort = null
     this.promise = new Promise((resolve, reject) => {
       this.callback = err => err ? reject(err) : resolve()
     })
@@ -92,6 +92,7 @@ class WSHandler {
     if (this.socket.destroyed) {
       abort()
     } else {
+      this.abort = abort
       this.socket.on('close', abort)
     }
   }
@@ -127,6 +128,7 @@ class WSHandler {
   }
 
   onError (err) {
+    this.socket.off('close', this.abort)
     this.callback(err)
   }
 }