snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/pkg/logger/context.go

@@ -0,0 +1,44 @@
+package logger
+
+import (
+	"context"
+	"github.com/go-chi/chi/v5/middleware"
+	"go.uber.org/zap"
+)
+
+type (
+	ctxLogKey struct{}
+)
+
+// ContextWithValue allows us to pack custom logger to context and pass that to the
+func ContextWithValue(ctx context.Context, logger *zap.Logger) context.Context {
+	return context.WithValue(ctx, ctxLogKey{}, logger)
+}
+
+// ContextValue retrieves logger from given context or falls back to
+// any of the logger passed to it. If no loggers are found it uses default logger from pkg/logger
+func ContextValue(ctx context.Context, fallbacks ...*zap.Logger) *zap.Logger {
+	if ctx != nil {
+		if ctxLogger := ctx.Value(ctxLogKey{}); ctxLogger != nil {
+			// This will panic if we somehow manage to set
+			return ctxLogger.(*zap.Logger)
+		}
+	}
+
+	for _, l := range fallbacks {
+		if l != nil {
+			return l
+		}
+	}
+
+	return Default()
+}
+
+// AddRequestID sets requestID field from context to logger and returns it
+func AddRequestID(ctx context.Context, logger *zap.Logger) *zap.Logger {
+	if reqID := middleware.GetReqID(ctx); reqID != "" {
+		logger = logger.With(zap.String("requestID", reqID))
+	}
+
+	return logger
+}