pgxatomic

README

pgxatomic

pgxatomic is a library of tools that allow you to implement transfer of clean control to transactions to a higher level by adding transaction in a context.Context using pgx driver.

Example Usage

1. You can use pgxatomic.Pool within repository implementation. It's simple wrapper around pgxpool.Pool which is wrapping Query, QueryRow and Exec methods with pgxatomic query functions.

type orderRepo struct {
    pool *pgxatomic.Pool // pgxpool.Pool wrapper
}

type order struct {
    ID uuid.UUID
    Cost int
}

func (r *orderRepo) Insert(ctx context.Context, cost int) order {
    rows, _ := r.pool.Query(ctx, "insert into order(cost) values ($1) RETURNING id, cost", cost)
    o, _ := pgx.CollectOneRow(rows, pgx.RowToStructByPos[order])
    return o
}

Or you can use Query, QueryRow, Exec functions directly from the library.

1. Run wrapped usecase method calls within txFunc using pgxatomic.runner.Run function

conf, _ := pgxpool.ParseConfig("postgres://user:pass@localhost:5432/postgres")
pool, _ := pgxpool.NewWithConfig(context.Background(), conf)

r, _ := pgxatomic.NewRunner(pool, pgx.TxOptions{})

_ = r.Run(context.Background(), func(txCtx context.Context) error {
    _ = orderService.Create(txCtx)
    _ = balanceService.Withdraw(txCtx)
    return nil
})

Error handling is omitted on purpose, handle all errors!

Credits

• Clean transactions in Golang hexagon

Documentation

Index

• func Exec(ctx context.Context, db interface{ ... }, sql string, args ...any) (pgconn.CommandTag, error)
• func NewRunner(db txStarter, o pgx.TxOptions) (runner, error)
• func Query(ctx context.Context, db interface{ ... }, sql string, args ...any) (pgx.Rows, error)
• func QueryRow(ctx context.Context, db interface{ ... }, sql string, args ...any) pgx.Row
• type Pool
  • func NewPool(p *pgxpool.Pool) (Pool, error)
  • func (p Pool) Exec(ctx context.Context, sql string, args ...any) (pgconn.CommandTag, error)
  • func (p Pool) Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)
  • func (p Pool) QueryRow(ctx context.Context, sql string, args ...any) pgx.Row

Functions

func Exec

func Exec(
	ctx context.Context,
	db interface {
		Exec(ctx context.Context, sql string, args ...any) (pgconn.CommandTag, error)
	},
	sql string,
	args ...any,
) (pgconn.CommandTag, error)

Exec is a wrapper around pgx Exec method.

func NewRunner

func NewRunner(db txStarter, o pgx.TxOptions) (runner, error)

func Query

func Query(
	ctx context.Context,
	db interface {
		Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)
	},
	sql string,
	args ...any,
) (pgx.Rows, error)

Query is a wrapper around pgx Query method.

func QueryRow

func QueryRow(
	ctx context.Context,
	db interface {
		QueryRow(ctx context.Context, sql string, args ...any) pgx.Row
	},
	sql string,
	args ...any,
) pgx.Row

QueryRow is a wrapper around pgx QueryRow method.

Types

type Pool

type Pool struct {
	// contains filtered or unexported fields
}

Pool wraps pgxpool.Pool query methods with pgxatomic corresponding functions which injects pgx.Tx into context.

func NewPool

func NewPool(p *pgxpool.Pool) (Pool, error)

func (Pool) Exec

func (p Pool) Exec(ctx context.Context, sql string, args ...any) (pgconn.CommandTag, error)

func (Pool) Query

func (p Pool) Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)

func (Pool) QueryRow

func (p Pool) QueryRow(ctx context.Context, sql string, args ...any) pgx.Row