29 lines
648 B
Go
29 lines
648 B
Go
package logging
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"os"
|
|
|
|
"gitea.michaelthomson.dev/mthomson/habits/internal/middleware"
|
|
)
|
|
|
|
type ContextHandler struct {
|
|
slog.Handler
|
|
}
|
|
|
|
func (h *ContextHandler) Handle(ctx context.Context, r slog.Record) error {
|
|
if requestID, ok := ctx.Value(middleware.TraceIdKey).(string); ok {
|
|
r.AddAttrs(slog.String(string(middleware.TraceIdKey), requestID))
|
|
}
|
|
return h.Handler.Handle(ctx, r)
|
|
}
|
|
|
|
func NewLogger() *slog.Logger {
|
|
baseHandler := slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{AddSource: true})
|
|
customHandler := &ContextHandler{Handler: baseHandler}
|
|
logger := slog.New(customHandler)
|
|
|
|
return logger
|
|
}
|