camper/pkg/database/tx.go

45 lines
914 B
Go

/*
* SPDX-FileCopyrightText: 2023 jordi fita mas <jfita@peritasoft.com>
* SPDX-License-Identifier: AGPL-3.0-only
*/
package database
import (
"context"
"github.com/jackc/pgconn"
"github.com/jackc/pgx/v4"
)
type Tx struct {
pgx.Tx
}
func (tx *Tx) MustCommit(ctx context.Context) {
if err := tx.Tx.Commit(ctx); err != nil {
panic(err)
}
}
func (tx *Tx) MustExec(ctx context.Context, sql string, args ...interface{}) pgconn.CommandTag {
tag, err := tx.Tx.Exec(ctx, sql, args...)
if err != nil {
panic(err)
}
return tag
}
func (tx *Tx) GetInt(ctx context.Context, sql string, args ...interface{}) (int, error) {
var result int
err := tx.QueryRow(ctx, sql, args...).Scan(&result)
return result, err
}
func (tx *Tx) MustGetInt(ctx context.Context, sql string, args ...interface{}) int {
if result, err := tx.GetInt(ctx, sql, args...); err == nil {
return result
} else {
panic(err)
}
}