142 lines
3.6 KiB
Go
142 lines
3.6 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
"github.com/lbryio/lbry.go/v2/extras/errors"
|
|
querytools "github.com/lbryio/lbry.go/v2/extras/query"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/volatiletech/sqlboiler/v4/boil"
|
|
)
|
|
|
|
func logQueryTime(logger *log.Logger, startTime time.Time) {
|
|
logger.Debugln("Query took " + time.Since(startTime).String())
|
|
}
|
|
|
|
// QueryLogger implements the Executor interface
|
|
type QueryLogger struct {
|
|
DB *sql.DB
|
|
Logger *log.Logger
|
|
Enabled bool
|
|
}
|
|
|
|
// Query implements the Executor Query function
|
|
func (d *QueryLogger) Query(query string, args ...interface{}) (*sql.Rows, error) {
|
|
if d.Logger != nil {
|
|
q, err := querytools.InterpolateParams(query, args...)
|
|
if err != nil {
|
|
return nil, errors.Err(err)
|
|
}
|
|
d.Logger.Debugln(q)
|
|
defer logQueryTime(d.Logger, time.Now())
|
|
}
|
|
return d.DB.Query(query, args...)
|
|
}
|
|
|
|
// Exec implements the Executor Exec function
|
|
func (d *QueryLogger) Exec(query string, args ...interface{}) (sql.Result, error) {
|
|
if d.Logger != nil {
|
|
q, err := querytools.InterpolateParams(query, args...)
|
|
if err != nil {
|
|
return nil, errors.Err(err)
|
|
}
|
|
d.Logger.Debugln(q)
|
|
defer logQueryTime(d.Logger, time.Now())
|
|
}
|
|
return d.DB.Exec(query, args...)
|
|
}
|
|
|
|
// QueryRow implements the Executor QueryRow function
|
|
func (d *QueryLogger) QueryRow(query string, args ...interface{}) *sql.Row {
|
|
if d.Logger != nil {
|
|
q, err := querytools.InterpolateParams(query, args...)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
d.Logger.Debugln(q)
|
|
defer logQueryTime(d.Logger, time.Now())
|
|
}
|
|
return d.DB.QueryRow(query, args...)
|
|
}
|
|
|
|
// Begin implements the Executor Begin function
|
|
func (d *QueryLogger) Begin() (boil.Transactor, error) {
|
|
if d.Logger != nil {
|
|
d.Logger.Debug("-> Beginning tx")
|
|
}
|
|
tx, err := d.DB.Begin()
|
|
if err != nil {
|
|
return tx, err
|
|
}
|
|
return &queryLoggerTx{Tx: tx, logger: d.Logger}, nil
|
|
}
|
|
|
|
// Close implements the Executor Close function
|
|
func (d *QueryLogger) Close() error {
|
|
if d.Logger != nil {
|
|
d.Logger.Print("Closing DB connection")
|
|
}
|
|
return d.DB.Close()
|
|
}
|
|
|
|
type queryLoggerTx struct {
|
|
Tx *sql.Tx
|
|
logger *log.Logger
|
|
}
|
|
|
|
// Query implements the Executor Query function on a transaction
|
|
func (t *queryLoggerTx) Query(query string, args ...interface{}) (*sql.Rows, error) {
|
|
if t.logger != nil {
|
|
q, err := querytools.InterpolateParams(query, args...)
|
|
if err != nil {
|
|
return nil, errors.Err(err)
|
|
}
|
|
t.logger.Debugln("-> " + q)
|
|
defer logQueryTime(t.logger, time.Now())
|
|
}
|
|
return t.Tx.Query(query, args...)
|
|
}
|
|
|
|
// Exec implements the Executor Exec function on a transaction
|
|
func (t *queryLoggerTx) Exec(query string, args ...interface{}) (sql.Result, error) {
|
|
if t.logger != nil {
|
|
q, err := querytools.InterpolateParams(query, args...)
|
|
if err != nil {
|
|
return nil, errors.Err(err)
|
|
}
|
|
t.logger.Debugln("-> " + q)
|
|
defer logQueryTime(t.logger, time.Now())
|
|
}
|
|
return t.Tx.Exec(query, args...)
|
|
}
|
|
|
|
// QueryRow implements the Executor QueryRow function on a transaction
|
|
func (t *queryLoggerTx) QueryRow(query string, args ...interface{}) *sql.Row {
|
|
if t.logger != nil {
|
|
q, err := querytools.InterpolateParams(query, args...)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
t.logger.Debugln("-> " + q)
|
|
defer logQueryTime(t.logger, time.Now())
|
|
}
|
|
return t.Tx.QueryRow(query, args...)
|
|
}
|
|
|
|
// Commit implements the Executor Commit function on a transaction
|
|
func (t *queryLoggerTx) Commit() error {
|
|
if t.logger != nil {
|
|
t.logger.Debug("-> Committing tx")
|
|
}
|
|
return t.Tx.Commit()
|
|
}
|
|
|
|
// RollBack implements the Executor Rollback function on a transaction
|
|
func (t *queryLoggerTx) Rollback() error {
|
|
if t.logger != nil {
|
|
t.logger.Debug("-> Rolling back tx")
|
|
}
|
|
return t.Tx.Rollback()
|
|
}
|