Abstract out net.Server.prototype._rejectPending

Does the same timeout action for maxConnections as it does for EMFILE.
This commit is contained in:
Ryan Dahl 2010-10-28 11:33:33 -07:00
Родитель 74d0a077ec
Коммит ad61d77fa3
2 изменённых файлов: 36 добавлений и 27 удалений

Просмотреть файл

@ -11,6 +11,8 @@ server = net.Server(function (socket) {
});
server.maxConnections = 128;
server.listen(9000);
var oldConnections, oldErrors;

Просмотреть файл

@ -959,7 +959,7 @@ function Server (/* [ options, ] listener */) {
self.watcher.callback = function () {
// Just in case we don't have a dummy fd.
getDummyFD();
if (self._acceptTimer) {
// Somehow the watcher got started again. Need to wait until
// the timer finishes.
@ -975,36 +975,17 @@ function Server (/* [ options, ] listener */) {
// Gracefully reject pending clients by freeing up a file
// descriptor.
rescueEMFILE(function() {
var acceptCount = 0;
// Accept and close the waiting clients one at a time.
// Single threaded programming ftw.
while (true) {
peerInfo = accept(self.fd);
if (!peerInfo) break;
close(peerInfo.fd);
// Don't become DoS'd by incoming requests
if (++acceptCount > 50) {
assert(!self._acceptTimer);
self.watcher.stop();
// Wait a second before accepting more.
self._acceptTimer = setTimeout(function () {
assert(parseInt(self.fd) >= 0);
self._acceptTimer = null;
self.watcher.start();
}, 1000);
break;
}
}
self._rejectPending();
});
return;
}
if (!peerInfo) return;
if (self.maxConnections && self.connections >= self.maxConnections) {
// Accept and close the connection.
// Close the connection we just had
close(peerInfo.fd);
// Reject all other pending connectins.
self._rejectPending();
return;
}
@ -1041,6 +1022,32 @@ exports.createServer = function () {
};
Server.prototype._rejectPending = function () {
var self = this;
var acceptCount = 0;
// Accept and close the waiting clients one at a time.
// Single threaded programming ftw.
while (true) {
peerInfo = accept(this.fd);
if (!peerInfo) return;
close(peerInfo.fd);
// Don't become DoS'd by incoming requests
if (++acceptCount > 50) {
assert(!this._acceptTimer);
this.watcher.stop();
// Wait a second before accepting more.
this._acceptTimer = setTimeout(function () {
assert(parseInt(self.fd) >= 0);
self._acceptTimer = null;
self.watcher.start();
}, 1000);
return;
}
}
};
// Listen on a UNIX socket
// server.listen("/tmp/socket");
//
@ -1122,7 +1129,7 @@ Server.prototype._startWatcher = function () {
Server.prototype._doListen = function () {
var self = this;
// Ensure we have a dummy fd for EMFILE conditions.
getDummyFD();
@ -1187,7 +1194,7 @@ function rescueEMFILE(callback) {
console.error("(node) Hit max file limit. Increase 'ulimit -n'.");
lastEMFILEWarning = now;
}
if (dummyFD) {
close(dummyFD);
dummyFD = null;
@ -1204,4 +1211,4 @@ function getDummyFD() {
dummyFD = null;
}
}
}
}