Add SetDefaultValues() func to the ksql.Config struct

pull/13/head
Vinícius Garcia 2021-10-30 12:08:03 -03:00
parent 45cb57cfa8
commit 4b75696bb4
2 changed files with 12 additions and 4 deletions

View File

@ -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

14
ksql.go
View File

@ -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