/
smartctl_exporter.go
118 lines (97 loc) · 2.18 KB
/
smartctl_exporter.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
116
117
118
package main
import (
"fmt"
"log"
"net/http"
"sync"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto"
"github.com/prometheus/client_golang/prometheus/promhttp"
)
var (
devices []*Device
results = sync.Map{}
collectors = map[string]*prometheus.GaugeVec{}
_flags = Flags{}
AppName = "smartctl_exporter_by_id"
Version = ""
BuildDate = ""
)
func WithMetrics(handler http.Handler) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
wg := sync.WaitGroup{}
wg.Add(len(devices))
for _, d := range devices {
go func(d *Device) {
results.Store(d.Name, GetAll(d))
wg.Done()
}(d)
}
wg.Wait()
for _, d := range devices {
if r, ok := results.Load(d.Name); ok {
r := r.(*Result)
passed := 0
if r.Passed {
passed = 1
}
collectors["device_status"].
WithLabelValues(d.Name, r.ModelName, r.SerialNumber, r.FirmwareVersion, d.LabelPath).
Set(float64(passed))
for k, v := range r.Attributes {
c, ok := collectors[k]
if !ok {
c = promauto.NewGaugeVec(
prometheus.GaugeOpts{
Namespace: "smartctl",
Name: k,
},
[]string{"device", "label_path"},
)
collectors[k] = c
}
c.WithLabelValues(d.Name, d.LabelPath).Set(v)
}
}
}
handler.ServeHTTP(w, r)
}
}
func main() {
_flags.init()
if *_flags.Version {
fmt.Printf(
"%s\n"+
"Version: \t%s\n"+
"Build date: \t%s\n",
AppName,
Version,
BuildDate)
return
}
devices = GetDevices()
deviceStatusCollector := promauto.NewGaugeVec(
prometheus.GaugeOpts{
Namespace: "smartctl",
Name: "device_status",
Help: "Device Status",
},
[]string{
"device",
"model_name",
"serial_number",
"firmware_version",
"label_path",
},
)
collectors["device_status"] = deviceStatusCollector
promHandler := promhttp.Handler()
hf := WithMetrics(promHandler)
if !*_flags.disableAuth {
hf = BasicAuth(hf)
}
http.HandleFunc(*_flags.Path, hf)
addr := fmt.Sprintf("%s:%d", *_flags.Address, *_flags.Port)
log.Printf("Listen: %s\n", addr)
log.Fatal(http.ListenAndServe(addr, nil))
}