diff --git a/ctx.go b/ctx.go index c6a019c..2c85a8c 100644 --- a/ctx.go +++ b/ctx.go @@ -2,16 +2,20 @@ package logger import "context" -var contextKey struct{} +type contextKey int + +const ( + loggerKey contextKey = iota +) // InjectInContext will inject a logger into the go-context func InjectInContext(ctx context.Context, l LoggerInterface) context.Context { - return context.WithValue(ctx, contextKey, l) + return context.WithValue(ctx, loggerKey, l) } // FromContext will retrieve a logger from the go-context or return defaultLogger func FromContext(ctx context.Context, defaultLogger LoggerInterface) LoggerInterface { - if _logger, ok := ctx.Value(contextKey).(LoggerInterface); ok { + if _logger, ok := ctx.Value(loggerKey).(LoggerInterface); ok { return _logger } return defaultLogger