/
prometheus_data.go
80 lines (67 loc) · 2.09 KB
/
prometheus_data.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
package promtop
import (
"context"
"log"
"sort"
"strconv"
"time"
"github.com/prometheus/client_golang/api"
v1 "github.com/prometheus/client_golang/api/prometheus/v1"
"github.com/prometheus/common/model"
"github.com/spf13/viper"
)
type PrometheusData struct {
}
func (p *PrometheusData) getClient() api.Client {
client, err := api.NewClient(api.Config{
Address: viper.GetString("prometheus_url"),
})
if err != nil {
log.Fatalf("Error creating client: %v", err)
}
return client
}
func (p *PrometheusData) GetInstances() []string {
client := p.getClient()
v1api := v1.NewAPI(client)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
result, warnings, err := v1api.Query(ctx, "up{job=\"node_exporter\"}", time.Now())
if err != nil {
log.Fatalf("Error querying Prometheus: %v", err)
}
if len(warnings) > 0 {
log.Fatalf("Warnings: %v\n", warnings)
}
instances := make([]string, 0, result.(model.Vector).Len())
for _, val := range result.(model.Vector) {
if val.Value == 1 {
instances = append(instances, string(val.Metric["instance"]))
}
}
return instances
}
func (p *PrometheusData) GetCpu(instance string) []float64 {
client := p.getClient()
v1api := v1.NewAPI(client)
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
result, warnings, err := v1api.Query(ctx, "100 - (avg by (instance,cpu) (rate(node_cpu_seconds_total{instance=\""+instance+"\",job=\"node_exporter\",mode=\"idle\"}[1m])) * 100)", time.Now())
if err != nil {
log.Fatalf("Error querying Prometheus: %v", err)
}
if len(warnings) > 0 {
log.Fatalf("Warnings: %v\n", warnings)
}
sort.Slice(result.(model.Vector), func(i, j int) bool {
cpu_i, _ := strconv.Atoi(string(result.(model.Vector)[i].Metric["cpu"]))
cpu_j, _ := strconv.Atoi(string(result.(model.Vector)[j].Metric["cpu"]))
return cpu_i < cpu_j
})
cpus := make([]float64, 0, result.(model.Vector).Len())
for _, val := range result.(model.Vector) {
// percent, err := strconv.ParseFloat(string(val.Value), 64)
cpus = append(cpus, float64(val.Value))
}
return cpus
}