This repository has been archived by the owner on Jul 11, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 279
/
logger.go
79 lines (61 loc) · 2.04 KB
/
logger.go
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
// Package logger implements utility routines to initialize the logging facility used by OSM components.
package logger
import (
"fmt"
"os"
"path"
"runtime"
"strings"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"github.com/openservicemesh/osm/pkg/constants"
)
// AllowedLevels is the list of allowed log levels for OSM Controller.
var AllowedLevels = []string{"debug", "info", "warn", "error", "fatal", "panic", "disabled", "trace"}
// CallerHook implements zerolog.Hook interface.
type CallerHook struct{}
// Run adds additional context
func (h CallerHook) Run(e *zerolog.Event, _ zerolog.Level, _ string) {
if _, file, line, ok := runtime.Caller(3); ok {
e.Str("file", fmt.Sprintf("%s:%d", path.Base(file), line))
}
}
func newLogger(component string) zerolog.Logger {
return log.With().Str("component", component).Logger().Hook(CallerHook{})
}
// New creates a new zerolog.Logger
func New(component string) zerolog.Logger {
if os.Getenv(constants.EnvVarHumanReadableLogMessages) == "true" {
return NewPretty(component)
}
return newLogger(component)
}
// NewPretty creates a new zerolog.Logger, which emits human-readable log messages
func NewPretty(component string) zerolog.Logger {
return newLogger(component).Output(zerolog.ConsoleWriter{Out: os.Stdout})
}
// SetLogLevel sets the global logging level
func SetLogLevel(verbosity string) error {
switch strings.ToLower(verbosity) {
// DebugLevel defines debug log level.
case "debug":
zerolog.SetGlobalLevel(zerolog.DebugLevel)
case "info":
zerolog.SetGlobalLevel(zerolog.InfoLevel)
case "warn":
zerolog.SetGlobalLevel(zerolog.WarnLevel)
case "error":
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
case "fatal":
zerolog.SetGlobalLevel(zerolog.FatalLevel)
case "panic":
zerolog.SetGlobalLevel(zerolog.PanicLevel)
case "disabled":
zerolog.SetGlobalLevel(zerolog.Disabled)
case "trace":
zerolog.SetGlobalLevel(zerolog.TraceLevel)
default:
return fmt.Errorf("Invalid log level '%s' specified. Please specify one of %v", verbosity, AllowedLevels)
}
return nil
}