Merge pull request #303 from ldinc/issue#282

Issue#282

pkg/cli/cli.go

@@ -197,7 +197,7 @@ func Run() {
 	}
 
 	// Start session cleanup worker
-	if options.Sessions {
+	if options.Sessions && !command.Opts.ConnectionIdleTimeoutDisabled {
 		go api.StartSessionCleanup()
 	}
 

pkg/client/client.go

@@ -367,7 +367,7 @@ func (client *Client) Close() error {
 }
 
 func (client *Client) IsIdle() bool {
-	return time.Since(client.lastQueryTime).Hours() > 1
+	return time.Since(client.lastQueryTime).Minutes() > command.Opts.ConnectionIdleTimeout
 }
 
 // Fetch all rows as strings for a single column

pkg/command/options.go

@@ -34,6 +34,8 @@ type Options struct {
 	ConnectBackend                string  `long:"connect-backend" description:"Enable database authentication through a third party backend"`
 	ConnectToken                  string  `long:"connect-token" description:"Authentication token for the third-party connect backend"`
 	ConnectHeaders                string  `long:"connect-headers" description:"List of headers to pass to the connect backend"`
+	ConnectionIdleTimeoutDisabled bool    `long:"disable-connection-idle-timeout" description:"Disable connection idle timeout" default:"false"`
+	ConnectionIdleTimeout         float64 `long:"idle-timeout" description:"Set connection idle timeout in minutes" default:"180"`
 }
 
 var Opts Options