46 lines
837 B
Go
46 lines
837 B
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"github.com/zepyrshut/rating-orama/internal/app"
|
|
"github.com/zepyrshut/rating-orama/internal/sqlc"
|
|
)
|
|
|
|
type pgxRepository struct {
|
|
*sqlc.Queries
|
|
pool *pgxpool.Pool
|
|
app *app.ExtendedApp
|
|
}
|
|
|
|
var _ ExtendedQuerier = &pgxRepository{}
|
|
|
|
func NewPGXRepo(pgx *pgxpool.Pool, app *app.ExtendedApp) ExtendedQuerier {
|
|
return &pgxRepository{
|
|
Queries: sqlc.New(pgx),
|
|
pool: pgx,
|
|
app: app,
|
|
}
|
|
}
|
|
|
|
func (r *pgxRepository) execTx(ctx context.Context, fn func(*sqlc.Queries) error) error {
|
|
tx, err := r.pool.Begin(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
q := sqlc.New(tx)
|
|
|
|
err = fn(q)
|
|
if err != nil {
|
|
if rbErr := tx.Rollback(ctx); rbErr != nil {
|
|
return fmt.Errorf("tx err: %v, rb err: %v", err, rbErr)
|
|
}
|
|
return err
|
|
}
|
|
|
|
return tx.Commit(ctx)
|
|
}
|