tx.go

package pgpkg

import (
	"database/sql"
	"strings"
)

type PkgTx struct {
	*sql.Tx
}

func (t *PkgTx) Exec(query string, args ...any) (sql.Result, error) {
	t.logQuery(query, args)
	return t.Tx.Exec(query, args...)
}

func (t *PkgTx) Query(query string, args ...any) (*sql.Rows, error) {
	t.logQuery(query, args)
	return t.Tx.Query(query, args...)
}

func (t *PkgTx) QueryRow(query string, args ...any) *sql.Row {
	t.logQuery(query, args)
	return t.Tx.QueryRow(query, args...)
}

func (t *PkgTx) logQuery(query string, args []any) {
	if !Options.Verbose {
		return
	}

	logText := query
	newLine := strings.IndexRune(logText, '\n')
	if newLine >= 0 {
		logText = logText[:newLine] + "…"
	}

	if args == nil || len(args) == 0 {
		Verbose.Println(logText)
	} else {
		Verbose.Println(query, args)
	}
}