goext/sq/database.go

package sq

import (
	"context"
	"database/sql"
	"github.com/jmoiron/sqlx"
	"sync"
)

type DB interface {
	Exec(ctx context.Context, sql string, prep PP) (sql.Result, error)
	Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error)
	Ping(ctx context.Context) error
	BeginTransaction(ctx context.Context, iso sql.IsolationLevel) (Tx, error)
	SetListener(listener Listener)
}

type database struct {
	db    *sqlx.DB
	txctr uint16
	lock  sync.Mutex
	lstr  Listener
}

func NewDB(db *sqlx.DB) DB {
	return &database{
		db:    db,
		txctr: 0,
		lock:  sync.Mutex{},
	}
}

func (db *database) SetListener(listener Listener) {
	db.lstr = listener
}

func (db *database) Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) {
	if db.lstr != nil {
		db.lstr.OnExec(nil, sql, &prep)
	}

	res, err := db.db.NamedExecContext(ctx, sql, prep)
	if err != nil {
		return nil, err
	}
	return res, nil
}

func (db *database) Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) {
	if db.lstr != nil {
		db.lstr.OnQuery(nil, sql, &prep)
	}

	rows, err := db.db.NamedQueryContext(ctx, sql, prep)
	if err != nil {
		return nil, err
	}
	return rows, nil
}

func (db *database) Ping(ctx context.Context) error {
	if db.lstr != nil {
		db.lstr.OnPing()
	}

	err := db.db.PingContext(ctx)
	if err != nil {
		return err
	}
	return nil
}

func (db *database) BeginTransaction(ctx context.Context, iso sql.IsolationLevel) (Tx, error) {
	db.lock.Lock()
	txid := db.txctr
	db.txctr += 1 // with overflow !
	db.lock.Unlock()

	if db.lstr != nil {
		db.lstr.OnTxBegin(txid)
	}

	xtx, err := db.db.BeginTxx(ctx, &sql.TxOptions{Isolation: iso})
	if err != nil {
		return nil, err
	}

	return NewTransaction(xtx, txid, db.lstr), nil
}