forked from Monibuca/utils
/
logger.go
87 lines (75 loc) · 1.95 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package utils
import (
"context"
"fmt"
"io"
"log"
"os"
"time"
colorable "github.com/mattn/go-colorable"
"github.com/logrusorgru/aurora"
)
// MultiLogWriter 多端写日志类
type MultiLogWriter struct {
writers []io.Writer
io.Writer
}
var logWriter MultiLogWriter
var multiLogger = log.New(&logWriter, "", log.LstdFlags)
var colorLogger = log.New(colorable.NewColorableStdout(), "", log.LstdFlags)
func init() {
log.SetOutput(io.MultiWriter(os.Stdout, &logWriter))
logWriter.Writer = io.MultiWriter()
}
// AddWriter 添加日志输出端
func AddWriter(wn io.Writer) {
logWriter.writers = append(logWriter.writers, wn)
logWriter.Writer = io.MultiWriter(logWriter.writers...)
}
// MayBeError 优雅错误判断加日志辅助函数
func MayBeError(info error) (hasError bool) {
if hasError = info != nil; hasError {
Print(aurora.Red(info))
}
return
}
func getNoColor(v ...interface{}) (noColor []interface{}) {
noColor = append(noColor, v...)
for i, value := range v {
if vv, ok := value.(aurora.Value); ok {
noColor[i] = vv.Value()
}
}
return
}
// Print 带颜色识别
func Print(v ...interface{}) {
noColor := getNoColor(v...)
colorLogger.Output(2, fmt.Sprint(v...))
multiLogger.Output(2, fmt.Sprint(noColor...))
}
// Printf calls Output to print to the standard logger.
// Arguments are handled in the manner of fmt.Printf.
func Printf(format string, v ...interface{}) {
noColor := getNoColor(v...)
colorLogger.Output(2, fmt.Sprintf(format, v...))
multiLogger.Output(2, fmt.Sprintf(format, noColor...))
}
// Println calls Output to print to the standard logger.
// Arguments are handled in the manner of fmt.Println.
func Println(v ...interface{}) {
noColor := getNoColor(v...)
colorLogger.Output(2, fmt.Sprintln(v...))
multiLogger.Output(2, fmt.Sprintln(noColor...))
}
type Event struct {
Timestamp time.Time
Level int
Label string
Tag string
}
type EventContext struct {
Name string
context.Context
EventChan chan *Event
}