Introduce own Numeric type and revert to stdlib from pgx

This commit is contained in:
2021-12-02 21:29:44 +00:00
parent 6f8a94ff5d
commit 4646356b2d
16 changed files with 136 additions and 57 deletions

View File

@ -8,7 +8,6 @@ import (
"time"
"github.com/google/uuid"
"github.com/jackc/pgtype"
)
const createTransaction = `-- name: CreateTransaction :one
@ -21,13 +20,13 @@ RETURNING id, date, memo, amount, account_id, payee_id
type CreateTransactionParams struct {
Date time.Time
Memo string
Amount pgtype.Numeric
Amount Numeric
AccountID uuid.UUID
PayeeID uuid.NullUUID
}
func (q *Queries) CreateTransaction(ctx context.Context, arg CreateTransactionParams) (Transaction, error) {
row := q.db.QueryRow(ctx, createTransaction,
row := q.db.QueryRowContext(ctx, createTransaction,
arg.Date,
arg.Memo,
arg.Amount,
@ -54,7 +53,7 @@ LIMIT 200
`
func (q *Queries) GetTransactionsForAccount(ctx context.Context, accountID uuid.UUID) ([]Transaction, error) {
rows, err := q.db.Query(ctx, getTransactionsForAccount, accountID)
rows, err := q.db.QueryContext(ctx, getTransactionsForAccount, accountID)
if err != nil {
return nil, err
}
@ -74,6 +73,9 @@ func (q *Queries) GetTransactionsForAccount(ctx context.Context, accountID uuid.
}
items = append(items, i)
}
if err := rows.Close(); err != nil {
return nil, err
}
if err := rows.Err(); err != nil {
return nil, err
}
@ -95,13 +97,13 @@ type GetTransactionsForBudgetRow struct {
ID uuid.UUID
Date time.Time
Memo string
Amount pgtype.Numeric
Amount Numeric
Account string
Payee string
}
func (q *Queries) GetTransactionsForBudget(ctx context.Context, budgetID uuid.UUID) ([]GetTransactionsForBudgetRow, error) {
rows, err := q.db.Query(ctx, getTransactionsForBudget, budgetID)
rows, err := q.db.QueryContext(ctx, getTransactionsForBudget, budgetID)
if err != nil {
return nil, err
}
@ -121,6 +123,9 @@ func (q *Queries) GetTransactionsForBudget(ctx context.Context, budgetID uuid.UU
}
items = append(items, i)
}
if err := rows.Close(); err != nil {
return nil, err
}
if err := rows.Err(); err != nil {
return nil, err
}