golang-sql/gosql/common/dbmethods.go

package common

import (
	"context"
	"database/sql"
	"regexp"
)

type DBMethods struct {
	DB *sql.DB

	Driver string
}

var r = regexp.MustCompile(`\$\d+`)

func (db *DBMethods) fixQuery(query string) string {
	if db.Driver == "mysql" {
		return r.ReplaceAllString(query, "?")
	}
	return query
}

func (db *DBMethods) Close() error {
	return db.DB.Close()
}

func (db *DBMethods) Exec(ctx context.Context, query string, args ...any) (sql.Result, error) {
	return db.DB.ExecContext(ctx, db.fixQuery(query), args...)
}

func (db *DBMethods) Ping(ctx context.Context) error {
	return db.DB.PingContext(ctx)
}

func (db *DBMethods) Prepare(ctx context.Context, query string) (*sql.Stmt, error) {
	return db.DB.PrepareContext(ctx, db.fixQuery(query))
}

func (db *DBMethods) Query(ctx context.Context, query string, args ...any) (*sql.Rows, error) {
	return db.DB.QueryContext(ctx, db.fixQuery(query), args...)
}

func (db *DBMethods) QueryRow(ctx context.Context, query string, args ...any) *sql.Row {
	return db.DB.QueryRowContext(ctx, db.fixQuery(query), args...)
}