-
Notifications
You must be signed in to change notification settings - Fork 6
/
base.go
125 lines (107 loc) · 2.69 KB
/
base.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
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
package log
import (
"github.com/griffin702/ginana/library/log/hook"
rotatelogs "github.com/lestrrat/go-file-rotatelogs"
"github.com/sirupsen/logrus"
"github.com/ulricqin/goutils/filetool"
"io"
"os"
"time"
)
var (
_level logrus.Level
_path string
_maxAge uint
_rotationTime uint
)
type logger struct {
log *logrus.Logger
logType int
}
func (l *logger) GetOutFile() (out io.Writer) {
return l.log.Out
}
func (l *logger) isStdOut() bool {
return l.logType == 0
}
func (l *logger) NewStdOut() {
cli := logrus.New()
cli.SetLevel(_level)
cli.Out = os.Stdout
cli.Formatter = &GiNanaStdFormatter{}
cli.AddHook(&hook.DefaultFieldHook{})
cli.AddHook(&hook.LineHook{})
l.log = cli
return
}
func (l *logger) NewFile() (cf func()) {
cli := logrus.New()
cli.SetLevel(_level)
out, _ := os.OpenFile(os.DevNull, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
cf = func() {
if out != nil {
_ = out.Close()
}
}
_ = filetool.InsureDir(_path)
logWriter, err := rotatelogs.New(
_path+"/log-%Y-%m-%d-%H-%M.log",
//rotatelogs.WithLinkName(d.path),
rotatelogs.WithMaxAge(time.Duration(_maxAge)*time.Hour), // 文件最大保存时间
rotatelogs.WithRotationTime(time.Duration(_rotationTime)*time.Hour), // 日志切割时间间隔
)
if err != nil {
return
}
cli.Out = logWriter
cli.Formatter = &GiNanaStdFormatter{
DisableColors: true,
}
//writeMap := lfshook.WriterMap{
// logrus.DebugLevel: logWriter,
// logrus.InfoLevel: logWriter,
// logrus.WarnLevel: logWriter,
// logrus.ErrorLevel: logWriter,
// logrus.FatalLevel: logWriter,
// logrus.PanicLevel: logWriter,
//}
//lfHook := lfshook.NewHook(writeMap, &logrus.TextFormatter{DisableColors: true})
cli.AddHook(&hook.DefaultFieldHook{})
cli.AddHook(&hook.LineHook{})
//cli.AddHook(lfHook)
l.log = cli
return
}
func (l *logger) Print(args ...interface{}) {
l.log.Print(args...)
}
func (l *logger) Println(args ...interface{}) {
l.log.Println(args...)
}
func (l *logger) Error(args ...interface{}) {
l.log.Error(args...)
}
func (l *logger) Warn(args ...interface{}) {
l.log.Warn(args...)
}
func (l *logger) Info(args ...interface{}) {
l.log.Info(args...)
}
func (l *logger) Debug(args ...interface{}) {
l.log.Debug(args...)
}
func (l *logger) Printf(format string, args ...interface{}) {
l.log.Printf(format, args...)
}
func (l *logger) Errorf(format string, args ...interface{}) {
l.log.Errorf(format, args...)
}
func (l *logger) Warnf(format string, args ...interface{}) {
l.log.Warnf(format, args...)
}
func (l *logger) Infof(format string, args ...interface{}) {
l.log.Infof(format, args...)
}
func (l *logger) Debugf(format string, args ...interface{}) {
l.log.Debugf(format, args...)
}