package main

import (
    "context"
    "log/slog"

    "github.com/google/uuid"
    "github.com/prathoss/logenricher"
)

func CorrelationIDExtractor(ctx context.Context) []slog.Attr {
    correlationID := ctx.Value("correlation-id").(uuid.UUID)
    return []slog.Attr{slog.String("correlation_id", correlationID.String())}
}

func main() {
    slog.SetDefault(slog.New(&logenricher.SlogHandlerWrapper{
        Handler: slog.NewTextHandler(
            os.Stdout,
            &slog.HandlerOptions{},
        ),
        Extractors: []logenricher.Extractor{
            CorrelationIDExtractor,
        },
    }))
}