refactor: a better DI-friendy logger implementation that doesn't suck
This commit is contained in:
@@ -8,40 +8,36 @@ import (
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
type Logger interface {
|
||||
Get() *zap.Logger
|
||||
Sync() error
|
||||
}
|
||||
|
||||
type loggerImpl struct {
|
||||
}
|
||||
|
||||
func NewLogger() Logger {
|
||||
return &loggerImpl{}
|
||||
}
|
||||
|
||||
var (
|
||||
logger *zap.Logger
|
||||
once sync.Once
|
||||
instance *zap.Logger
|
||||
once sync.Once
|
||||
)
|
||||
|
||||
func (l *loggerImpl) Get() *zap.Logger {
|
||||
func NewLogger() *zap.Logger {
|
||||
once.Do(func() {
|
||||
var err error
|
||||
cfg := config.GetConfig()
|
||||
var err error
|
||||
|
||||
if cfg.Environment == "production" {
|
||||
logger, err = zap.NewProduction()
|
||||
instance, err = zap.NewProduction()
|
||||
} else {
|
||||
logger, err = zap.NewDevelopment()
|
||||
instance, err = zap.NewDevelopment()
|
||||
}
|
||||
if err != nil {
|
||||
panic(err)
|
||||
panic("failed to initialize logger: " + err.Error())
|
||||
}
|
||||
})
|
||||
return logger
|
||||
return instance
|
||||
}
|
||||
|
||||
func (l *loggerImpl) Sync() error {
|
||||
return logger.Sync()
|
||||
type SyncLogger struct {
|
||||
*zap.Logger
|
||||
}
|
||||
|
||||
func NewSyncLogger(logger *zap.Logger) *SyncLogger {
|
||||
return &SyncLogger{logger}
|
||||
}
|
||||
|
||||
func (s *SyncLogger) Close() error {
|
||||
return s.Sync()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user