mirror of https://github.com/VinGarcia/ksql.git
42 lines
969 B
Go
42 lines
969 B
Go
package kpgx
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
"github.com/vingarcia/ksql"
|
|
"github.com/vingarcia/ksql/sqldialect"
|
|
)
|
|
|
|
// NewFromPgxPool builds a ksql.DB from a *pgxpool.Pool instance
|
|
func NewFromPgxPool(pool *pgxpool.Pool) (db ksql.DB, err error) {
|
|
return ksql.NewWithAdapter(NewPGXAdapter(pool), sqldialect.PostgresDialect{})
|
|
}
|
|
|
|
// New instantiates a new ksql.Client using pgx as the backend driver
|
|
func New(
|
|
ctx context.Context,
|
|
connectionString string,
|
|
config ksql.Config,
|
|
) (db ksql.DB, err error) {
|
|
config.SetDefaultValues()
|
|
|
|
pgxConf, err := pgxpool.ParseConfig(connectionString)
|
|
if err != nil {
|
|
return ksql.DB{}, err
|
|
}
|
|
|
|
pgxConf.MaxConns = int32(config.MaxOpenConns)
|
|
|
|
pool, err := pgxpool.ConnectConfig(ctx, pgxConf)
|
|
if err != nil {
|
|
return ksql.DB{}, err
|
|
}
|
|
if err = pool.Ping(ctx); err != nil {
|
|
return ksql.DB{}, err
|
|
}
|
|
|
|
db, err = ksql.NewWithAdapter(NewPGXAdapter(pool), sqldialect.PostgresDialect{})
|
|
return db, err
|
|
}
|