[log] change helper func
This commit is contained in:
parent
761b1553aa
commit
d0cae7b6ed
|
@ -17,21 +17,21 @@ func NewFilter(next Logger, options ...Option) Logger {
|
||||||
return l
|
return l
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewFilterByLevel wraps next and implements filtering based on a given level.
|
// AllowLevel returns an option for the given level or error if no option exist
|
||||||
// Error is returned if level is not info, error or debug.
|
// for such level.
|
||||||
func NewFilterByLevel(next Logger, lvl string) (Logger, error) {
|
func AllowLevel(lvl string) (Option, error) {
|
||||||
var option Option
|
|
||||||
switch lvl {
|
switch lvl {
|
||||||
case "info":
|
|
||||||
option = AllowInfo()
|
|
||||||
case "debug":
|
case "debug":
|
||||||
option = AllowDebug()
|
return AllowDebug(), nil
|
||||||
|
case "info":
|
||||||
|
return AllowInfo(), nil
|
||||||
case "error":
|
case "error":
|
||||||
option = AllowError()
|
return AllowError(), nil
|
||||||
|
case "none":
|
||||||
|
return AllowNone(), nil
|
||||||
default:
|
default:
|
||||||
return nil, fmt.Errorf("Expected either \"info\", \"debug\" or \"error\" log level, given %v", lvl)
|
return nil, fmt.Errorf("Expected either \"info\", \"debug\", \"error\" or \"none\" level, given %s", lvl)
|
||||||
}
|
}
|
||||||
return NewFilter(next, option), nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type filter struct {
|
type filter struct {
|
||||||
|
|
Loading…
Reference in New Issue