51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package ipld_eth_statedb
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgconn"
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
)
|
|
|
|
var _ Driver = &PGXDriver{}
|
|
|
|
// PGXDriver driver, implements Driver
|
|
type PGXDriver struct {
|
|
ctx context.Context
|
|
db *pgxpool.Pool
|
|
}
|
|
|
|
// NewPGXDriver returns a new pgx driver for Postgres
|
|
func NewPGXDriver(ctx context.Context, config Config) (*PGXDriver, error) {
|
|
db, err := NewPGXPool(ctx, config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &PGXDriver{ctx: ctx, db: db}, nil
|
|
}
|
|
|
|
// NewPGXDriverFromPool returns a new pgx driver for Postgres
|
|
func NewPGXDriverFromPool(ctx context.Context, db *pgxpool.Pool) (*PGXDriver, error) {
|
|
return &PGXDriver{ctx: ctx, db: db}, nil
|
|
}
|
|
|
|
// QueryRow satisfies sql.Database
|
|
func (driver *PGXDriver) QueryRow(ctx context.Context, sql string, args ...interface{}) ScannableRow {
|
|
return driver.db.QueryRow(ctx, sql, args...)
|
|
}
|
|
|
|
// Exec satisfies sql.Database
|
|
func (pgx *PGXDriver) Exec(ctx context.Context, sql string, args ...interface{}) (Result, error) {
|
|
res, err := pgx.db.Exec(ctx, sql, args...)
|
|
return resultWrapper{ct: res}, err
|
|
}
|
|
|
|
type resultWrapper struct {
|
|
ct pgconn.CommandTag
|
|
}
|
|
|
|
// RowsAffected satisfies sql.Result
|
|
func (r resultWrapper) RowsAffected() (int64, error) {
|
|
return r.ct.RowsAffected(), nil
|
|
}
|