MaysWind
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package datastore
|
|
|
|
import (
|
|
xorm "xorm.io/xorm/log"
|
|
|
|
"github.com/mayswind/lab/pkg/log"
|
|
"github.com/mayswind/lab/pkg/settings"
|
|
)
|
|
|
|
// XOrmLoggerAdapter represents the logger adapter for xorm
|
|
type XOrmLoggerAdapter struct {
|
|
enable bool
|
|
logLevel settings.Level
|
|
}
|
|
|
|
// Debug logs debug log
|
|
func (logger XOrmLoggerAdapter) Debug(v ...interface{}) {
|
|
log.SqlQuery(v...)
|
|
}
|
|
|
|
// Debugf logs debug log with custom format
|
|
func (logger XOrmLoggerAdapter) Debugf(format string, v ...interface{}) {
|
|
log.SqlQueryf(format, v...)
|
|
}
|
|
|
|
// Info logs info log
|
|
func (logger XOrmLoggerAdapter) Info(v ...interface{}) {
|
|
log.SqlQuery(v...)
|
|
}
|
|
|
|
// Infof logs info log with custom format
|
|
func (logger XOrmLoggerAdapter) Infof(format string, v ...interface{}) {
|
|
log.SqlQueryf(format, v...)
|
|
}
|
|
|
|
// Warn logs warn log
|
|
func (logger XOrmLoggerAdapter) Warn(v ...interface{}) {
|
|
log.SqlQuery(v...)
|
|
}
|
|
|
|
// Warnf logs warn log with custom format
|
|
func (logger XOrmLoggerAdapter) Warnf(format string, v ...interface{}) {
|
|
log.SqlQueryf(format, v...)
|
|
}
|
|
|
|
// Error logs error log
|
|
func (logger XOrmLoggerAdapter) Error(v ...interface{}) {
|
|
log.SqlQuery(v...)
|
|
}
|
|
|
|
// Errorf logs error log with custom format
|
|
func (logger XOrmLoggerAdapter) Errorf(format string, v ...interface{}) {
|
|
log.SqlQueryf(format, v...)
|
|
}
|
|
|
|
// Level returns the logger level
|
|
func (logger XOrmLoggerAdapter) Level() xorm.LogLevel {
|
|
if logger.logLevel == settings.LOGLEVEL_DEBUG {
|
|
return xorm.LOG_DEBUG
|
|
} else if logger.logLevel == settings.LOGLEVEL_INFO {
|
|
return xorm.LOG_INFO
|
|
} else if logger.logLevel == settings.LOGLEVEL_WARN {
|
|
return xorm.LOG_WARNING
|
|
} else if logger.logLevel == settings.LOGLEVEL_ERROR {
|
|
return xorm.LOG_ERR
|
|
}
|
|
|
|
return xorm.LOG_INFO
|
|
}
|
|
|
|
// SetLevel sets the logger level
|
|
func (logger XOrmLoggerAdapter) SetLevel(l xorm.LogLevel) {
|
|
if l == xorm.LOG_DEBUG {
|
|
logger.logLevel = settings.LOGLEVEL_DEBUG
|
|
} else if l == xorm.LOG_INFO {
|
|
logger.logLevel = settings.LOGLEVEL_INFO
|
|
} else if l == xorm.LOG_WARNING {
|
|
logger.logLevel = settings.LOGLEVEL_WARN
|
|
} else if l == xorm.LOG_ERR {
|
|
logger.logLevel = settings.LOGLEVEL_ERROR
|
|
}
|
|
|
|
logger.logLevel = settings.LOGLEVEL_INFO
|
|
}
|
|
|
|
// ShowSQL sets whether write sql to log
|
|
func (logger XOrmLoggerAdapter) ShowSQL(show ...bool) {
|
|
logger.enable = len(show) > 0 && show[0]
|
|
}
|
|
|
|
// IsShowSQL returns whether write sql to log
|
|
func (logger XOrmLoggerAdapter) IsShowSQL() bool {
|
|
return logger.enable
|
|
}
|
|
|
|
// NewXOrmLoggerAdapter returns a new XOrmLoggerAdapter instance
|
|
func NewXOrmLoggerAdapter(showSql bool, logLevel settings.Level) xorm.Logger {
|
|
return XOrmLoggerAdapter{
|
|
enable: showSql,
|
|
logLevel: logLevel,
|
|
}
|
|
}
|