argoyle
aaa111dd20
Add a context with timeout to handle graceful shutdown of the HTTP server. Update error handling during the server's close to include context-aware shutdown. Ensure that the server properly logs only non-closed errors when listening.
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package logging
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"os"
|
|
)
|
|
|
|
type Logger interface {
|
|
Info(msg string, args ...any)
|
|
Warn(msg string, args ...any)
|
|
Error(msg string, args ...any)
|
|
}
|
|
|
|
var defaultLogger *slog.Logger
|
|
|
|
type contextKey string
|
|
|
|
const loggerKey = contextKey("logger")
|
|
|
|
func SetupLogger(logLevel, serviceName, buildVersion string) *slog.Logger {
|
|
var leveler slog.LevelVar
|
|
|
|
err := leveler.UnmarshalText([]byte(logLevel))
|
|
|
|
defaultLogger = slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
|
|
AddSource: false,
|
|
Level: leveler.Level(),
|
|
ReplaceAttr: nil,
|
|
})).With("service", serviceName).With("version", buildVersion)
|
|
|
|
if err != nil {
|
|
defaultLogger.With("err", err).Error("Failed to parse log level")
|
|
os.Exit(1)
|
|
}
|
|
slog.SetDefault(defaultLogger)
|
|
return defaultLogger
|
|
}
|
|
|
|
// ContextWithLogger returns a new Context with the logger attached
|
|
func ContextWithLogger(ctx context.Context, logger *slog.Logger) context.Context {
|
|
return context.WithValue(ctx, loggerKey, logger)
|
|
}
|
|
|
|
// LoggerFromContext returns a logger from the passed context or the default logger
|
|
func LoggerFromContext(ctx context.Context) *slog.Logger {
|
|
logger := ctx.Value(loggerKey)
|
|
if l, ok := logger.(*slog.Logger); ok {
|
|
return l
|
|
}
|
|
return defaultLogger
|
|
}
|