forked from hbdmapi/huobi_futures_Golang
/
performance_logger.go
76 lines (63 loc) · 1.36 KB
/
performance_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
package log
import (
"fmt"
"log"
"os"
"strings"
"time"
)
var performanceLogger *PerformanceLogger
var logEnabled = false
type PerformanceLogger struct {
logger *log.Logger
enable bool
file *os.File
index int
start time.Time
}
func Enable(enable bool) {
logEnabled = enable
if logEnabled && performanceLogger == nil {
performanceLogger = new(PerformanceLogger).init()
}
}
func GetInstance() *PerformanceLogger {
return performanceLogger
}
func (p *PerformanceLogger) init() *PerformanceLogger {
if logEnabled {
var err error
fileName := time.Now().Format("20060102_150405.txt")
p.file, err = os.OpenFile(fileName, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
if err != nil {
log.Fatalln("Failed to open file: ", fileName)
}
p.logger = log.New(p.file, "", 0)
p.index = 1
}
return p
}
func (p *PerformanceLogger) Start() {
if logEnabled {
p.start = time.Now()
}
}
func (p *PerformanceLogger) StopAndLog(method string, url string) {
if logEnabled {
duration := time.Since(p.start).Milliseconds()
// Strip parameters
i := strings.IndexByte(url, '?')
var path string
if i > 0 {
path = url[0:i]
} else {
path = url
}
// Log the header before first record
if p.index == 1 {
p.logger.Println("Index, Duration(ms), URL")
}
p.logger.Println(fmt.Sprintf("%d, %d, %s %s", p.index, duration, method, path))
p.index++
}
}