logger: conditional caller field
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
@@ -8,6 +8,7 @@ import (
|
|||||||
"slices"
|
"slices"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"go.unistack.org/micro/v4/logger"
|
||||||
"go.unistack.org/micro/v4/meter"
|
"go.unistack.org/micro/v4/meter"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -42,8 +43,10 @@ type Options struct {
|
|||||||
Fields []interface{}
|
Fields []interface{}
|
||||||
// ContextAttrFuncs contains funcs that executed before log func on context
|
// ContextAttrFuncs contains funcs that executed before log func on context
|
||||||
ContextAttrFuncs []ContextAttrFunc
|
ContextAttrFuncs []ContextAttrFunc
|
||||||
// callerSkipCount number of frmaes to skip
|
// callerSkipCount number of frames to skip
|
||||||
CallerSkipCount int
|
CallerSkipCount int
|
||||||
|
// AddCaller enables to get caller
|
||||||
|
AddCaller bool
|
||||||
// The logging level the logger should log
|
// The logging level the logger should log
|
||||||
Level Level
|
Level Level
|
||||||
// AddSource enabled writing source file and position in log
|
// AddSource enabled writing source file and position in log
|
||||||
@@ -83,6 +86,12 @@ func NewOptions(opts ...Option) Options {
|
|||||||
return options
|
return options
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func WithCallerEnabled(b bool) logger.Option {
|
||||||
|
return func(o *Options) {
|
||||||
|
o.AddCaller = b
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// WithFatalFinalizers set logger.Fatal finalizers
|
// WithFatalFinalizers set logger.Fatal finalizers
|
||||||
func WithFatalFinalizers(fncs ...func(context.Context)) Option {
|
func WithFatalFinalizers(fncs ...func(context.Context)) Option {
|
||||||
return func(o *Options) {
|
return func(o *Options) {
|
||||||
|
|||||||
@@ -37,11 +37,11 @@ var (
|
|||||||
|
|
||||||
type wrapper struct {
|
type wrapper struct {
|
||||||
h slog.Handler
|
h slog.Handler
|
||||||
level atomic.Int64
|
level int64
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *wrapper) Enabled(ctx context.Context, level slog.Level) bool {
|
func (h *wrapper) Enabled(ctx context.Context, level slog.Level) bool {
|
||||||
return level >= slog.Level(int(h.level.Load()))
|
return level >= slog.Level(atomic.LoadInt64(&h.level))
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *wrapper) Handle(ctx context.Context, rec slog.Record) error {
|
func (h *wrapper) Handle(ctx context.Context, rec slog.Record) error {
|
||||||
@@ -49,11 +49,17 @@ func (h *wrapper) Handle(ctx context.Context, rec slog.Record) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (h *wrapper) WithAttrs(attrs []slog.Attr) slog.Handler {
|
func (h *wrapper) WithAttrs(attrs []slog.Attr) slog.Handler {
|
||||||
return h.h.WithAttrs(attrs)
|
return &wrapper{
|
||||||
|
h: h.h.WithAttrs(attrs),
|
||||||
|
level: atomic.LoadInt64(&h.level),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *wrapper) WithGroup(name string) slog.Handler {
|
func (h *wrapper) WithGroup(name string) slog.Handler {
|
||||||
return h.h.WithGroup(name)
|
return &wrapper{
|
||||||
|
h: h.h.WithGroup(name),
|
||||||
|
level: atomic.LoadInt64(&h.level),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr {
|
func (s *slogLogger) renameAttr(_ []string, a slog.Attr) slog.Attr {
|
||||||
@@ -115,10 +121,13 @@ func (s *slogLogger) Clone(opts ...logger.Option) logger.Logger {
|
|||||||
|
|
||||||
attrs, _ := s.argsAttrs(options.Fields)
|
attrs, _ := s.argsAttrs(options.Fields)
|
||||||
l := &slogLogger{
|
l := &slogLogger{
|
||||||
handler: &wrapper{h: s.handler.h.WithAttrs(attrs)},
|
handler: &wrapper{
|
||||||
opts: options,
|
h: s.handler.h.WithAttrs(attrs),
|
||||||
|
level: atomic.LoadInt64(&s.handler.level),
|
||||||
|
},
|
||||||
|
opts: options,
|
||||||
}
|
}
|
||||||
l.handler.level.Store(int64(loggerToSlogLevel(options.Level)))
|
atomic.StoreInt64(&l.handler.level, int64(loggerToSlogLevel(options.Level)))
|
||||||
|
|
||||||
return l
|
return l
|
||||||
}
|
}
|
||||||
@@ -131,9 +140,9 @@ func (s *slogLogger) V(level logger.Level) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *slogLogger) Level(level logger.Level) {
|
func (s *slogLogger) Level(level logger.Level) {
|
||||||
|
atomic.StoreInt64(&s.handler.level, int64(loggerToSlogLevel(level)))
|
||||||
s.mu.Lock()
|
s.mu.Lock()
|
||||||
s.opts.Level = level
|
s.opts.Level = level
|
||||||
s.handler.level.Store(int64(loggerToSlogLevel(level)))
|
|
||||||
s.mu.Unlock()
|
s.mu.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -154,8 +163,11 @@ func (s *slogLogger) Fields(fields ...interface{}) logger.Logger {
|
|||||||
}
|
}
|
||||||
|
|
||||||
attrs, _ := s.argsAttrs(fields)
|
attrs, _ := s.argsAttrs(fields)
|
||||||
l.handler = &wrapper{h: s.handler.h.WithAttrs(attrs)}
|
l.handler = &wrapper{
|
||||||
l.handler.level.Store(int64(loggerToSlogLevel(l.opts.Level)))
|
h: s.handler.h.WithAttrs(attrs),
|
||||||
|
level: atomic.LoadInt64(&s.handler.level),
|
||||||
|
}
|
||||||
|
atomic.StoreInt64(&l.handler.level, int64(loggerToSlogLevel(l.opts.Level)))
|
||||||
|
|
||||||
return l
|
return l
|
||||||
}
|
}
|
||||||
@@ -200,8 +212,11 @@ func (s *slogLogger) Init(opts ...logger.Option) error {
|
|||||||
h = slog.NewJSONHandler(s.opts.Out, handleOpt)
|
h = slog.NewJSONHandler(s.opts.Out, handleOpt)
|
||||||
}
|
}
|
||||||
|
|
||||||
s.handler = &wrapper{h: h.WithAttrs(attrs)}
|
s.handler = &wrapper{
|
||||||
s.handler.level.Store(int64(loggerToSlogLevel(s.opts.Level)))
|
h: h.WithAttrs(attrs),
|
||||||
|
level: atomic.LoadInt64(&s.handler.level),
|
||||||
|
}
|
||||||
|
atomic.StoreInt64(&s.handler.level, int64(loggerToSlogLevel(s.opts.Level)))
|
||||||
s.mu.Unlock()
|
s.mu.Unlock()
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
@@ -290,10 +305,17 @@ func (s *slogLogger) printLog(ctx context.Context, lvl logger.Level, msg string,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var pcs [1]uintptr
|
var pcs uintptr
|
||||||
runtime.Callers(s.opts.CallerSkipCount, pcs[:]) // skip [Callers, printLog, LogLvlMethod]
|
|
||||||
r := slog.NewRecord(s.opts.TimeFunc(), loggerToSlogLevel(lvl), msg, pcs[0])
|
if s.opts.AddCaller {
|
||||||
|
var caller [1]uintptr
|
||||||
|
runtime.Callers(s.opts.CallerSkipCount, caller[:]) // skip [Callers, printLog, LogLvlMethod]
|
||||||
|
pcs = caller[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
r := slog.NewRecord(s.opts.TimeFunc(), loggerToSlogLevel(lvl), msg, pcs)
|
||||||
r.AddAttrs(attrs...)
|
r.AddAttrs(attrs...)
|
||||||
|
|
||||||
_ = s.handler.Handle(ctx, r)
|
_ = s.handler.Handle(ctx, r)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user