From 4c2f555b8158a58fa558bbe36c040d48686bc6ea Mon Sep 17 00:00:00 2001
From: liuxuewen <675073505@qq.com>
Date: Fri, 21 Jul 2017 12:04:21 +0800
Subject: [PATCH] add the rows number that the sql result affected or returned

---
 logger.go | 2 ++
 main.go   | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/logger.go b/logger.go
index 117b0403..4324a2e4 100644
--- a/logger.go
+++ b/logger.go
@@ -7,6 +7,7 @@ import (
 	"os"
 	"reflect"
 	"regexp"
+	"strconv"
 	"time"
 	"unicode"
 )
@@ -87,6 +88,7 @@ var LogFormatter = func(values ...interface{}) (messages []interface{}) {
 			}
 
 			messages = append(messages, sql)
+			messages = append(messages, fmt.Sprintf(" \n\033[36;31m[%v]\033[0m ", strconv.FormatInt(values[5].(int64), 10)+" rows affected or returned "))
 		} else {
 			messages = append(messages, "\033[31;1m")
 			messages = append(messages, values[2:]...)
diff --git a/main.go b/main.go
index dac06848..0f2fd1f5 100644
--- a/main.go
+++ b/main.go
@@ -739,6 +739,6 @@ func (s *DB) log(v ...interface{}) {
 
 func (s *DB) slog(sql string, t time.Time, vars ...interface{}) {
 	if s.logMode == 2 {
-		s.print("sql", fileWithLineNum(), NowFunc().Sub(t), sql, vars)
+		s.print("sql", fileWithLineNum(), NowFunc().Sub(t), sql, vars, s.RowsAffected)
 	}
 }