Browse Source

优化日志输出

lxg 3 years ago
parent
commit
d1009b69d7
2 changed files with 5 additions and 3 deletions
  1. 2 0
      cmd/main.go
  2. 3 3
      pkg/logger/gorm.go

+ 2 - 0
cmd/main.go

@@ -42,5 +42,7 @@ func main() {
 	}
 	fmt.Println(crud.Attach(&User{}))
 	crud.Routes(nil)
+	crud.WithDB(crud.DB().Debug())
 	time.Sleep(time.Second * 5)
+	crud.DB().Exec("select * from aaxxaa")
 }

+ 3 - 3
pkg/logger/gorm.go

@@ -39,9 +39,9 @@ func (l Logger) Trace(ctx context.Context, begin time.Time, fc func() (sql strin
 	case err != nil && l.LogLevel >= logger.Error && (!errors.Is(err, gorm.ErrRecordNotFound)):
 		sql, rows := fc()
 		if rows == -1 {
-			l.Error(ctx, sql)
+			l.Error(ctx, "%s %s", sql, err.Error())
 		} else {
-			l.Error(ctx, sql)
+			l.Error(ctx, "%s %s", sql, err.Error())
 		}
 	case elapsed > l.SlowThreshold && l.SlowThreshold != 0:
 		sql, _ := fc()
@@ -52,7 +52,7 @@ func (l Logger) Trace(ctx context.Context, begin time.Time, fc func() (sql strin
 		if rows == -1 {
 			log.Debugf("[SQL] %s", sql)
 		} else {
-			log.Debugf("[SQL] %s -> %d", sql, rows)
+			log.Debugf("[SQL] %s: %d", sql, rows)
 		}
 	}
 }