logging update with context

internal/logging/logging.go

@@ -0,0 +1,26 @@
+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
+}