make connection pool params configurable

2020-05-11 10:50:05 -05:00
parent fb6fdfcc13
commit 8dc31e4ca5
2 changed files with 35 additions and 10 deletions
@@ -17,6 +17,8 @@
 package postgres

 import (
+	"time"
+
 	"github.com/jmoiron/sqlx"
 	_ "github.com/lib/pq" //postgres driver
 	"github.com/vulcanize/vulcanizedb/pkg/config"
@@ -35,6 +37,16 @@ func NewDB(databaseConfig config.Database, node core.Node) (*DB, error) {
 	if connectErr != nil {
 		return &DB{}, ErrDBConnectionFailed(connectErr)
 	}
+	if databaseConfig.MaxOpen > 0 {
+		db.SetMaxOpenConns(databaseConfig.MaxOpen)
+	}
+	if databaseConfig.MaxIdle > 0 {
+		db.SetMaxIdleConns(databaseConfig.MaxIdle)
+	}
+	if databaseConfig.MaxLifetime > 0 {
+		lifetime := time.Duration(databaseConfig.MaxLifetime) * time.Second
+		db.SetConnMaxLifetime(lifetime)
+	}
 	pg := DB{DB: db, Node: node}
 	nodeErr := pg.CreateNode(&node)
 	if nodeErr != nil {