16 defaultLogger = Logger{log.New(os.Stdout, "\r\n", 0)}
17 sqlRegexp = regexp.MustCompile(`\?`)
18 numericPlaceHolderRegexp = regexp.MustCompile(`\$\d+`)
21 func isPrintable(s string) bool {
23 if !unicode.IsPrint(r) {
30 var LogFormatter = func(values ...interface{}) (messages []interface{}) {
34 formattedValues []string
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])
40 messages = []interface{}{source, currentTime}
44 messages = append(messages, fmt.Sprintf(" \033[36;1m[%.2fms]\033[0m ", float64(values[2].(time.Duration).Nanoseconds()/1e4)/100.0))
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))
57 formattedValues = append(formattedValues, "'<binary>'")
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))
63 formattedValues = append(formattedValues, "NULL")
66 formattedValues = append(formattedValues, fmt.Sprintf("'%v'", value))
69 formattedValues = append(formattedValues, "NULL")
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")
81 formattedValuesLength := len(formattedValues)
82 for index, value := range sqlRegexp.Split(values[3].(string), -1) {
84 if index < formattedValuesLength {
85 sql += formattedValues[index]
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 "))
93 messages = append(messages, "\033[31;1m")
94 messages = append(messages, values[2:]...)
95 messages = append(messages, "\033[0m")
102 type logger interface {
103 Print(v ...interface{})
106 // LogWriter log writer interface
107 type LogWriter interface {
108 Println(v ...interface{})
111 // Logger default logger
116 // Print format & print log
117 func (logger Logger) Print(values ...interface{}) {
118 logger.Println(LogFormatter(values...)...)