updated close function for safety

http-party · Aug 14, 2014 · 8be9d94 · 8be9d94
1 parent a3d0219
commit 8be9d94
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/http-proxy/index.js b/lib/http-proxy/index.js
@@ -134,8 +134,16 @@ ProxyServer.prototype.listen = function(port, hostname) {
 
 ProxyServer.prototype.close = function(callback) {
   if (this._server) {
-    this._server.close(callback);
-    this._server = null;
+    // Wrap callback to nullify server after all open connections are closed.
+    var callback_wrapper = function() {
+        this._server = null;
+
+        if (callback) {
+            callback(arguments);
+        }
+    };
+
+    this._server.close(callback_wrapper);
   }
 };