-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
86 lines (71 loc) · 1.65 KB
/
main.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
80
81
82
83
84
85
86
package main
import (
"flag"
"os"
"time"
mytools "zapexample/tools"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
type LogLevel int
const (
Warn LogLevel = iota
Info
Debug
Error
)
const dateTimeFormat = "[02/01/2006 15:04:05]"
const LOG_FILE = "log.txt"
var logger *zap.SugaredLogger = nil
func main() {
interval := flag.Int("i", 5, "Greetings Interval")
flag.Parse()
initLogger()
logger.Infow("Zap Package level logging example", "i", *interval)
logger.Debug("Logger initialized")
mytools.SetLogger(
GetLogger(LogLevel(Debug)),
)
logger.Debug("Calling greetings from main")
mytools.SayGreetings(*interval)
}
func initLogger() {
// Initializing with default level for package main
logger = GetLogger(LogLevel(Info))
}
func GetLogger(level LogLevel) *zap.SugaredLogger {
// Configure logger options
var l zapcore.Level
config := zap.NewProductionEncoderConfig()
config.EncodeTime = func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
enc.AppendString(t.Format(dateTimeFormat))
}
logFile, errLogFile := os.OpenFile(LOG_FILE, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
if errLogFile != nil {
defer logFile.Close()
panic("Failed to open log file " + LOG_FILE)
}
// Set the initial logging level
switch level {
case Warn:
l = zap.WarnLevel
case Debug:
l = zap.DebugLevel
case Error:
l = zap.ErrorLevel
default:
if logger == nil {
l = zap.InfoLevel
} else {
return logger
}
}
var err error
_logger := zap.New(zapcore.NewCore(zapcore.NewConsoleEncoder(config),
zapcore.AddSync(logFile), l)).Sugar()
if err != nil {
panic("failed to initialize logger: " + err.Error())
}
// defer _logger.Sync()
return _logger
}