forked from mirror/gorm
72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
package gorm
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"reflect"
|
|
"regexp"
|
|
"time"
|
|
)
|
|
|
|
type logger interface {
|
|
Print(v ...interface{})
|
|
}
|
|
|
|
type LogWriter interface {
|
|
Println(v ...interface{})
|
|
}
|
|
|
|
type Logger struct {
|
|
LogWriter
|
|
}
|
|
|
|
var defaultLogger = Logger{log.New(os.Stdout, "\r\n", 0)}
|
|
|
|
// Format log
|
|
var sqlRegexp = regexp.MustCompile(`(\$\d+)|\?`)
|
|
|
|
func (logger Logger) Print(values ...interface{}) {
|
|
if len(values) > 1 {
|
|
level := values[0]
|
|
currentTime := "\n\033[33m[" + NowFunc().Format("2006-01-02 15:04:05") + "]\033[0m"
|
|
source := fmt.Sprintf("\033[35m(%v)\033[0m", values[1])
|
|
messages := []interface{}{source, currentTime}
|
|
|
|
if level == "sql" {
|
|
// duration
|
|
messages = append(messages, fmt.Sprintf(" \033[36;1m[%.2fms]\033[0m ", float64(values[2].(time.Duration).Nanoseconds()/1e4)/100.0))
|
|
// sql
|
|
var formatedValues []interface{}
|
|
for _, value := range values[4].([]interface{}) {
|
|
indirectValue := reflect.Indirect(reflect.ValueOf(value))
|
|
if indirectValue.IsValid() {
|
|
value = indirectValue.Interface()
|
|
if t, ok := value.(time.Time); ok {
|
|
formatedValues = append(formatedValues, fmt.Sprintf("'%v'", t.Format(time.RFC3339)))
|
|
} else if b, ok := value.([]byte); ok {
|
|
formatedValues = append(formatedValues, fmt.Sprintf("'%v'", string(b)))
|
|
} else if r, ok := value.(driver.Valuer); ok {
|
|
if value, err := r.Value(); err == nil && value != nil {
|
|
formatedValues = append(formatedValues, fmt.Sprintf("'%v'", value))
|
|
} else {
|
|
formatedValues = append(formatedValues, "NULL")
|
|
}
|
|
} else {
|
|
formatedValues = append(formatedValues, fmt.Sprintf("'%v'", value))
|
|
}
|
|
} else {
|
|
formatedValues = append(formatedValues, fmt.Sprintf("'%v'", value))
|
|
}
|
|
}
|
|
messages = append(messages, fmt.Sprintf(sqlRegexp.ReplaceAllString(values[3].(string), "%v"), formatedValues...))
|
|
} else {
|
|
messages = append(messages, "\033[31;1m")
|
|
messages = append(messages, values[2:]...)
|
|
messages = append(messages, "\033[0m")
|
|
}
|
|
logger.Println(messages...)
|
|
}
|
|
}
|