generated from xmidt-org/.go-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
logger.go
115 lines (92 loc) · 2.15 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
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
package main
import (
"bytes"
"fmt"
"io"
"os"
)
const (
ErrorLabel = "ERROR"
PassLabel = "PASS"
FailLabel = "FAIL"
indent = " "
)
type SampleResult struct {
Name string
Err error
}
type TemplateResult struct {
Name string
Err error
SampleResults []SampleResult
}
type Logger interface {
Debugf(message string, args ...interface{}) error
Errorf(message string, args ...interface{}) error
Result(TemplateResult) error
}
type ConsoleLogger struct {
Out io.Writer
Err io.Writer
Verbose bool
buffer bytes.Buffer
}
func (cr *ConsoleLogger) out() io.Writer {
if cr.Out != nil {
return cr.Out
}
return os.Stdout
}
func (cr *ConsoleLogger) err() io.Writer {
if cr.Err != nil {
return cr.Err
}
return os.Stderr
}
func (cr *ConsoleLogger) Debugf(format string, args ...interface{}) (err error) {
if !cr.Verbose {
_, err = fmt.Fprintf(cr.out(), format+"\n", args...)
}
return
}
func (cr *ConsoleLogger) Errorf(format string, args ...interface{}) (err error) {
_, err = fmt.Fprintf(cr.err(), format+"\n", args...)
return
}
func (cr *ConsoleLogger) Result(tr TemplateResult) (err error) {
cr.buffer.Reset()
var (
headerWritten = false
// closure that ensures a header line is only written one time.
// since we may not know if a header will need to be written until
// iterating over the sample results, we can just blindly call this
// closure before every output.
headerOnce = func() {
if !headerWritten {
fmt.Fprintln(&cr.buffer, tr.Name)
}
headerWritten = true
}
)
if cr.Verbose {
// always write the header for verbose output,
headerOnce()
} else if tr.Err != nil {
headerOnce()
fmt.Fprintf(&cr.buffer, "%s%-5.5s\t%s\n", indent, ErrorLabel, tr.Err)
}
for _, sr := range tr.SampleResults {
if sr.Err != nil {
headerOnce()
fmt.Fprintf(&cr.buffer, "%s%-5.5s\t%s\t%s\n", indent, FailLabel, sr.Name, sr.Err)
} else if cr.Verbose {
headerOnce()
fmt.Fprintf(&cr.buffer, "%s%-5.5s\t%s\n", indent, PassLabel, sr.Name)
}
}
if cr.buffer.Len() > 0 {
// we've already terminated each line with a newline
_, err = fmt.Fprint(cr.out(), cr.buffer.String())
}
return
}