forked from akash-network/node
/
printer.go
78 lines (64 loc) · 1.57 KB
/
printer.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
package session
import (
"time"
)
type Printer interface {
Flush() error
Data() *PrinterData
}
type PrinterResult map[string]string
type PrinterResultMode uint
const (
PrinterResultModeKV PrinterResultMode = iota
PrinterResultModeList
)
type PrinterData struct {
Result []map[string]string `json:"result,omitempty"`
Raw interface{} `json:"raw,omitempty"`
Log []PrinterLog `json:"log,omitempty"`
resultMode PrinterResultMode
}
func (d *PrinterData) ResultMode() PrinterResultMode {
return d.resultMode
}
func NewPrinterDataList() *PrinterData {
return &PrinterData{resultMode: PrinterResultModeList}
}
func NewPrinterDataKV() *PrinterData {
return &PrinterData{resultMode: PrinterResultModeKV}
}
func (d *PrinterData) AddResultKV(key, value string) *PrinterData {
if len(d.Result) == 0 {
d.Result = make([]map[string]string, 1, 1)
}
if d.Result[0] == nil {
d.Result[0] = map[string]string{key: value}
} else {
d.Result[0][key] = value
}
return d
}
func (d *PrinterData) AddResultList(results ...PrinterResult) *PrinterData {
if len(d.Result) == 0 {
d.Result = make([]map[string]string, 0, 0)
}
for _, res := range results {
obj := make(map[string]string)
for k, v := range res {
obj[k] = v
}
d.Result = append(d.Result, obj)
}
return d
}
type PrinterLogLevel string
const (
PrinterLogLevelInfo PrinterLogLevel = "info"
PrinterLogLevelDebug = "debug"
PrinterLogLevelError = "error"
)
type PrinterLog struct {
Timestamp time.Time
Message interface{}
Level PrinterLogLevel
}