Skip to content

Commit 10e217e

Browse files
liu-xuewenjinzhu
authored andcommitted
Print affected rows (go-gorm#1541)
* fix better * add the rows number that the sql result affected or returned
1 parent 2a14638 commit 10e217e

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

logger.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77
"os"
88
"reflect"
99
"regexp"
10+
"strconv"
1011
"time"
1112
"unicode"
1213
)
@@ -87,6 +88,7 @@ var LogFormatter = func(values ...interface{}) (messages []interface{}) {
8788
}
8889

8990
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 "))
9092
} else {
9193
messages = append(messages, "\033[31;1m")
9294
messages = append(messages, values[2:]...)

main.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -702,7 +702,7 @@ func (s *DB) GetErrors() []error {
702702
////////////////////////////////////////////////////////////////////////////////
703703

704704
func (s *DB) clone() *DB {
705-
db := DB{
705+
db := &DB{
706706
db: s.db,
707707
parent: s.parent,
708708
logger: s.logger,
@@ -723,8 +723,8 @@ func (s *DB) clone() *DB {
723723
db.search = s.search.clone()
724724
}
725725

726-
db.search.db = &db
727-
return &db
726+
db.search.db = db
727+
return db
728728
}
729729

730730
func (s *DB) print(v ...interface{}) {
@@ -739,6 +739,6 @@ func (s *DB) log(v ...interface{}) {
739739

740740
func (s *DB) slog(sql string, t time.Time, vars ...interface{}) {
741741
if s.logMode == 2 {
742-
s.print("sql", fileWithLineNum(), NowFunc().Sub(t), sql, vars)
742+
s.print("sql", fileWithLineNum(), NowFunc().Sub(t), sql, vars, s.RowsAffected)
743743
}
744744
}

0 commit comments

Comments
 (0)