2021-06-29 15:07:08 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2021-08-09 16:31:42 +00:00
|
|
|
"database/sql"
|
|
|
|
"fmt"
|
2021-06-29 15:07:08 +00:00
|
|
|
"log"
|
2021-08-09 16:31:42 +00:00
|
|
|
"strings"
|
2021-06-29 15:07:08 +00:00
|
|
|
"time"
|
|
|
|
|
2021-08-09 16:31:42 +00:00
|
|
|
"github.com/spf13/cast"
|
|
|
|
)
|
2021-06-29 15:07:08 +00:00
|
|
|
|
|
|
|
const dbHooksBegin contextKey = "begin"
|
|
|
|
|
2021-08-09 16:31:42 +00:00
|
|
|
func (db *database) dbBefore(ctx context.Context, query string, args ...interface{}) context.Context {
|
|
|
|
if !db.debug {
|
|
|
|
return ctx
|
|
|
|
}
|
|
|
|
return context.WithValue(ctx, dbHooksBegin, time.Now())
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *database) dbAfter(ctx context.Context, query string, args ...interface{}) {
|
|
|
|
if !db.debug {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
dur := time.Since(ctx.Value(dbHooksBegin).(time.Time))
|
|
|
|
var logBuilder strings.Builder
|
|
|
|
logBuilder.WriteString("\nQuery: ")
|
|
|
|
logBuilder.WriteString(`"`)
|
|
|
|
logBuilder.WriteString(query)
|
|
|
|
logBuilder.WriteString(`"`)
|
|
|
|
if len(args) > 0 {
|
|
|
|
logBuilder.WriteString("\nArgs: ")
|
|
|
|
for i, arg := range args {
|
|
|
|
if i > 0 {
|
|
|
|
logBuilder.WriteString(", ")
|
|
|
|
}
|
|
|
|
if named, ok := arg.(sql.NamedArg); ok && named.Name != "" {
|
|
|
|
logBuilder.WriteString("(")
|
|
|
|
logBuilder.WriteString(named.Name)
|
|
|
|
logBuilder.WriteString(`) "`)
|
|
|
|
logBuilder.WriteString(argToString(named.Value))
|
|
|
|
logBuilder.WriteString(`"`)
|
|
|
|
} else {
|
|
|
|
logBuilder.WriteString(`"`)
|
|
|
|
logBuilder.WriteString(argToString(arg))
|
|
|
|
logBuilder.WriteString(`"`)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
logBuilder.WriteString("\nDuration: ")
|
|
|
|
logBuilder.WriteString(dur.String())
|
|
|
|
log.Println(logBuilder.String())
|
2021-06-29 15:07:08 +00:00
|
|
|
}
|
|
|
|
|
2021-08-09 16:31:42 +00:00
|
|
|
func argToString(arg interface{}) string {
|
|
|
|
val := cast.ToString(arg)
|
|
|
|
if val == "" {
|
|
|
|
val = fmt.Sprintf("%v", arg)
|
|
|
|
}
|
|
|
|
return val
|
2021-06-29 15:07:08 +00:00
|
|
|
}
|