Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
feat(gorm): gorm v2 migration (#170)
* feat(gorm): upgrade gorm to v2 + add new db method for closing TODO: add log support through logrus to gorm db * feat(gorm.v2): add logrus support + fix linter issues
- Loading branch information
Showing
6 changed files
with
300 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,97 @@ | ||
package db | ||
|
||
import ( | ||
"context" | ||
"fmt" | ||
"time" | ||
|
||
"github.com/sirupsen/logrus" | ||
"gorm.io/gorm/logger" | ||
"gorm.io/gorm/utils" | ||
) | ||
|
||
/********************************************* | ||
* Custom logger definition for Gorm to use Logrus | ||
* Implement gorm Logger iterface | ||
* Based on : https://github.com/go-gorm/gorm/blob/master/logger/logger.go | ||
*********************************************/ | ||
|
||
// GormLogger is a wrapper class that implement Gorm logger interface | ||
type GormLogger struct { | ||
GormLoggerConfig | ||
} | ||
|
||
// GormLoggerConfig handle GormLogger config (log level, slow threshold) | ||
type GormLoggerConfig struct { | ||
LogLevel logger.LogLevel | ||
SlowThreshold time.Duration | ||
} | ||
|
||
var ( | ||
// LogrusGormLogger default GormLogger instance for Gorm logging through Logrus | ||
LogrusGormLogger = GormLogger{ | ||
GormLoggerConfig{ | ||
LogLevel: logger.Warn, | ||
SlowThreshold: 200 * time.Millisecond, | ||
}, | ||
} | ||
) | ||
|
||
// LogMode log mode | ||
func (l *GormLogger) LogMode(level logger.LogLevel) logger.Interface { | ||
newlogger := *l | ||
l.LogLevel = level | ||
return &newlogger | ||
} | ||
|
||
// Info print info | ||
func (l *GormLogger) Info(_ context.Context, msg string, data ...interface{}) { | ||
if l.LogLevel >= logger.Info { | ||
logrus.Info(msg, append([]interface{}{utils.FileWithLineNum()}, data...)) | ||
} | ||
} | ||
|
||
// Warn print warn messages | ||
func (l *GormLogger) Warn(_ context.Context, msg string, data ...interface{}) { | ||
if l.LogLevel >= logger.Warn { | ||
logrus.Warn(msg, append([]interface{}{utils.FileWithLineNum()}, data...)) | ||
} | ||
} | ||
|
||
// Error print error messages | ||
func (l *GormLogger) Error(_ context.Context, msg string, data ...interface{}) { | ||
if l.LogLevel >= logger.Error { | ||
logrus.Warn(msg, append([]interface{}{utils.FileWithLineNum()}, data...)) | ||
} | ||
} | ||
|
||
// Trace print sql message | ||
func (l *GormLogger) Trace(_ context.Context, begin time.Time, fc func() (string, int64), err error) { | ||
if l.LogLevel > 0 { | ||
elapsed := time.Since(begin) | ||
switch { | ||
case err != nil && l.LogLevel >= logger.Error: | ||
sql, rows := fc() | ||
if rows == -1 { | ||
logrus.Trace(utils.FileWithLineNum(), err, float64(elapsed.Nanoseconds())/1e6, "-", sql) | ||
} else { | ||
logrus.Trace(utils.FileWithLineNum(), err, float64(elapsed.Nanoseconds())/1e6, rows, sql) | ||
} | ||
case elapsed > l.SlowThreshold && l.SlowThreshold != 0 && l.LogLevel >= logger.Warn: | ||
sql, rows := fc() | ||
slowLog := fmt.Sprintf("SLOW SQL >= %v", l.SlowThreshold) | ||
if rows == -1 { | ||
logrus.Trace(utils.FileWithLineNum(), slowLog, float64(elapsed.Nanoseconds())/1e6, "-", sql) | ||
} else { | ||
logrus.Trace(utils.FileWithLineNum(), slowLog, float64(elapsed.Nanoseconds())/1e6, rows, sql) | ||
} | ||
case l.LogLevel >= logger.Info: | ||
sql, rows := fc() | ||
if rows == -1 { | ||
logrus.Trace(utils.FileWithLineNum(), float64(elapsed.Nanoseconds())/1e6, "-", sql) | ||
} else { | ||
logrus.Trace(utils.FileWithLineNum(), float64(elapsed.Nanoseconds())/1e6, rows, sql) | ||
} | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.