/
resource.go
84 lines (65 loc) · 2.19 KB
/
resource.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
package collector
import (
"encoding/json"
"fmt"
"os/exec"
"strings"
"github.com/prometheus/client_golang/prometheus"
kingpin "gopkg.in/alecthomas/kingpin.v2"
)
var (
resourceSubsystem = "resource"
resourceLabels = []string{"resource"}
resourceDesc = map[string]*prometheus.Desc{
"max": colPromDesc(resourceSubsystem, "maximum", "Resource Maximum.", resourceLabels),
"used": colPromDesc(resourceSubsystem, "used", "Resource Used.", resourceLabels),
}
totalResourceErrors = 0.0
cumulusCLResPath = kingpin.Flag("cumulus.cl-resource-query.path", "Path of cl-resource-query.").Default("/usr/cumulus/bin/cl-resource-query").String()
)
func init() {
registerCollector(resourceSubsystem, enabledByDefault, NewResourceCollector)
}
// ResourceCollector collects resource metrics, implemented as per the Collector interface.
type ResourceCollector struct{}
// NewResourceCollector returns a new ResourceCollector.
func NewResourceCollector() Collector {
return &ResourceCollector{}
}
// Get metrics and send to the Prometheus.Metric channel.
func (c *ResourceCollector) Get(ch chan<- prometheus.Metric) (float64, error) {
jsonResources, err := getResourceStats()
if err != nil {
totalResourceErrors++
return totalResourceErrors, fmt.Errorf("cannot get resources: %s", err)
}
if err := processResourceStats(ch, jsonResources); err != nil {
totalResourceErrors++
return totalResourceErrors, err
}
return totalResourceErrors, nil
}
func getResourceStats() ([]byte, error) {
args := []string{"-j"}
output, err := exec.Command(*cumulusCLResPath, args...).Output()
if err != nil {
return nil, err
}
return output, nil
}
func processResourceStats(ch chan<- prometheus.Metric, jsonResourceSum []byte) error {
dataMap := map[string]*resourceEntry{}
if err:= json.Unmarshal(jsonResourceSum, &dataMap); err != nil {
return fmt.Errorf("cannot unmarshal resource json: %s", err)
}
for _, v := range dataMap {
newGauge(ch, resourceDesc["max"], v.Max, strings.ToLower(v.Name))
newGauge(ch, resourceDesc["used"], v.Count, strings.ToLower(v.Name))
}
return nil
}
type resourceEntry struct {
Count float64 `json:"count"`
Max float64 `json:"max"`
Name string `json:"name"`
}