/
engine_tokudb.go
81 lines (69 loc) · 1.74 KB
/
engine_tokudb.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
// Scrape `SHOW ENGINE TOKUDB STATUS`.
package collector
import (
"context"
"database/sql"
"strings"
"github.com/prometheus/client_golang/prometheus"
)
const (
// Subsystem.
tokudb = "engine_tokudb"
// Query.
engineTokudbStatusQuery = `SHOW ENGINE TOKUDB STATUS`
)
func sanitizeTokudbMetric(metricName string) string {
replacements := map[string]string{
">": "",
",": "",
":": "",
"(": "",
")": "",
" ": "_",
"-": "_",
"+": "and",
"/": "and",
}
for r := range replacements {
metricName = strings.Replace(metricName, r, replacements[r], -1)
}
return metricName
}
// ScrapeEngineTokudbStatus scrapes from `SHOW ENGINE TOKUDB STATUS`.
type ScrapeEngineTokudbStatus struct{}
// Name of the Scraper.
func (ScrapeEngineTokudbStatus) Name() string {
return "engine_tokudb_status"
}
// Help returns additional information about Scraper.
func (ScrapeEngineTokudbStatus) Help() string {
return "Collect from SHOW ENGINE TOKUDB STATUS"
}
// Version of MySQL from which scraper is available.
func (ScrapeEngineTokudbStatus) Version() float64 {
return 5.6
}
// Scrape collects data.
func (ScrapeEngineTokudbStatus) Scrape(ctx context.Context, db *sql.DB, ch chan<- prometheus.Metric) error {
tokudbRows, err := db.QueryContext(ctx, engineTokudbStatusQuery)
if err != nil {
return err
}
defer tokudbRows.Close()
var temp, key string
var val sql.RawBytes
for tokudbRows.Next() {
if err := tokudbRows.Scan(&temp, &key, &val); err != nil {
return err
}
key = strings.ToLower(key)
if floatVal, ok := parseStatus(val); ok {
ch <- prometheus.MustNewConstMetric(
newDesc(tokudb, sanitizeTokudbMetric(key), "Generic metric from SHOW ENGINE TOKUDB STATUS."),
prometheus.UntypedValue,
floatVal,
)
}
}
return nil
}