blob: 8c1739f2f329b98569534bcc72dd5b4bc44dc348 (
plain)
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
|
package logger
import (
"github.com/spiral/endure"
"github.com/spiral/roadrunner/v2/plugins/config"
"go.uber.org/zap"
)
// 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
channels ChannelConfig
}
// Init logger service.
func (z *ZapLogger) Init(cfg config.Provider) (err error) {
err = cfg.UnmarshalKey(ServiceName, &z.cfg)
if err != nil {
return err
}
err = cfg.UnmarshalKey(ServiceName, &z.channels)
if err != nil {
return err
}
z.base, err = z.cfg.BuildLogger()
return err
}
// DefaultLogger returns default logger.
func (z *ZapLogger) DefaultLogger() (*zap.Logger, error) {
return z.base, nil
}
// 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, error) {
if cfg, ok := z.channels.Channels[name]; ok {
return cfg.BuildLogger()
}
return z.base.Named(name), nil
}
// NamedLogger returns logger dedicated to the specific channel. Similar to Named() but also reads the core params.
func (z *ZapLogger) ServiceLogger(n endure.Named) (*zap.Logger, error) {
return z.NamedLogger(n.Name())
}
// Provides declares factory methods.
func (z *ZapLogger) Provides() []interface{} {
return []interface{}{
z.DefaultLogger,
z.ServiceLogger,
}
}
|
