Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

release idle connections after timeout #2507

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
2 changes: 2 additions & 0 deletions .gitignore
Expand Up @@ -2,4 +2,6 @@
coverage/
node_modules/
npm-debug.log
.vscode
.idea
package-lock.json
4 changes: 4 additions & 0 deletions Readme.md
Expand Up @@ -459,6 +459,10 @@ constructor. In addition to those options pools accept a few extras:
pool will immediately call back with an error. (Default: `true`)
* `connectionLimit`: The maximum number of connections to create at once.
(Default: `10`)
* `maxIdle`: The maximum number of idle connections.
(Default: `connectionLimit`)
* `idleTimeout`: The maximum idle time of a connection in the pool, in milliseconds.
(Default: `60000`)
* `queueLimit`: The maximum number of connection requests the pool will queue
before returning an error from `getConnection`. If set to `0`, there is no
limit to the number of queued connection requests. (Default: `0`)
Expand Down
23 changes: 22 additions & 1 deletion lib/Pool.js
Expand Up @@ -17,6 +17,10 @@ function Pool(options) {
this._freeConnections = [];
this._connectionQueue = [];
this._closed = false;
if (this.config.maxIdle < this.config.connectionLimit) {
// create idle connection timeout automatically release job
this._removeIdleTimeoutConnections();
}
}

Pool.prototype.getConnection = function (cb) {
Expand All @@ -34,7 +38,7 @@ Pool.prototype.getConnection = function (cb) {
var pool = this;

if (this._freeConnections.length > 0) {
connection = this._freeConnections.shift();
connection = this._freeConnections.pop();
this.acquireConnection(connection, cb);
return;
}
Expand Down Expand Up @@ -277,6 +281,23 @@ Pool.prototype._removeConnection = function(connection) {
this.releaseConnection(connection);
};

Pool.prototype._removeIdleTimeoutConnections = function() {
if (this._removeIdleTimeoutConnectionsTimer) {
clearTimeout(this._removeIdleTimeoutConnectionsTimer);
}

var _this = this;
this._removeIdleTimeoutConnectionsTimer = setTimeout(function () {
try {
while (_this._freeConnections.length > _this.config.maxIdle && Date.now() - _this._freeConnections[0].lastActiveTime > _this.config.idleTimeout) {
_this._purgeConnection(_this._freeConnections[0]);
}
} finally {
_this._removeIdleTimeoutConnections();
}
}, 1000);
};

Pool.prototype.escape = function(value) {
return mysql.escape(value, this.config.connectionConfig.stringifyObjects, this.config.connectionConfig.timezone);
};
Expand Down
6 changes: 6 additions & 0 deletions lib/PoolConfig.js
Expand Up @@ -17,6 +17,12 @@ function PoolConfig(options) {
this.connectionLimit = (options.connectionLimit === undefined)
? 10
: Number(options.connectionLimit);
this.maxIdle = (options.maxIdle === undefined)
? this.connectionLimit
: Number(options.maxIdle);
this.idleTimeout = (options.idleTimeout === undefined)
? 60000
: Number(options.idleTimeout);
this.queueLimit = (options.queueLimit === undefined)
? 0
: Number(options.queueLimit);
Expand Down
6 changes: 6 additions & 0 deletions lib/PoolConnection.js
Expand Up @@ -9,6 +9,9 @@ function PoolConnection(pool, options) {
Connection.call(this, options);
this._pool = pool;

// The last active time of this connection
this.lastActiveTime = Date.now();

// Bind connection to pool domain
if (Events.usingDomains) {
this.domain = pool.domain;
Expand All @@ -32,6 +35,9 @@ PoolConnection.prototype.release = function release() {
return undefined;
}

// update last active time
this.lastActiveTime = Date.now();

return pool.releaseConnection(this);
};

Expand Down
47 changes: 47 additions & 0 deletions test/unit/pool/test-idle-connection-timeout.js
@@ -0,0 +1,47 @@
var assert = require('assert');
var common = require('../../common');
var pool = common.createPool({
port : common.fakeServerPort,
connectionLimit : 10,
maxIdle : 1,
idleTimeout : 5000
});

var server = common.createFakeServer();

server.listen(common.fakeServerPort, function(err){
assert.ifError(err);

pool.once('release', function(connection) {
assert.ok(connection);
});

pool.getConnection(function (err, connection1) {
assert.ifError(err);
assert.ok(connection1);
pool.getConnection(function (err, connection2) {
assert.ifError(err);
assert.ok(connection2);
assert.notEqual(connection1, connection2);
connection1.release();
connection2.release();
assert.equal(pool._allConnections.length, 2);
assert.equal(pool._freeConnections.length, 2);
setTimeout(function() {
assert.equal(pool._allConnections.length, 1);
assert.equal(pool._freeConnections.length, 1);
pool.getConnection(function (err, connection3) {
assert.ifError(err);
assert.ok(connection3);
assert.equal(connection3, connection2);
assert.equal(pool._allConnections.length, 1);
assert.equal(pool._freeConnections.length, 0);
connection3.release();
connection3.destroy();
server.destroy();
setTimeout(function () { process.exit(0); }, 1000);
});
}, 7000);
});
});
});