Add callback for close API

webpack · Sep 24, 2016 · 1cf6549 · 1cf6549
1 parent 943e024
commit 1cf6549
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/Server.js b/lib/Server.js
@@ -380,13 +380,14 @@ Server.prototype.listen = function() {
 	return returnValue;
 }
 
-Server.prototype.close = function() {
+Server.prototype.close = function(callback) {
 	this.sockets.forEach(function(sock) {
 		sock.close();
 	});
 	this.sockets = [];
-	this.middleware.close();
-	this.listeningApp.close();
+	this.listeningApp.close(function() {
+		this.middleware.close(callback);
+	}.bind(this));
 }
 
 Server.prototype.sockWrite = function(sockets, type, data) {