check if the logger is the implement of log.ContextLogger first

This commit is contained in:
KowloonZh 2020-06-29 03:12:41 +00:00
parent 9b41b879a7
commit bbc13e5087
1 changed files with 2 additions and 2 deletions

View File

@ -143,10 +143,10 @@ func (engine *Engine) Logger() log.ContextLogger {
func (engine *Engine) SetLogger(logger interface{}) {
var realLogger log.ContextLogger
switch t := logger.(type) {
case log.Logger:
realLogger = log.NewLoggerAdapter(t)
case log.ContextLogger:
realLogger = t
case log.Logger:
realLogger = log.NewLoggerAdapter(t)
}
engine.logger = realLogger
engine.DB().Logger = realLogger