sql log

[quan-xiansheng]

Hello, may I ask how sqlx prints sql statements? If it is not supported, may I ask whether this function can be extended later? This function is very common in business, I often struggle with gorm and sqlx because of this problem, and I personally like the simple use of sqlx.以上翻译结果来自有道神经网络翻译（YNMT）· 通用场景

[wen-templari]

Perhaps you could try something like this: sqlhooks, and print SQL yourself inside the hooks.

package repo

import (
    "context"
    "database/sql"
    "os"
    "time"

    "github.com/go-sql-driver/mysql"
    "github.com/nbtca/saturday/util"
    "github.com/qustavo/sqlhooks/v2"
    "github.com/sirupsen/logrus"

    "github.com/jmoiron/sqlx"
)

var db *sqlx.DB

// Hooks satisfies the sqlhooks.Hooks interface
type Hooks struct{}

// Before hook will print the query with it's args and return the context with the timestamp
func (h *Hooks) Before(ctx context.Context, query string, args ...interface{}) (context.Context, error) {
    return context.WithValue(ctx, "begin", time.Now()), nil
}

// After hook will get the timestamp registered on the Before hook and print the elapsed time
func (h *Hooks) After(ctx context.Context, query string, args ...interface{}) (context.Context, error) {
    begin := ctx.Value("begin").(time.Time)
    util.Logger.WithFields(logrus.Fields{
        "query":   query,
        "args":    args,
        "elapsed": time.Since(begin),
    }).Debug("SQL executed")
    return ctx, nil
}

func InitDB() {
    var err error
    sql.Register("mysqlWithHooks", sqlhooks.Wrap(&mysql.MySQLDriver{}, &Hooks{}))
    db, err = sqlx.Connect("mysqlWithHooks", os.Getenv("DB_URL"))

    if err != nil {
        util.Logger.Fatal(err)
    }
    db.SetMaxOpenConns(1000)               // The default is 0 (unlimited)
    db.SetMaxIdleConns(10)                 // defaultMaxIdleConns = 2
    db.SetConnMaxLifetime(time.Minute * 5) // 0, connections are reused forever.
}

func SetDB(dbx *sqlx.DB) {
    db = dbx
}

func CloseDB() {
    db.Close()
}