archive-gh-me/node-postgres
1
0
Fork 0
mirror of https://github.com/brianc/node-postgres.git synced 2026-01-18 15:55:05 +00:00
Code Issues Projects Releases Wiki Activity

Make tcp-keepalive configurable (#1058)

Browse Source
This commit is contained in:
Brian C 2016-06-21 21:38:10 -05:00 committed by GitHub
parent ad65c7b1bc
commit 0aa62f2854
3 changed files with 25 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/client.js
View File

@ -25,7 +25,8 @@ var Client = function(config) {
this.connection = c.connection || new Connection({
stream: c.stream,
ssl: this.connectionParameters.ssl
ssl: this.connectionParameters.ssl,
keepAlive: c.keepAlive || false
});
this.queryQueue = [];
this.binary = c.binary || defaults.binary;

5
lib/connection.js
View File

@ -11,6 +11,7 @@ var Connection = function(config) {
EventEmitter.call(this);
config = config || {};
this.stream = config.stream || new net.Stream();
this._keepAlive = config.keepAlive;
this.lastBuffer = false;
this.lastOffset = 0;
this.buffer = null;
@ -47,8 +48,10 @@ Connection.prototype.connect = function(port, host) {
var self = this;
this.stream.on('connect', function() {
if (self._keepAlive) {
self.stream.setKeepAlive(true);
}
self.emit('connect');
self.stream.setKeepAlive(true);
});
this.stream.on('error', function(error) {

27
test/unit/connection/startup-tests.js
View File

@ -7,13 +7,18 @@ test('connection can take existing stream', function() {
});
test('using closed stream', function() {
var stream = new MemoryStream();
stream.readyState = 'closed';
stream.connect = function(port, host) {
this.connectCalled = true;
this.port = port;
this.host = host;
}
var makeStream = function() {
var stream = new MemoryStream();
stream.readyState = 'closed';
stream.connect = function(port, host) {
this.connectCalled = true;
this.port = port;
this.host = host;
}
return stream;
};
var stream = makeStream();
var con = new Connection({stream: stream});
@ -46,6 +51,10 @@ test('using closed stream', function() {
test('after stream emits connected event init TCP-keepalive', function() {
var stream = makeStream();
var con = new Connection({ stream: stream, keepAlive: true });
con.connect(123, 'test');
var res = false;
stream.setKeepAlive = function(bit) {
@ -53,7 +62,9 @@ test('using closed stream', function() {
};
assert.ok(stream.emit('connect'));
assert.equal(res, true);
setTimeout(function() {
assert.equal(res, true);
})
});
});