gologger.go
3.2 KB
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
package gologger
import (
"fmt"
"os"
"time"
"sync"
)
var muEv = &sync.Mutex{}
var muEr = &sync.Mutex{}
const MaxLogFileSize5MB int64 = 5*1024*1024
const MaxLogFileSize1MB int64 = 1*1024*1024
const MaxLogFileSize500KB int64 = 500*1024
const MaxLogFileSize100KB int64 = 100*1024
const MaxLogFileSize512B int64 = 512
const ErrDirName = "error-logs"
const EvtDirName = "event-logs"
type Logger struct {
eventf *os.File
eventFileName string
errorf *os.File
errorFileName string
}
func NewLogger(eventFileName, errorFileName string) (*Logger, error) {
var l Logger
var err error
timestamp := "_" + time.Now().Format(time.RFC3339)
// create files
l.eventFileName = eventFileName
l.errorFileName = errorFileName
l.eventf, err = os.Create("./" + EvtDirName + "/" +
eventFileName + timestamp + ".txt")
if err != nil {
fmt.Printf("logger: new: couldn't create event log file\n")
return &l, err
}
l.errorf, err = os.Create("./" + ErrDirName + "/" +
errorFileName + timestamp + ".txt")
if err != nil {
fmt.Printf("logger: new: couldn't create error log file\n")
return &l, err
}
// create directories to keep log files
err = os.Mkdir(EvtDirName, os.ModePerm)
if err != nil {
if !os.IsExist(err) {
fmt.Printf("logger: mkdir: couldn't create event log directory\n")
return &l, err
}
}
err = os.Mkdir(ErrDirName, os.ModePerm)
if err != nil {
if !os.IsExist(err) {
fmt.Printf("logger: new: couldn't create error log directory\n")
return &l, err
}
}
return &l, nil
}
func (l *Logger) Close() {
// close files
err := l.eventf.Close()
if err != nil {
fmt.Printf("logger: on exit: couldn't close event log file\n")
}
err = l.errorf.Close()
if err != nil {
fmt.Printf("logger: on exit: couldn't close error log file\n")
}
}
func (l *Logger) splitEventLog() {
timestamp := "_" + time.Now().Format(time.RFC3339)
evfstats, _ := l.eventf.Stat()
if evfstats.Size() >= MaxLogFileSize100KB {
// close old file
err := l.eventf.Close()
if err != nil {
fmt.Printf("logger: split: couldn't close event file\n")
return
}
// open new file
var errnew error
l.eventf, errnew = os.Create("./" + EvtDirName + "/" +
l.eventFileName + timestamp + ".txt")
if errnew != nil {
fmt.Printf("logger: split: couldn't create event log file\n")
}
}
}
func (l *Logger) splitErrorLog() {
timestamp := "_" + time.Now().Format(time.RFC3339)
erfstats, _ := l.errorf.Stat()
if erfstats.Size() >= MaxLogFileSize100KB {
// close old file
err := l.errorf.Close()
if err != nil {
fmt.Printf("logger: split: couldn't close error file\n")
return
}
// open new file
var errnew error
l.errorf, errnew = os.Create("./" + ErrDirName + "/" +
l.errorFileName + timestamp + ".txt")
if errnew != nil {
fmt.Printf("logger: split: couldn't create error log file\n")
}
}
}
func (l *Logger) LogEvent(event string) {
muEv.Lock()
defer muEv.Unlock()
l.eventf.WriteString(time.Now().Format(time.RFC3339) + ": " + event + "\n")
l.splitEventLog()
}
func (l *Logger) LogError(comment string, err error) {
muEr.Lock()
defer muEr.Unlock()
l.errorf.WriteString(time.Now().Format(time.RFC3339) + ": " + err.Error() + " ::::: " + comment + "\n")
l.splitErrorLog()
}