forked from go-ozzo/ozzo-log
-
Notifications
You must be signed in to change notification settings - Fork 2
/
target_console.go
109 lines (97 loc) · 2.45 KB
/
target_console.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
// Copyright 2015 Qiang Xue. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package log
import (
"errors"
"fmt"
"io"
"os"
"github.com/admpub/color"
)
var colorBrushes = map[Leveler]*color.Color{
LevelDebug: color.New(color.FgHiCyan), // cyan
LevelInfo: color.New(color.FgHiGreen), // green
LevelWarn: color.New(color.FgHiYellow), // yellow
LevelError: color.New(color.FgHiRed), // red
LevelFatal: color.New(color.FgHiMagenta), // magenta
}
const (
ColorFlag = iota
ColorRow
)
// ConsoleTarget writes filtered log messages to console window.
type ConsoleTarget struct {
*Filter
ColorMode bool // whether to use colors to differentiate log levels
ColorType int
Writer io.Writer // the writer to write log messages
close chan bool
outputFunc func(*ConsoleTarget, *Entry)
}
// NewConsoleTarget creates a ConsoleTarget.
// The new ConsoleTarget takes these default options:
// MaxLevel: LevelDebug, ColorMode: true, Writer: os.Stdout
func NewConsoleTarget() *ConsoleTarget {
return &ConsoleTarget{
Filter: &Filter{MaxLevel: LevelDebug},
ColorMode: DefaultConsoleColorize,
ColorType: ColorFlag,
Writer: os.Stdout,
close: make(chan bool),
}
}
// Open prepares ConsoleTarget for processing log messages.
func (t *ConsoleTarget) Open(io.Writer) error {
t.Filter.Init()
if t.Writer == nil {
return errors.New("ConsoleTarget.Writer cannot be nil")
}
if t.ColorMode {
switch t.ColorType {
case ColorFlag:
t.outputFunc = func(t *ConsoleTarget, e *Entry) {
fmt.Fprintln(t.Writer, t.ColorizeFlag(e))
}
default:
t.outputFunc = func(t *ConsoleTarget, e *Entry) {
fmt.Fprintln(t.Writer, t.ColorizeRow(e))
}
}
} else {
t.outputFunc = func(t *ConsoleTarget, e *Entry) {
fmt.Fprintln(t.Writer, e.String())
}
}
return nil
}
// Process writes a log message using Writer.
func (t *ConsoleTarget) Process(e *Entry) {
if e == nil {
t.close <- true
return
}
if !t.Allow(e) {
return
}
t.outputFunc(t, e)
}
func (t *ConsoleTarget) ColorizeFlag(e *Entry) string {
s := e.Level.Tag()
cs := e.Level.Color()
if cs != nil {
return cs.SprintFunc()(s) + e.String()
}
return s + e.String()
}
func (t *ConsoleTarget) ColorizeRow(e *Entry) string {
cs := e.Level.Color()
if cs != nil {
return cs.SprintFunc()(e.String())
}
return e.String()
}
// Close closes the console target.
func (t *ConsoleTarget) Close() {
<-t.close
}