65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package logging
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
var (
|
|
loggingConfig = zap.NewProductionConfig()
|
|
)
|
|
|
|
func ConfigureLogging(
|
|
level zap.AtomicLevel,
|
|
developmentLogging bool,
|
|
initialFields map[string]interface{},
|
|
) {
|
|
loggingConfig.Level = level
|
|
loggingConfig.Development = developmentLogging
|
|
if initialFields != nil {
|
|
loggingConfig.InitialFields = initialFields
|
|
}
|
|
}
|
|
|
|
func ParseLevel(levelString string) zap.AtomicLevel {
|
|
switch strings.ToLower(levelString) {
|
|
case "debug":
|
|
return zap.NewAtomicLevelAt(zapcore.DebugLevel)
|
|
case "info":
|
|
return zap.NewAtomicLevelAt(zapcore.InfoLevel)
|
|
case "warn":
|
|
return zap.NewAtomicLevelAt(zapcore.WarnLevel)
|
|
case "error":
|
|
return zap.NewAtomicLevelAt(zapcore.ErrorLevel)
|
|
case "fatal":
|
|
return zap.NewAtomicLevelAt(zapcore.FatalLevel)
|
|
default:
|
|
return zap.NewAtomicLevelAt(zapcore.InfoLevel)
|
|
}
|
|
}
|
|
|
|
func CreateLogger() (Logger, error) {
|
|
if zapLogger, err := loggingConfig.Build(); err != nil {
|
|
return nil, err
|
|
} else {
|
|
return NewLogger(zapLogger), nil
|
|
}
|
|
}
|
|
|
|
func CreateTestLogger(tb testing.TB) Logger {
|
|
return &testLogger{
|
|
tb: tb,
|
|
encoder: zapcore.NewConsoleEncoder(zap.NewDevelopmentEncoderConfig()),
|
|
}
|
|
}
|
|
|
|
func MustCreateLogger() Logger {
|
|
if logger, err := CreateLogger(); err != nil {
|
|
panic(err)
|
|
} else {
|
|
return logger
|
|
}
|
|
}
|