forked from elastic/beats
-
Notifications
You must be signed in to change notification settings - Fork 0
/
status.go
114 lines (94 loc) · 2.38 KB
/
status.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
/*
Package status fetches MySQL server status metrics.
For more information on the query it uses, see:
http://dev.mysql.com/doc/refman/5.7/en/show-status.html
*/
package status
/*
TODO @ruflin, 20160315
* Complete fields read
* Complete template
* Complete dashboards
*/
import (
"database/sql"
"github.com/elastic/beats/libbeat/common"
"github.com/elastic/beats/libbeat/logp"
"github.com/elastic/beats/metricbeat/mb"
"github.com/elastic/beats/metricbeat/module/mysql"
"github.com/pkg/errors"
)
var (
debugf = logp.MakeDebug("mysql-status")
)
func init() {
if err := mb.Registry.AddMetricSet("mysql", "status", New); err != nil {
panic(err)
}
}
// MetricSet for fetching MySQL server status.
type MetricSet struct {
mb.BaseMetricSet
dsn string
db *sql.DB
}
// New creates and returns a new MetricSet instance.
func New(base mb.BaseMetricSet) (mb.MetricSet, error) {
// Unpack additional configuration options.
config := struct {
Hosts []string `config:"hosts" validate:"nonzero,required"`
Username string `config:"username"`
Password string `config:"password"`
}{
Username: "",
Password: "",
}
err := base.Module().UnpackConfig(&config)
if err != nil {
return nil, err
}
// Create and validate the data source name.
dsn, err := mysql.CreateDSN(base.Host(), config.Username, config.Password, base.Module().Config().Timeout)
if err != nil {
return nil, err
}
return &MetricSet{
BaseMetricSet: base,
dsn: dsn,
}, nil
}
// Fetch fetches status messages from a mysql host.
func (m *MetricSet) Fetch() (event common.MapStr, err error) {
if m.db == nil {
var err error
m.db, err = mysql.NewDB(m.dsn)
if err != nil {
return nil, errors.Wrap(err, "mysql-status fetch failed")
}
}
status, err := m.loadStatus(m.db)
if err != nil {
return nil, err
}
return eventMapping(status), nil
}
// loadStatus loads all status entries from the given database into an array.
func (m *MetricSet) loadStatus(db *sql.DB) (map[string]string, error) {
// Returns the global status, also for versions previous 5.0.2
rows, err := db.Query("SHOW /*!50002 GLOBAL */ STATUS;")
if err != nil {
return nil, err
}
defer rows.Close()
mysqlStatus := map[string]string{}
for rows.Next() {
var name string
var value string
err = rows.Scan(&name, &value)
if err != nil {
return nil, err
}
mysqlStatus[name] = value
}
return mysqlStatus, nil
}