go语言里的sqlx请问怎么打印最终于执行的sql语句

[feikeq]

没有这功能在调式的时候很不方便啊！

[yzx9]

You could try wrapping the database driver with sqlhooks. This suggestion originally came from a StackOverflow post, though I haven't been able to locate it again.

var logger *slog.Logger
const SlowThreshold = 300 * time.Millisecond

func init() {
	hook := customHook{
		slowThreshold: SlowThreshold,
	}
	sql.Register("mysql-custom", sqlhooks.Wrap(&mysql.MySQLDriver{}, &hook))
}

type customHook struct {
	slowThreshold time.Duration
}

// make sure hook implement `sqlhooks.Hooks`
var _ interface{ sqlhooks.Hooks } = (*customHook)(nil)

type hookKey string

const hookStartAt hookKey = "startAt"

func (h *customHook) Before(ctx context.Context, query string, args ...any) (context.Context, error) {
	ctx = context.WithValue(ctx, hookStartAt, time.Now())
	return ctx, nil
}

// not run if on error
func (h *customHook) After(ctx context.Context, query string, args ...any) (context.Context, error) {
	if startAt, ok := ctx.Value(hookStartAt).(time.Time); ok {
		if took := time.Since(startAt); took >= h.slowThreshold { // warn if slow
			logger.Warn("slow query",
				slog.String("query", string(readFriendly(query))),
				slog.Any("args", args),
				slog.Int64("took", int64(took)))
		}
	}

	return ctx, nil
}

type ErrDown struct {
	err   error
	query string
	args  []any
	took  time.Duration
}

// OnError will be called if any error happens
func (h *customHook) OnError(ctx context.Context, err error, query string, args ...any) error {
	// Not a user error: driver is telling sql package that an
	// optional interface method is not implemented. There is
	// nothing to instrument here.
	// https://golang.org/pkg/database/sql/driver/#ErrSkip
	// https://github.com/DataDog/dd-trace-go/issues/270
	if errors.Is(err, driver.ErrSkip) {
		return nil
	}

	if err == nil {
		return nil
	}

	custom := ErrDown{
		err:   err,
		query: query,
		args:  args,
	}
	if startAt, ok := ctx.Value(hookStartAt).(time.Time); ok {
		custom.took = time.Since(startAt)
	}
	return custom
}