[log] change helper func

This commit is contained in:
Anton Kaliaev 2017-05-16 11:51:29 +02:00
parent 761b1553aa
commit d0cae7b6ed
No known key found for this signature in database
GPG Key ID: 7B6881D965918214
1 changed files with 10 additions and 10 deletions

View File

@ -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 {