Skip to content

Commit 61d8db3

Browse files
committed
allow set logger for log function
1 parent ed8500d commit 61d8db3

1 file changed

Lines changed: 14 additions & 4 deletions

File tree

log/log.go

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,19 +31,29 @@ func Setup(config Config) {
3131
}
3232
}
3333

34-
// Build is a panic-free version of Setup.
35-
func Build(config Config) error {
34+
// Set logger for global log fuctions
35+
func Set(logger *zap.Logger) {
36+
log.Store(logger.Sugar())
37+
}
38+
39+
// NewLogger creates logger
40+
func NewLogger(config Config) (*zap.Logger, error) {
3641
var cfg zap.Config
3742
if config.Debug {
3843
cfg = zap.NewDevelopmentConfig()
3944
} else {
4045
cfg = zap.NewProductionConfig()
4146
}
42-
logger, err := cfg.Build()
47+
return cfg.Build()
48+
}
49+
50+
// Build is a panic-free version of Setup.
51+
func Build(config Config) error {
52+
logger, err := NewLogger(config)
4353
if err != nil {
4454
return fmt.Errorf("logger creation failed: %w", err)
4555
}
46-
log.Store(logger.Sugar())
56+
Set(logger)
4757
return nil
4858
}
4959

0 commit comments

Comments
 (0)