Marc Brugger
119 lines
2.3 KiB
Go
119 lines
2.3 KiB
Go
package log
|
|
|
|
import (
|
|
"os"
|
|
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
const (
|
|
logHistorySize = 50
|
|
envLogLevel = "LOG_LEVEL"
|
|
envLogFormat = "LOG_FORMAT"
|
|
)
|
|
|
|
var (
|
|
rootLogger *zap.Logger
|
|
logs []string
|
|
)
|
|
|
|
// GetLogger returns a named logger.
|
|
func GetLogger(name string) *zap.SugaredLogger {
|
|
return rootLogger.Named(name).Sugar()
|
|
}
|
|
|
|
func init() {
|
|
level := zap.InfoLevel
|
|
|
|
if lvl, ok := os.LookupEnv(envLogLevel); ok {
|
|
if err := level.Set(lvl); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
format := "console"
|
|
if fmt, ok := os.LookupEnv(envLogFormat); ok {
|
|
format = fmt
|
|
}
|
|
|
|
cfg := zap.Config{
|
|
Level: zap.NewAtomicLevelAt(level),
|
|
Development: false,
|
|
Encoding: format,
|
|
EncoderConfig: zap.NewProductionEncoderConfig(),
|
|
OutputPaths: []string{"stdout"},
|
|
ErrorOutputPaths: []string{"stderr"},
|
|
}
|
|
cfg.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
|
|
cfg.EncoderConfig.EncodeDuration = zapcore.StringDurationEncoder
|
|
cfg.EncoderConfig.EncodeLevel = zapcore.CapitalLevelEncoder
|
|
|
|
opt := zap.WrapCore(func(c zapcore.Core) zapcore.Core {
|
|
return zapcore.NewTee(c, &logList{
|
|
enc: zapcore.NewConsoleEncoder(cfg.EncoderConfig),
|
|
LevelEnabler: cfg.Level,
|
|
})
|
|
})
|
|
|
|
rootLogger, _ = cfg.Build(opt)
|
|
}
|
|
|
|
type logList struct {
|
|
zapcore.LevelEnabler
|
|
enc zapcore.Encoder
|
|
}
|
|
|
|
func (l *logList) clone() *logList {
|
|
return &logList{
|
|
LevelEnabler: l.LevelEnabler,
|
|
enc: l.enc.Clone(),
|
|
}
|
|
}
|
|
|
|
func (l *logList) With(fields []zapcore.Field) zapcore.Core {
|
|
clone := l.clone()
|
|
addFields(clone.enc, fields)
|
|
return clone
|
|
}
|
|
|
|
func (l *logList) Check(ent zapcore.Entry, ce *zapcore.CheckedEntry) *zapcore.CheckedEntry {
|
|
if l.Enabled(ent.Level) {
|
|
return ce.AddCore(ent, l)
|
|
}
|
|
return ce
|
|
}
|
|
|
|
func (l *logList) Write(ent zapcore.Entry, fields []zapcore.Field) error {
|
|
buf, err := l.enc.EncodeEntry(ent, fields)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
logs = append(logs, buf.String())
|
|
|
|
if len(logs) > logHistorySize {
|
|
logs = logs[len(logs)-logHistorySize:]
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (l *logList) Sync() error {
|
|
return nil
|
|
}
|
|
|
|
// Logs get the current logs.
|
|
func Logs() []string {
|
|
return logs
|
|
}
|
|
|
|
// Clear the current logs.
|
|
func Clear() {
|
|
logs = nil
|
|
}
|
|
|
|
func addFields(enc zapcore.ObjectEncoder, fields []zapcore.Field) {
|
|
for i := range fields {
|
|
fields[i].AddTo(enc)
|
|
}
|
|
}
|