Merge pull request #1081 from benbotto/keep-alive

Keep alive
sidorares · Dec 13, 2019 · f9b9f96 · f9b9f96
2 parents d8cac2a + 288d997
commit f9b9f96
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/connection.js b/lib/connection.js
@@ -43,6 +43,12 @@ class Connection extends EventEmitter {
     } else {
       this.stream = opts.config.stream;
     }
+
+    // Enable keep-alive on the socket.  It's disabled by default, but the
+    // user can enable it and supply an initial delay.
+    if (typeof this.stream.setKeepAlive === 'function' && this.config.enableKeepAlive)
+      this.stream.setKeepAlive(true, this.config.keepAliveInitialDelay);
+
     this._internalId = _connectionId++;
     this._commands = new Queue();
     this._command = null;

diff --git a/lib/connection_config.js b/lib/connection_config.js
@@ -18,10 +18,12 @@ const validOptions = {
   dateStrings: 1,
   debug: 1,
   decimalNumbers: 1,
+  enableKeepAlive: 1,
   flags: 1,
   host: 1,
   insecureAuth: 1,
   isServer: 1,
+  keepAliveInitialDelay: 1,
   localAddress: 1,
   maxPreparedStatements: 1,
   multipleStatements: 1,
@@ -93,6 +95,8 @@ class ConnectionConfig {
     this.debug = options.debug;
     this.trace = options.trace !== false;
     this.stringifyObjects = options.stringifyObjects || false;
+    this.enableKeepAlive = !!options.enableKeepAlive;
+    this.keepAliveInitialDelay = options.keepAliveInitialDelay || 0;
     if (
       options.timezone &&
       !/^(?:local|Z|[ +-]\d\d:\d\d)$/.test(options.timezone)