Only enables keep-alive on lib-created TCP connections.

Author: benbotto

lib/connection.js

@@ -36,6 +36,10 @@ class Connection extends EventEmitter {
           opts.config.port,
           opts.config.host
         );
+
+        // Enable keep-alive on the socket.  It's disabled by default, but the
+        // user can enable it and supply an initial delay.
+        this.stream.setKeepAlive(true, this.config.keepAliveInitialDelay);
       }
       // if stream is a function, treat it as "stream agent / factory"
     } else if (typeof opts.config.stream === 'function')  {
@@ -44,11 +48,6 @@ class Connection extends EventEmitter {
       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;