From 4b75696bb49cb57a7a3088c941a381c3127810ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vin=C3=ADcius=20Garcia?= <vingarcia00@gmail.com>
Date: Sat, 30 Oct 2021 12:08:03 -0300
Subject: [PATCH] Add SetDefaultValues() func to the ksql.Config struct

---
 kpgx/kpgx.go |  2 ++
 ksql.go      | 14 ++++++++++----
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/kpgx/kpgx.go b/kpgx/kpgx.go
index dd3f957..87ad683 100644
--- a/kpgx/kpgx.go
+++ b/kpgx/kpgx.go
@@ -13,6 +13,8 @@ func New(
 	connectionString string,
 	config ksql.Config,
 ) (db ksql.DB, err error) {
+	config.SetDefaultValues()
+
 	pgxConf, err := pgxpool.ParseConfig(connectionString)
 	if err != nil {
 		return ksql.DB{}, err
diff --git a/ksql.go b/ksql.go
index 723fd9d..7ce7bdc 100644
--- a/ksql.go
+++ b/ksql.go
@@ -76,12 +76,20 @@ type Config struct {
 	MaxOpenConns int
 }
 
+func (c *Config) SetDefaultValues() {
+	if c.MaxOpenConns == 0 {
+		c.MaxOpenConns = 1
+	}
+}
+
 // New instantiates a new KissSQL client
 func New(
 	dbDriver string,
 	connectionString string,
 	config Config,
 ) (DB, error) {
+	config.SetDefaultValues()
+
 	db, err := sql.Open(dbDriver, connectionString)
 	if err != nil {
 		return DB{}, err
@@ -90,10 +98,6 @@ func New(
 		return DB{}, err
 	}
 
-	if config.MaxOpenConns == 0 {
-		config.MaxOpenConns = 1
-	}
-
 	db.SetMaxOpenConns(config.MaxOpenConns)
 
 	return NewWithAdapter(SQLAdapter{db}, dbDriver)
@@ -108,6 +112,8 @@ func NewWithPGX(
 	connectionString string,
 	config Config,
 ) (db DB, err error) {
+	config.SetDefaultValues()
+
 	pgxConf, err := pgxpool.ParseConfig(connectionString)
 	if err != nil {
 		return DB{}, err