diff --git a/logger/options.go b/logger/options.go index 791de0e3..8bc045de 100644 --- a/logger/options.go +++ b/logger/options.go @@ -63,6 +63,13 @@ func WithCallerSkipCount(c int) Option { } } +// 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 {