Files
vuls/pkg/log/log.go
2022-11-15 11:26:26 +09:00

23 lines
447 B
Go

package log
import (
"context"
"go.uber.org/zap"
)
type ctxLogger struct{}
// ContextWithLogger adds logger to context
func ContextWithLogger(ctx context.Context, l *zap.Logger) context.Context {
return context.WithValue(ctx, ctxLogger{}, l)
}
// LoggerFromContext returns logger from context
func LoggerFromContext(ctx context.Context) *zap.Logger {
if l, ok := ctx.Value(ctxLogger{}).(*zap.Logger); ok {
return l
}
return zap.L()
}