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
|
package log
import (
"os"
"github.com/sirupsen/logrus"
)
const (
// GitalyLogDirEnvKey defines the environment variable used to specify the Gitaly log directory
GitalyLogDirEnvKey = "GITALY_LOG_DIR"
// LogTimestampFormat defines the timestamp format in log files
LogTimestampFormat = "2006-01-02T15:04:05.000Z"
)
var (
defaultLogger = logrus.StandardLogger()
grpcGo = logrus.New()
// Loggers is convenient when you want to apply configuration to all
// loggers
Loggers = []*logrus.Logger{defaultLogger, grpcGo}
)
func init() {
// This ensures that any log statements that occur before
// the configuration has been loaded will be written to
// stdout instead of stderr
for _, l := range Loggers {
l.Out = os.Stdout
}
}
// Configure sets the format and level on all loggers. It applies level
// mapping to the GrpcGo logger.
func Configure(loggers []*logrus.Logger, format string, level string) {
var formatter logrus.Formatter
switch format {
case "json":
formatter = &logrus.JSONFormatter{TimestampFormat: LogTimestampFormat}
case "text":
formatter = &logrus.TextFormatter{TimestampFormat: LogTimestampFormat}
case "":
// Just stick with the default
default:
logrus.WithField("format", format).Fatal("invalid logger format")
}
logrusLevel, err := logrus.ParseLevel(level)
if err != nil {
logrusLevel = logrus.InfoLevel
}
for _, l := range loggers {
if l == grpcGo {
l.SetLevel(mapGrpcLogLevel(logrusLevel))
} else {
l.SetLevel(logrusLevel)
}
if formatter != nil {
l.Formatter = formatter
}
}
}
func mapGrpcLogLevel(level logrus.Level) logrus.Level {
// Honor grpc-go's debug settings: https://github.com/grpc/grpc-go#how-to-turn-on-logging
logLevel := os.Getenv("GRPC_GO_LOG_SEVERITY_LEVEL")
if logLevel != "" {
switch logLevel {
case "ERROR", "error":
return logrus.ErrorLevel
case "WARNING", "warning":
return logrus.WarnLevel
case "INFO", "info":
return logrus.InfoLevel
}
}
// grpc-go is too verbose at level 'info'. So when config.toml requests
// level info, we tell grpc-go to log at 'warn' instead.
if level == logrus.InfoLevel {
return logrus.WarnLevel
}
return level
}
// Default is the default logrus logger
func Default() *logrus.Entry { return defaultLogger.WithField("pid", os.Getpid()) }
// GrpcGo is a dedicated logrus logger for the grpc-go library. We use it
// to control the library's chattiness.
func GrpcGo() *logrus.Entry { return grpcGo.WithField("pid", os.Getpid()) }
|