Add LogLevel to pq driver

ShiftLeftSecurity · Aug 22, 2018 · b1dd52c · b1dd52c
1 parent c7b663a
commit b1dd52c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/db/postgres/connection.go b/db/postgres/connection.go
@@ -81,6 +81,10 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 			return nil, errors.Wrap(err, "parsing connection string")
 		}
 		if ci != nil {
+			llevel, llevelErr := pgx.LogLevelFromString(string(ci.LogLevel))
+			if llevelErr != nil {
+				llevel = pgx.LogLevelError
+			}
 			config.ConnConfig = csconfig.Merge(pgx.ConnConfig{
 				Host:     ci.Host,
 				Port:     ci.Port,
@@ -92,6 +96,7 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 				UseFallbackTLS:    ci.UseFallbackTLS,
 				FallbackTLSConfig: ci.FallbackTLSConfig,
 				Logger:            logging.NewPgxLogAdapter(ci.Logger),
+				LogLevel:          int(llevel),
 			})
 		} else {
 			defaultLogger := log.New(os.Stdout, "logger: ", log.Lshortfile)

diff --git a/db/postgrespq/connection.go b/db/postgrespq/connection.go
@@ -50,6 +50,10 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 	var config pgx.ConnPoolConfig
 	var conLogger logging.Logger
 	if ci != nil {
+		llevel, llevelErr := pgx.LogLevelFromString(string(ci.LogLevel))
+		if llevelErr != nil {
+			llevel = pgx.LogLevelError
+		}
 		conLogger = ci.Logger
 		config = pgx.ConnPoolConfig{
 			ConnConfig: pgx.ConnConfig{
@@ -63,6 +67,7 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 				UseFallbackTLS:    ci.UseFallbackTLS,
 				FallbackTLSConfig: ci.FallbackTLSConfig,
 				Logger:            logging.NewPgxLogAdapter(conLogger),
+				LogLevel:          int(llevel),
 			},
 			MaxConnections: ci.MaxConnPoolConns,
 		}
@@ -76,6 +81,10 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 			return nil, errors.Wrap(err, "parsing connection string")
 		}
 		if ci != nil {
+			llevel, llevelErr := pgx.LogLevelFromString(string(ci.LogLevel))
+			if llevelErr != nil {
+				llevel = pgx.LogLevelError
+			}
 			config.ConnConfig = csconfig.Merge(pgx.ConnConfig{
 				Host:     ci.Host,
 				Port:     ci.Port,
@@ -87,6 +96,7 @@ func (c *Connector) Open(ci *connection.Information) (connection.DB, error) {
 				UseFallbackTLS:    ci.UseFallbackTLS,
 				FallbackTLSConfig: ci.FallbackTLSConfig,
 				Logger:            logging.NewPgxLogAdapter(ci.Logger),
+				LogLevel:          int(llevel),
 			})
 		} else {
 			defaultLogger := log.New(os.Stdout, "logger: ", log.Lshortfile)