mirror of https://github.com/VinGarcia/ksql.git
Add SetDefaultValues() func to the ksql.Config struct
parent
45cb57cfa8
commit
4b75696bb4
|
@ -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
14
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
|
||||
|
|
Loading…
Reference in New Issue