vendor/github.com/jinzhu/gorm/logger.go

   1 package gorm
   2
   3 import (
   4         "database/sql/driver"
   5         "fmt"
   6         "log"
   7         "os"
   8         "reflect"
   9         "regexp"
  10         "strconv"
  11         "time"
  12         "unicode"
  13 )
  14
  15 var (
  16         defaultLogger            = Logger{log.New(os.Stdout, "\r\n", 0)}
  17         sqlRegexp                = regexp.MustCompile(`\?`)
  18         numericPlaceHolderRegexp = regexp.MustCompile(`\$\d+`)
  19 )
  20
  21 func isPrintable(s string) bool {
  22         for _, r := range s {
  23                 if !unicode.IsPrint(r) {
  24                         return false
  25                 }
  26         }
  27         return true
  28 }
  29
  30 var LogFormatter = func(values ...interface{}) (messages []interface{}) {
  31         if len(values) > 1 {
  32                 var (
  33                         sql             string
  34                         formattedValues []string
  35                         level           = values[0]
  36                         currentTime     = "\n\033[33m[" + NowFunc().Format("2006-01-02 15:04:05") + "]\033[0m"
  37                         source          = fmt.Sprintf("\033[35m(%v)\033[0m", values[1])
  38                 )
  39
  40                 messages = []interface{}{source, currentTime}
  41
  42                 if level == "sql" {
  43                         // duration
  44                         messages = append(messages, fmt.Sprintf(" \033[36;1m[%.2fms]\033[0m ", float64(values[2].(time.Duration).Nanoseconds()/1e4)/100.0))
  45                         // sql
  46
  47                         for _, value := range values[4].([]interface{}) {
  48                                 indirectValue := reflect.Indirect(reflect.ValueOf(value))
  49                                 if indirectValue.IsValid() {
  50                                         value = indirectValue.Interface()
  51                                         if t, ok := value.(time.Time); ok {
  52                                                 formattedValues = append(formattedValues, fmt.Sprintf("'%v'", t.Format("2006-01-02 15:04:05")))
  53                                         } else if b, ok := value.([]byte); ok {
  54                                                 if str := string(b); isPrintable(str) {
  55                                                         formattedValues = append(formattedValues, fmt.Sprintf("'%v'", str))
  56                                                 } else {
  57                                                         formattedValues = append(formattedValues, "'<binary>'")
  58                                                 }
  59                                         } else if r, ok := value.(driver.Valuer); ok {
  60                                                 if value, err := r.Value(); err == nil && value != nil {
  61                                                         formattedValues = append(formattedValues, fmt.Sprintf("'%v'", value))
  62                                                 } else {
  63                                                         formattedValues = append(formattedValues, "NULL")
  64                                                 }
  65                                         } else {
  66                                                 formattedValues = append(formattedValues, fmt.Sprintf("'%v'", value))
  67                                         }
  68                                 } else {
  69                                         formattedValues = append(formattedValues, "NULL")
  70                                 }
  71                         }
  72
  73                         // differentiate between $n placeholders or else treat like ?
  74                         if numericPlaceHolderRegexp.MatchString(values[3].(string)) {
  75                                 sql = values[3].(string)
  76                                 for index, value := range formattedValues {
  77                                         placeholder := fmt.Sprintf(`\$%d([^\d]|$)`, index+1)
  78                                         sql = regexp.MustCompile(placeholder).ReplaceAllString(sql, value+"$1")
  79                                 }
  80                         } else {
  81                                 formattedValuesLength := len(formattedValues)
  82                                 for index, value := range sqlRegexp.Split(values[3].(string), -1) {
  83                                         sql += value
  84                                         if index < formattedValuesLength {
  85                                                 sql += formattedValues[index]
  86                                         }
  87                                 }
  88                         }
  89
  90                         messages = append(messages, sql)
  91                         messages = append(messages, fmt.Sprintf(" \n\033[36;31m[%v]\033[0m ", strconv.FormatInt(values[5].(int64), 10)+" rows affected or returned "))
  92                 } else {
  93                         messages = append(messages, "\033[31;1m")
  94                         messages = append(messages, values[2:]...)
  95                         messages = append(messages, "\033[0m")
  96                 }
  97         }
  98
  99         return
 100 }
 101
 102 type logger interface {
 103         Print(v ...interface{})
 104 }
 105
 106 // LogWriter log writer interface
 107 type LogWriter interface {
 108         Println(v ...interface{})
 109 }
 110
 111 // Logger default logger
 112 type Logger struct {
 113         LogWriter
 114 }
 115
 116 // Print format & print log
 117 func (logger Logger) Print(values ...interface{}) {
 118         logger.Println(LogFormatter(values...)...)
 119 }