Add conn max lifetime

ShiftLeftSecurity · Aug 4, 2018 · 0e993da · 0e993da
1 parent fe32670
commit 0e993da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/db/connection/connection.go b/db/connection/connection.go
@@ -33,6 +33,7 @@ type Information struct {
 	User             string
 	Password         string
 	QueryExecTimeout *time.Duration //optionally set the maximum time a query can take to execute
+	ConnMaxLifetime  *time.Duration
 
 	TLSConfig         *tls.Config // config for TLS connection -- nil disables TLS
 	UseFallbackTLS    bool        // Try FallbackTLSConfig if connecting with TLSConfig fails. Used for preferring TLS, but allowing unencrypted, or vice-versa

diff --git a/db/postgrespq/connection.go b/db/postgrespq/connection.go
@@ -109,6 +109,9 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 	if err != nil {
 		return nil, errors.Wrap(err, "connecting to postgres database")
 	}
+	if ci.ConnMaxLifetime != nil {
+		conn.SetConnMaxLifetime(*ci.ConnMaxLifetime)
+	}
 	return &DB{
 		conn:        conn,
 		logger:      conLogger,