36 lines
706 B
Go
36 lines
706 B
Go
|
package logs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NoLevel(logger zerolog.Logger, level zerolog.Level) zerolog.Logger {
|
||
|
|
return logger.Hook(NewNoLevelHook(logger.GetLevel(), level))
|
||
|
|
}
|
||
|
|
|
||
|
|
type NoLevelHook struct {
|
||
|
|
minLevel zerolog.Level
|
||
|
|
level zerolog.Level
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNoLevelHook(minLevel zerolog.Level, level zerolog.Level) *NoLevelHook {
|
||
|
|
return &NoLevelHook{minLevel: minLevel, level: level}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n NoLevelHook) Run(e *zerolog.Event, level zerolog.Level, _ string) {
|
||
|
|
if n.minLevel > n.level {
|
||
|
|
e.Discard()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if level == zerolog.NoLevel {
|
||
|
|
e.Str("level", n.level.String())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func msgFunc(i ...any) func() string {
|
||
|
|
return func() string { return fmt.Sprint(i...) }
|
||
|
|
}
|