wp-go/db/db.go

package db

import (
	"context"
	_ "github.com/go-sql-driver/mysql"
	"github.com/jmoiron/sqlx"
	"github/fthvgb1/wp-go/config"
	"log"
	"os"
	"strings"
)

var Db *sqlx.DB

type SqlxDb struct {
	sqlx *sqlx.DB
}

func NewSqlxDb(sqlx *sqlx.DB) *SqlxDb {
	return &SqlxDb{sqlx: sqlx}
}

func (r SqlxDb) Select(ctx context.Context, dest any, sql string, params ...any) error {
	if os.Getenv("SHOW_SQL") == "true" {
		log.Printf(strings.Replace(sql, "?", "'%v'", -1), params...)
	}
	return r.sqlx.Select(dest, sql, params...)
}

func (r SqlxDb) Get(ctx context.Context, dest any, sql string, params ...any) error {
	if os.Getenv("SHOW_SQL") == "true" {
		log.Printf(strings.Replace(sql, "?", "'%v'", -1), params...)
	}
	return r.sqlx.Get(dest, sql, params...)
}

func InitDb() error {
	dsn := config.Conf.Mysql.Dsn.GetDsn()
	var err error
	Db, err = sqlx.Open("mysql", dsn)
	if err != nil {
		return err
	}
	if config.Conf.Mysql.Pool.ConnMaxIdleTime != 0 {
		Db.SetConnMaxIdleTime(config.Conf.Mysql.Pool.ConnMaxLifetime)
	}
	if config.Conf.Mysql.Pool.MaxIdleConn != 0 {
		Db.SetMaxIdleConns(config.Conf.Mysql.Pool.MaxIdleConn)
	}
	if config.Conf.Mysql.Pool.MaxOpenConn != 0 {
		Db.SetMaxOpenConns(config.Conf.Mysql.Pool.MaxOpenConn)
	}
	if config.Conf.Mysql.Pool.ConnMaxLifetime != 0 {
		Db.SetConnMaxLifetime(config.Conf.Mysql.Pool.ConnMaxLifetime)
	}
	return err
}