-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger.go
81 lines (69 loc) · 1.83 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
80
81
package log
import (
"fmt"
"github.com/aioncore/p2p/pkg/service/utils"
"github.com/go-kit/log"
"github.com/go-kit/log/level"
"io"
"os"
"path/filepath"
"time"
)
var (
debugLogger log.Logger
infoLogger log.Logger
errorLogger log.Logger
)
type Level string
const (
DEBUG Level = "debug"
INFO Level = "info"
ERROR Level = "error"
)
func InitLogger(filePath string, serviceType string) {
debugLogger = log.NewLogfmtLogger(os.Stdout)
debugLogger = level.NewFilter(debugLogger, level.AllowAll())
infoLogFile, err := utils.OpenFile(filepath.Join(filePath, serviceType, "log", "info.log"), os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
if err != nil {
panic(err)
}
infoLogger = log.NewLogfmtLogger(io.MultiWriter(os.Stdout, infoLogFile))
infoLogger = level.NewFilter(infoLogger, level.AllowAll())
errorLogFile, err := utils.OpenFile(filepath.Join(filePath, serviceType, "log", "error.log"), os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
if err != nil {
panic(err)
}
errorLogger = log.NewLogfmtLogger(io.MultiWriter(os.Stdout, errorLogFile))
errorLogger = level.NewFilter(errorLogger, level.AllowAll())
}
func Debug(format string, a ...any) {
message := fmt.Sprintf(format, a...)
logMessage(DEBUG, message)
}
func Info(format string, a ...any) {
message := fmt.Sprintf(format, a...)
logMessage(INFO, message)
}
func Error(format string, a ...any) {
message := fmt.Sprintf(format, a...)
logMessage(ERROR, message)
}
func logMessage(level Level, message string) {
logger := levelLogger(level)
err := logger.Log("time", time.Now(), "message", message)
if err != nil {
panic(err)
}
}
func levelLogger(logLevel Level) log.Logger {
switch logLevel {
case DEBUG:
return level.Debug(debugLogger)
case INFO:
return level.Info(infoLogger)
case ERROR:
return level.Error(errorLogger)
default:
return level.Info(infoLogger)
}
}