package exec

import (
	"context"
	"database/sql"

	"gitee.com/knowgo/logo"
)

type Executor struct {
	db     *sql.DB
	logger logo.Logger
	logSQL LogSQL
}

func NewExecutor(db *sql.DB, logger logo.Logger, logSQL LogSQL) *Executor {
	return &Executor{db: db, logger: logger, logSQL: logSQL}
}

func (e *Executor) QueryRow(sql string, args ...interface{}) *sql.Row {
	if e.logSQL.LogSQL() {
		e.logger.Info(sql, args)
	}
	return e.db.QueryRow(sql, args...)
}

func (e *Executor) QueryRows(sql string, args ...interface{}) (*sql.Rows, error) {
	if e.logSQL.LogSQL() {
		e.logger.Info(sql, args)
	}
	return e.db.Query(sql, args...)
}

func (e *Executor) QueryContextRow(ctx context.Context, sql string, args ...interface{}) *sql.Row {
	if e.logSQL.LogSQL() {
		e.logger.Info(sql, args)
	}
	return e.db.QueryRowContext(ctx, sql, args...)
}

func (e *Executor) QueryContextRows(ctx context.Context, sql string, args ...interface{}) (*sql.Rows, error) {
	if e.logSQL.LogSQL() {
		e.logger.Info(sql, args)
	}
	return e.db.QueryContext(ctx, sql, args...)
}

func (e *Executor) Exec(sql string, args ...interface{}) (sql.Result, error) {
	if e.logSQL.LogSQL() {
		e.logger.Info(sql, args)
	}
	return e.db.Exec(sql, args...)
}

func (e *Executor) ExecContext(ctx context.Context, sql string, args ...interface{}) (sql.Result, error) {
	if e.logSQL.LogSQL() {
		e.logger.Info(sql, args)
	}
	return e.db.ExecContext(ctx, sql, args...)
}