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,
|
connectionString string,
|
||||||
config ksql.Config,
|
config ksql.Config,
|
||||||
) (db ksql.DB, err error) {
|
) (db ksql.DB, err error) {
|
||||||
|
config.SetDefaultValues()
|
||||||
|
|
||||||
pgxConf, err := pgxpool.ParseConfig(connectionString)
|
pgxConf, err := pgxpool.ParseConfig(connectionString)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ksql.DB{}, err
|
return ksql.DB{}, err
|
||||||
|
|
14
ksql.go
14
ksql.go
|
@ -76,12 +76,20 @@ type Config struct {
|
||||||
MaxOpenConns int
|
MaxOpenConns int
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Config) SetDefaultValues() {
|
||||||
|
if c.MaxOpenConns == 0 {
|
||||||
|
c.MaxOpenConns = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// New instantiates a new KissSQL client
|
// New instantiates a new KissSQL client
|
||||||
func New(
|
func New(
|
||||||
dbDriver string,
|
dbDriver string,
|
||||||
connectionString string,
|
connectionString string,
|
||||||
config Config,
|
config Config,
|
||||||
) (DB, error) {
|
) (DB, error) {
|
||||||
|
config.SetDefaultValues()
|
||||||
|
|
||||||
db, err := sql.Open(dbDriver, connectionString)
|
db, err := sql.Open(dbDriver, connectionString)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return DB{}, err
|
return DB{}, err
|
||||||
|
@ -90,10 +98,6 @@ func New(
|
||||||
return DB{}, err
|
return DB{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if config.MaxOpenConns == 0 {
|
|
||||||
config.MaxOpenConns = 1
|
|
||||||
}
|
|
||||||
|
|
||||||
db.SetMaxOpenConns(config.MaxOpenConns)
|
db.SetMaxOpenConns(config.MaxOpenConns)
|
||||||
|
|
||||||
return NewWithAdapter(SQLAdapter{db}, dbDriver)
|
return NewWithAdapter(SQLAdapter{db}, dbDriver)
|
||||||
|
@ -108,6 +112,8 @@ func NewWithPGX(
|
||||||
connectionString string,
|
connectionString string,
|
||||||
config Config,
|
config Config,
|
||||||
) (db DB, err error) {
|
) (db DB, err error) {
|
||||||
|
config.SetDefaultValues()
|
||||||
|
|
||||||
pgxConf, err := pgxpool.ParseConfig(connectionString)
|
pgxConf, err := pgxpool.ParseConfig(connectionString)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return DB{}, err
|
return DB{}, err
|
||||||
|
|
Loading…
Reference in New Issue