操作系统： CentOS 6.9_x64
go语言版本： 1.8.3
问题描述
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
/*
golang log example
*/
package main
import (
"log"
"os"
)
func main() {
logFile,err := os.Create("test1.log")
defer logFile.Close()
if err != nil {
log.Fatalln("open file error!")
}
logger := log.New(logFile,"[Debug]",log.Ldate | log.Ltime | log.Lshortfile)
logger.Println("test debug message")
logger.SetPrefix("[Info]")
logger.Println("test info message")
}
|
运行效果：
|
1
2
3
4
5
6
|
[root@local t2]# go build logTest1.go
[root@local t2]# ./logTest1
[root@local t2]# cat test1.log
[Debug]2017/06/13 23:18:36 logTest1.go:19: test debug message
[Info]2017/06/13 23:18:36 logTest1.go:21: test info message
[root@local t2]#
|
go语言的log模块没有提供log rotate接口，但实际开发中我们需要该功能：
我们不希望单个日志过大，否则文本编辑器无法打开，查看比较困难；
更不希望占用太大的存储空间，可以指定最多存多少个日志文件。
解决方案
借助带缓冲的channel来实现。
示例代码如下：
|
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
|
/*
golang log rotate example
*/
package main
import (
"fmt"
"log"
"os"
"time"
)
const (
BACKUP_COUNT = 5
MAX_FILE_BYTES = 2 * 1024
)
func doRotate(fPrefix string) {
for j := BACKUP_COUNT; j >= 1; j-- {
curFileName := fmt.Sprintf("%s_%d.log",fPrefix,j)
k := j-1
preFileName := fmt.Sprintf("%s_%d.log",fPrefix,k)
if k == 0 {
preFileName = fmt.Sprintf("%s.log", fPrefix)
}
_,err := os.Stat(curFileName)
if err == nil {
os.Remove(curFileName)
fmt.Println("remove : ", curFileName)
}
_,err = os.Stat(preFileName)
if err == nil {
fmt.Println("rename : ", preFileName, " => ", curFileName)
err = os.Rename(preFileName, curFileName)
if err != nil {
fmt.Println(err)
}
}
}
}
func NewLogger(fPrefix string) (*log.Logger, *os.File) {
var logger *log.Logger
fileName := fmt.Sprintf("%s.log", fPrefix)
fmt.Println("fileName :", fileName)
logFile, err := os.OpenFile(fileName, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
fmt.Println("open file error!")
} else {
logger = log.New(logFile, "[Debug]", log.Ldate|log.Ltime|log.Lshortfile)
}
return logger, logFile
}
func logWorker(msgQueue <-chan string) {
fPrefix := "msg"
logger, logFile := NewLogger(fPrefix)
for msg := range msgQueue {
logger.Println(msg)
fi, err2 := logFile.Stat()
if err2 == nil {
if fi.Size() > MAX_FILE_BYTES {
logFile.Close()
doRotate(fPrefix)
logger,logFile = NewLogger(fPrefix)
}
}
}
logFile.Close()
}
func main() {
msgQueue := make(chan string, 1000)
go logWorker(msgQueue)
for j := 1; j <= 1000; j++ {
msgQueue <- fmt.Sprintf("msg_%d", j)
time.Sleep(1 * time.Second)
}
close(msgQueue)
return
}
|
运行效果如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
|
[root@local t2]# ./logRotateTest1
fileName : msg.log
rename : msg.log => msg_1.log
fileName : msg.log
rename : msg_1.log => msg_2.log
rename : msg.log => msg_1.log
fileName : msg.log
rename : msg_2.log => msg_3.log
rename : msg_1.log => msg_2.log
rename : msg.log => msg_1.log
fileName : msg.log
^C
|
讨论
这里只是个简单的示例代码，实现了log rotate，更多功能需自行开发。
好，就这些了，希望对你有帮助。
以上这篇golang之log rotate详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持服务器之家。