frostfs-node/pkg/util/logger/test/logger.go

37 lines
659 B
Go
Raw Normal View History

package test
import (
2020-07-24 13:54:03 +00:00
"github.com/nspcc-dev/neofs-node/pkg/util/logger"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
const sampling = 1000
2020-07-24 13:54:03 +00:00
// NewLogger creates a new logger.
//
// If debug, development logger is created.
func NewLogger(debug bool) *logger.Logger {
var l logger.Logger
l.Logger = zap.L()
if debug {
cfg := zap.NewDevelopmentConfig()
cfg.Sampling = &zap.SamplingConfig{
Initial: sampling,
Thereafter: sampling,
}
cfg.EncoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
log, err := cfg.Build()
if err != nil {
panic("could not prepare logger: " + err.Error())
}
l.Logger = log
}
return &l
}