1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
package logger
import (
"github.com/fatih/color"
"github.com/spiral/endure"
"github.com/spiral/roadrunner/v2/plugins/config"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"strings"
"time"
)
// ServiceName declares service name.
const ServiceName = "logs"
type LogFactory interface {
// GlobalLogger returns global log instance.
GlobalLogger() *zap.Logger
// NamedLogger returns logger dedicated to the specific channel. Similar to Named() but also reads the core params.
NamedLogger(name string) *zap.Logger
}
// ZapLogger manages zap logger.
type ZapLogger struct {
base *zap.Logger
cfg Config
}
func (z *ZapLogger) Init(cfg config.Provider) (err error) {
err = cfg.UnmarshalKey(ServiceName, &z.cfg)
if err != nil {
return err
}
if z.base == nil {
cfg := zap.Config{
Level: zap.NewAtomicLevelAt(zap.DebugLevel),
Encoding: "console",
EncoderConfig: zapcore.EncoderConfig{
MessageKey: "message",
LevelKey: "level",
TimeKey: "time",
NameKey: "name",
EncodeName: func(s string, enc zapcore.PrimitiveArrayEncoder) {
if len(s) < 12 {
s = s + strings.Repeat(" ", 12-len(s))
}
enc.AppendString(color.HiGreenString(s))
},
EncodeLevel: func(level zapcore.Level, enc zapcore.PrimitiveArrayEncoder) {
switch level {
case zapcore.DebugLevel:
enc.AppendString(color.HiWhiteString(level.CapitalString()))
case zapcore.InfoLevel:
enc.AppendString(color.HiCyanString(level.CapitalString()))
case zapcore.WarnLevel:
enc.AppendString(color.HiYellowString(level.CapitalString()))
case zapcore.ErrorLevel, zapcore.DPanicLevel:
enc.AppendString(color.HiRedString(level.CapitalString()))
case zapcore.PanicLevel, zapcore.FatalLevel:
enc.AppendString(color.HiMagentaString(level.CapitalString()))
}
},
EncodeTime: func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
enc.AppendString(t.UTC().Format("2006/01/02 15:04:05"))
},
EncodeCaller: zapcore.ShortCallerEncoder,
},
OutputPaths: []string{"stderr"},
ErrorOutputPaths: []string{"stderr"},
}
z.base, err = cfg.Build()
if err != nil {
return err
}
}
return nil
}
// GlobalLogger returns global log instance.
func (z *ZapLogger) GlobalLogger() *zap.Logger {
return z.base
}
// NamedLogger returns logger dedicated to the specific channel. Similar to Named() but also reads the core params.
func (z *ZapLogger) NamedLogger(name string) *zap.Logger {
// todo: automatically configure
return z.base.Named(name)
}
// Provides declares factory methods.
func (z *ZapLogger) Provides() []interface{} {
return []interface{}{
z.DefaultLogger,
z.AllocateLogger,
}
}
// AllocateLogger allocates logger for the service.
func (z *ZapLogger) AllocateLogger(n endure.Named) (*zap.Logger, error) {
return z.NamedLogger(n.Name()), nil
}
// DefaultLogger returns default logger.
func (z *ZapLogger) DefaultLogger() (*zap.Logger, error) {
return z.GlobalLogger(), nil
}
|
