forked from StephanU/beats
-
Notifications
You must be signed in to change notification settings - Fork 0
/
status.go
97 lines (84 loc) · 2.03 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
package status
import (
"errors"
"io"
"io/ioutil"
"net"
"net/http"
"net/url"
"strings"
"github.com/elastic/beats/libbeat/common/cfgwarn"
"github.com/elastic/beats/libbeat/logp"
"github.com/elastic/beats/libbeat/common"
"github.com/elastic/beats/metricbeat/mb"
"github.com/elastic/beats/metricbeat/module/uwsgi"
)
func init() {
mb.Registry.AddMetricSet("uwsgi", "status", New, uwsgi.HostParser)
}
// MetricSet for fetching uwsgi metrics from StatServer.
type MetricSet struct {
mb.BaseMetricSet
}
// New creates a new instance of the MetricSet.
func New(base mb.BaseMetricSet) (mb.MetricSet, error) {
cfgwarn.Experimental("The uWSGI status metricset is experimental")
return &MetricSet{BaseMetricSet: base}, nil
}
func fetchStatData(URL string) ([]byte, error) {
var reader io.Reader
u, err := url.Parse(URL)
if err != nil {
logp.Err("parsing uwsgi stats url failed: ", err)
return nil, err
}
switch u.Scheme {
case "tcp":
conn, err := net.Dial(u.Scheme, u.Host)
if err != nil {
return nil, err
}
defer conn.Close()
reader = conn
case "unix":
path := strings.Replace(URL, "unix://", "", -1)
conn, err := net.Dial(u.Scheme, path)
if err != nil {
return nil, err
}
defer conn.Close()
reader = conn
case "http", "https":
res, err := http.Get(u.String())
if err != nil {
return nil, err
}
defer res.Body.Close()
if res.StatusCode != 200 {
logp.Err("failed to fetch uwsgi status with code: ", res.StatusCode)
return nil, errors.New("http failed")
}
reader = res.Body
default:
return nil, errors.New("unknown scheme")
}
data, err := ioutil.ReadAll(reader)
if err != nil {
logp.Err("uwsgi data read failed: ", err)
return nil, err
}
return data, nil
}
// Fetch methods implements the data gathering and data conversion to the right
// format.
func (m *MetricSet) Fetch() ([]common.MapStr, error) {
content, err := fetchStatData(m.HostData().URI)
if err != nil {
return []common.MapStr{
common.MapStr{
"error": err.Error(),
},
}, err
}
return eventsMapping(content)
}