81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package logger
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
type Option func(*Options)
|
|
|
|
type Options struct {
|
|
// The logging level the logger should log at. default is `InfoLevel`
|
|
Level Level
|
|
// fields to always be logged
|
|
Fields map[string]interface{}
|
|
// It's common to set this to a file, or leave it default which is `os.Stderr`
|
|
Out io.Writer
|
|
// Caller skip frame count for file:line info
|
|
CallerSkipCount int
|
|
// Alternative options
|
|
Context context.Context
|
|
}
|
|
|
|
func NewOptions(opts ...Option) Options {
|
|
options := Options{
|
|
Level: InfoLevel,
|
|
Fields: make(map[string]interface{}),
|
|
Out: os.Stderr,
|
|
CallerSkipCount: 2,
|
|
Context: context.Background(),
|
|
}
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
return options
|
|
}
|
|
|
|
// WithFields set default fields for the logger
|
|
func WithFields(fields map[string]interface{}) Option {
|
|
return func(args *Options) {
|
|
args.Fields = fields
|
|
}
|
|
}
|
|
|
|
// WithLevel set default level for the logger
|
|
func WithLevel(level Level) Option {
|
|
return func(args *Options) {
|
|
args.Level = level
|
|
}
|
|
}
|
|
|
|
// WithOutput set default output writer for the logger
|
|
func WithOutput(out io.Writer) Option {
|
|
return func(args *Options) {
|
|
args.Out = out
|
|
}
|
|
}
|
|
|
|
// WithCallerSkipCount set frame count to skip
|
|
func WithCallerSkipCount(c int) Option {
|
|
return func(args *Options) {
|
|
args.CallerSkipCount = c
|
|
}
|
|
}
|
|
|
|
// WithContext set context
|
|
func WithContext(ctx context.Context) Option {
|
|
return func(args *Options) {
|
|
args.Context = ctx
|
|
}
|
|
}
|
|
|
|
func SetOption(k, v interface{}) Option {
|
|
return func(o *Options) {
|
|
if o.Context == nil {
|
|
o.Context = context.Background()
|
|
}
|
|
o.Context = context.WithValue(o.Context, k, v)
|
|
}
|
|
}
|