Feat: SetPrintLevel
This commit is contained in:
parent
1f997319a3
commit
5364eb4794
25
wrap.go
25
wrap.go
@ -41,8 +41,10 @@ var _ StdCompatLogger = &log.Logger{}
|
|||||||
|
|
||||||
type Logger struct {
|
type Logger struct {
|
||||||
*zerolog.Logger
|
*zerolog.Logger
|
||||||
prefix string
|
|
||||||
*sync.RWMutex
|
*sync.RWMutex
|
||||||
|
|
||||||
|
prefix string
|
||||||
|
printLevel zerolog.Level
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *Logger) SetPrefix(prefix string) {
|
func (l *Logger) SetPrefix(prefix string) {
|
||||||
@ -51,6 +53,12 @@ func (l *Logger) SetPrefix(prefix string) {
|
|||||||
l.Unlock()
|
l.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (l *Logger) SetPrintLevel(level zerolog.Level) {
|
||||||
|
l.Lock()
|
||||||
|
l.printLevel = level
|
||||||
|
l.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
func (l *Logger) Prefix() string {
|
func (l *Logger) Prefix() string {
|
||||||
l.RLock()
|
l.RLock()
|
||||||
defer l.RUnlock()
|
defer l.RUnlock()
|
||||||
@ -59,7 +67,19 @@ func (l *Logger) Prefix() string {
|
|||||||
|
|
||||||
func (l *Logger) Println(v ...interface{}) {
|
func (l *Logger) Println(v ...interface{}) {
|
||||||
l.RLock()
|
l.RLock()
|
||||||
l.Logger.Info().Msg(fmt.Sprintln(v...))
|
l.Logger.WithLevel(l.printLevel).Msg(fmt.Sprintln(v...))
|
||||||
|
l.RUnlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Printf(format string, v ...interface{}) {
|
||||||
|
l.RLock()
|
||||||
|
l.Logger.WithLevel(l.printLevel).Msgf(format, v...)
|
||||||
|
l.RUnlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Print(v ...interface{}) {
|
||||||
|
l.RLock()
|
||||||
|
l.Logger.WithLevel(l.printLevel).Msg(fmt.Sprint(v...))
|
||||||
l.RUnlock()
|
l.RUnlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -225,6 +245,7 @@ func (h prefixHook) Run(e *zerolog.Event, _ zerolog.Level, _ string) {
|
|||||||
func Wrap(l zerolog.Logger) *Logger {
|
func Wrap(l zerolog.Logger) *Logger {
|
||||||
wrapped := &Logger{
|
wrapped := &Logger{
|
||||||
RWMutex: &sync.RWMutex{},
|
RWMutex: &sync.RWMutex{},
|
||||||
|
printLevel: zerolog.InfoLevel,
|
||||||
}
|
}
|
||||||
p := prefixHook{wrapped}
|
p := prefixHook{wrapped}
|
||||||
l = l.Hook(p)
|
l = l.Hook(p)
|
||||||
|
Loading…
Reference in New Issue
Block a user