golang sqlc query log print

golang에서 sqlc를 사용하는데 쿼리를 log에 출력하는 옵션이 없어서 찾아 보니

 

driver의 hook을 설정해서 쿼리의 실행전과 실행 후에 쿼리를 찍는 소소가 있었다.

 

그런데 sqlc는 쿼리를 prepared 형식으로 던지기 때문에 

 

parameter와 쿼리가 분리되어서 출력 되는 문제가 있었다. 

 

그래서 gorm의 쿼리를 출력하는 소스를 찾아 보니 logger 폴더에 parameter를 찾아서 찍어 주는 소스가 존재 했다.

 

두개를 합치면 다음과 같은 소스로 원하는 쿼리를 출력해 볼 수 있다.

 

1. database driver hook

https://github.com/qustavo/sqlhooks

GitHub - qustavo/sqlhooks: Attach hooks to any database/sql driver

 

2. gorm logger 

https://github.com/go-gorm/gorm/blob/master/logger/logger.go

 

 

package main

import (
	"context"
	"fmt"
	"time"

	"gorm.io/gorm/logger"
	_ "github.com/qustavo/sqlhooks/v2"
	_ "github.com/go-sql-driver/mysql"	
)

// Hooks satisfies the sqlhook.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) {
	fmt.Printf("[QUERY] [%s] %s\n",
		time.Since(ctx.Value("begin").(time.Time)),
		logger.ExplainSQL(query, nil, `'`, args...),
	)
	return ctx, nil
}

func main() {
	// DB를 접속하는 경우
	// ctx := context.Background()

	// // First, register the wrapper
	// sql.Register("mysqlWithHooks", sqlhooks.Wrap(&mysql.MySQLDriver{}, &Hooks{}))
	// db, err := sql.Open("mysqlWithHooks", dsn)

	sql := "SELECT * FROM person WHERE id = ?"
	vars := []interface{}{1}

	explain := logger.ExplainSQL(sql, nil, `'`, vars...)

	fmt.Println(explain)
}

 

결과

SELECT * FROM person WHERE id = ? [1]
SELECT * FROM person WHERE id = 1