forked from ghj1976/tailMail
-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger.go
67 lines (56 loc) · 1.38 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
package tailMail
import (
"fmt"
"log"
"os"
"os/exec"
"path"
"path/filepath"
"time"
)
// 日志文件句柄
var logfile *os.File
// 执行结果是否输出到文件上,如果 true 表示输出到文件,否则输出到屏幕上。
var OutPutLog = false
// 如果要输出到文件,准备好文件
func InitLogFile(dir string) {
if OutPutLog {
logfilename := getLogFileName(dir)
logfile, err := os.OpenFile(logfilename, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0700)
if err != nil {
fmt.Println(err)
os.Exit(-1)
}
log.SetOutput(logfile)
}
}
// 应用退出时 关闭文件句柄
func LoggerFinish() {
if OutPutLog {
logfile.Close()
}
}
// 获取当前文件名
func getLogFileName(dir string) (filename string) {
file, _ := exec.LookPath(os.Args[0])
fileName := filepath.Base(file)
logfileName := fmt.Sprintf("%s.%s.log", fileName, time.Now().Format("20060102"))
dirPath := path.Join(dir, "log")
if !isDirExists(dirPath) {
// 参考 http://stackoverflow.com/questions/14249467/os-mkdir-and-os-mkdirall-permission-value
os.Mkdir(dirPath, 0700)
}
logfileName = path.Join(dirPath, logfileName)
fmt.Println("输出信息情况日志文件:", logfileName)
return logfileName
}
// 判断目录是否存在
func isDirExists(dir string) bool {
fi, err := os.Stat(dir)
if err != nil {
return os.IsExist(err)
} else {
return fi.IsDir()
}
panic("not reached")
}