27 lines
576 B
Go
27 lines
576 B
Go
package logging
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"os"
|
|
)
|
|
|
|
type ContextHandler struct {
|
|
slog.Handler
|
|
}
|
|
|
|
func (h *ContextHandler) Handle(ctx context.Context, r slog.Record) error {
|
|
if requestID, ok := ctx.Value("trace_id").(string); ok {
|
|
r.AddAttrs(slog.String("trace_id", requestID))
|
|
}
|
|
return h.Handler.Handle(ctx, r)
|
|
}
|
|
|
|
func NewLogger() *slog.Logger {
|
|
baseHandler := slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{AddSource: false})
|
|
customHandler := &ContextHandler{Handler: baseHandler}
|
|
logger := slog.New(customHandler)
|
|
|
|
return logger
|
|
}
|