onasty/internal/store/psqlutil/psqlutil.go(view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
package psqlutil
import (
"context"
pgxuuid "github.com/jackc/pgx-gofrs-uuid"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
)
type DB struct{ *pgxpool.Pool }
func Connect(ctx context.Context, dsn string) (*DB, error) {
dbConf, err := pgxpool.ParseConfig(dsn)
if err != nil {
return nil, err
}
dbConf.AfterConnect = func(_ context.Context, c *pgx.Conn) error {
pgxuuid.Register(c.TypeMap())
return nil
}
db, err := pgxpool.NewWithConfig(ctx, dbConf)
if err != nil {
return nil, err
}
if err := db.Ping(ctx); err != nil {
return nil, err
}
return &DB{
Pool: db,
}, nil
}
func (db *DB) Close() error {
db.Pool.Close()
return nil
}
|