25 lines
517 B
Go
25 lines
517 B
Go
package logger
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type key int
|
|
|
|
const (
|
|
loggerKey key = iota
|
|
)
|
|
|
|
// NewContextWithLogger returns a new context with log added.
|
|
func NewContextWithLogger(ctx context.Context, log *zap.Logger) context.Context {
|
|
return context.WithValue(ctx, loggerKey, log)
|
|
}
|
|
|
|
// LoggerFromContext returns the zap.Logger associated with ctx or nil if no logger has been assigned.
|
|
func LoggerFromContext(ctx context.Context) *zap.Logger {
|
|
l, _ := ctx.Value(loggerKey).(*zap.Logger)
|
|
return l
|
|
}
|