68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package zero
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/micro/go-micro/v2/logger"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type formatterKey struct{}
|
|
type levelKey struct{}
|
|
type levelFieldKey struct{}
|
|
type outKey struct{}
|
|
type hooksKey struct{}
|
|
type reportCallerKey struct{}
|
|
type exitKey struct{}
|
|
type prettyKey struct{}
|
|
type useColorKey struct{}
|
|
|
|
type Options struct {
|
|
logger.Options
|
|
}
|
|
|
|
func WithTimeFieldFormat(formatter zerolog.Formatter) logger.Option {
|
|
return setOption(formatterKey{}, formatter)
|
|
}
|
|
|
|
func WithLevelFieldName(levelFieldName string) logger.Option {
|
|
return setOption(levelFieldKey{}, levelFieldName)
|
|
}
|
|
|
|
func WithPretty(pretty bool) logger.Option {
|
|
return setOption(prettyKey{}, pretty)
|
|
}
|
|
func WithColor(useColor bool) logger.Option {
|
|
return setOption(useColorKey{}, useColor)
|
|
}
|
|
|
|
func WithLevel(lvl logger.Level) logger.Option {
|
|
return setOption(levelKey{}, lvl)
|
|
}
|
|
|
|
func WithOut(out io.Writer) logger.Option {
|
|
return setOption(outKey{}, out)
|
|
}
|
|
|
|
func WithHooks(hooks []zerolog.Hook) logger.Option {
|
|
return setOption(hooksKey{}, hooks)
|
|
}
|
|
|
|
func WithReportCaller(reportCaller bool) logger.Option {
|
|
return setOption(reportCallerKey{}, reportCaller)
|
|
}
|
|
|
|
func WithExitFunc(exit func(int)) logger.Option {
|
|
return setOption(exitKey{}, exit)
|
|
}
|
|
|
|
func setOption(k, v interface{}) logger.Option {
|
|
return func(o *logger.Options) {
|
|
if o.Context == nil {
|
|
o.Context = context.Background()
|
|
}
|
|
o.Context = context.WithValue(o.Context, k, v)
|
|
}
|
|
}
|