/
parser.go
112 lines (89 loc) · 1.98 KB
/
parser.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
package parser
import (
"encoding/json"
"errors"
"fmt"
"gopkg.in/yaml.v2"
"strconv"
"strings"
"time"
)
type Item struct {
Cpu string
Memory string
}
type Container struct {
Chart Item
Text Item
}
type Arcentry struct {
ApiKey string `yaml:"apiKey"`
DocId string `yaml:"docId"`
}
type Config struct {
Config struct {
Arcentry Arcentry
Watch struct {
Interval string
}
Containers []map[string]Container
}
}
type Stat struct {
Container string
Memory struct {
Raw string
Percent string
}
Cpu string
}
type ChartDocument struct {
Data [][]float64 `json:"data"`
MaxY float64 `json:"maxY"`
MinY float64 `json:"minY"`
YAxisSpace string `json:"yAxisSpace"`
}
type TextDocument struct {
Text string `json:"text"`
}
type Request struct {
Objects map[string]interface{} `json:"objects"`
}
func ReadConfig(bytes []byte) (config *Config, err error) {
err = yaml.Unmarshal(bytes, &config)
if err != nil {
return nil, err
}
return config, nil
}
func ReadStat(content string) (stat *Stat, err error) {
err = json.Unmarshal([]byte(content), &stat)
if err != nil {
return nil, err
}
return stat, nil
}
func ParseRequest(content interface{}) (req []byte, err error) {
bytes, err := json.Marshal(content)
if err != nil {
return nil, err
}
return bytes, nil
}
func parseUnit(str string, dur time.Duration, unit string) (time.Duration, error) {
i, err := strconv.Atoi(strings.Replace(str, unit, "", -1))
if err != nil {
return 5 * time.Second, err
}
return time.Duration(i) * dur, nil
}
func ParseDuration(duration string) (dur time.Duration, err error) {
if strings.Contains(duration, "s") {
return parseUnit(duration, time.Second, "s")
} else if strings.Contains(duration, "m") {
return parseUnit(duration, time.Minute, "m")
} else if strings.Contains(duration, "h") {
return parseUnit(duration, time.Hour, "h")
}
return 5 * time.Second, errors.New(fmt.Sprintf("invalid duration format: %s", duration))
}