You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"time"
|
|
|
|
"git.runcible.io/learning/pulley/internal/config"
|
|
pgx "github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
// https://medium.com/@neelkanthsingh.jr/understanding-database-connection-pools-and-the-pgx-library-in-go-3087f3c5a0c
|
|
|
|
// pgxpool.New implicitly creates a dbConfig from the database uri params which can include connection params
|
|
|
|
// newPgxConfig creates a pgxpool Config from the application service config
|
|
// func newPgxConfig(cfg config.ServiceConfig) (pgxpool.Config, error) {
|
|
// dbconfig, err := pgxpool.ParseConfig(cfg.DatabaseUri)
|
|
// }
|
|
|
|
// OpenPgPool creates a db pool and tests connectivity
|
|
func OpenPgPool(ctx context.Context, cfg config.ServiceConfig) (*pgxpool.Pool, error) {
|
|
pool, err := pgxpool.New(ctx, cfg.DatabaseUri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
timoutCtx, close := context.WithTimeout(ctx, 5*time.Second)
|
|
defer close()
|
|
|
|
if err := pool.Ping(timoutCtx); err != nil {
|
|
slog.Error("Error in attempting first postgres database connection")
|
|
pool.Close()
|
|
return nil, err
|
|
}
|
|
return pool, nil
|
|
}
|
|
|
|
type PgxIface interface {
|
|
Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)
|
|
QueryRow(ctx context.Context, sql string, args ...any) pgx.Row
|
|
BeginTx(ctx context.Context, txOptions pgx.TxOptions) (pgx.Tx, error)
|
|
}
|
|
|
|
// TODO determine if this is even useful. Sqlc does produce
|
|
type DBQuierier struct {
|
|
db *pgxpool.Pool
|
|
}
|
|
|
|
func (q *DBQuierier) Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error) {
|
|
return q.db.Query(ctx, sql, args...)
|
|
}
|
|
|
|
func (q *DBQuierier) QueryRow(ctx context.Context, sql string, args ...any) pgx.Row {
|
|
return q.db.QueryRow(ctx, sql, args...)
|
|
}
|
|
|
|
func (q *DBQuierier) BeginTx(ctx context.Context, txOptions pgx.TxOptions) (pgx.Tx, error) {
|
|
return q.db.BeginTx(ctx, txOptions)
|
|
}
|