123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- package db
- import (
- "database/sql/driver"
- "fmt"
- "log"
- "os"
- "reflect"
- "regexp"
- "strconv"
- "time"
- "unicode"
- )
- var (
- DefaultGormLogger = GormLogger{log.New(os.Stdout, "[ORM] ", 0)}
- sqlRegexp = regexp.MustCompile(`\?`)
- numericPlaceHolderRegexp = regexp.MustCompile(`\$\d+`)
- )
- func isPrintable(s string) bool {
- for _, r := range s {
- if !unicode.IsPrint(r) {
- return false
- }
- }
- return true
- }
- var LogFormatter = func(values ...interface{}) (messages []interface{}) {
- if len(values) > 1 {
- var (
- sql string
- formattedValues []string
- level = values[0]
- currentTime = time.Now().Format("2006/01/02 15:04:05")
- // source = fmt.Sprintf(" %v", values[1])
- )
- messages = []interface{}{currentTime}
- if level == "sql" {
- // duration
- messages = append(messages, fmt.Sprintf("[%.2fms]", float64(values[2].(time.Duration).Nanoseconds()/1e4)/100.0))
- // sql
- 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 {
- formattedValues = append(formattedValues, fmt.Sprintf("'%v'", t.Format("2006-01-02 15:04:05")))
- } else if b, ok := value.([]byte); ok {
- if str := string(b); isPrintable(str) {
- formattedValues = append(formattedValues, fmt.Sprintf("'%v'", str))
- } else {
- formattedValues = append(formattedValues, "'<binary>'")
- }
- } else if r, ok := value.(driver.Valuer); ok {
- if value, err := r.Value(); err == nil && value != nil {
- formattedValues = append(formattedValues, fmt.Sprintf("'%v'", value))
- } else {
- formattedValues = append(formattedValues, "NULL")
- }
- } else {
- formattedValues = append(formattedValues, fmt.Sprintf("'%v'", value))
- }
- } else {
- formattedValues = append(formattedValues, "NULL")
- }
- }
- // differentiate between $n placeholders or else treat like ?
- if numericPlaceHolderRegexp.MatchString(values[3].(string)) {
- sql = values[3].(string)
- for index, value := range formattedValues {
- placeholder := fmt.Sprintf(`\$%d([^\d]|$)`, index+1)
- sql = regexp.MustCompile(placeholder).ReplaceAllString(sql, value+"$1")
- }
- } else {
- formattedValuesLength := len(formattedValues)
- for index, value := range sqlRegexp.Split(values[3].(string), -1) {
- sql += value
- if index < formattedValuesLength {
- sql += formattedValues[index]
- }
- }
- }
- messages = append(messages, sql)
- messages = append(messages, fmt.Sprintf("\r\n%v", strconv.FormatInt(values[5].(int64), 10)+" rows affected or returned"))
- } else {
- messages = append(messages, values[2:]...)
- }
- }
- return
- }
- type logger interface {
- Print(v ...interface{})
- }
- // Logger default logger
- type GormLogger struct {
- *log.Logger
- }
- // Print format & print log
- func (logger GormLogger) Print(values ...interface{}) {
- logger.Println(LogFormatter(values...)...)
- }
|