This repository has been archived by the owner on Jan 21, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 6
/
store.go
109 lines (91 loc) · 2.33 KB
/
store.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
package instruments
import (
"encoding/json"
"fmt"
"strconv"
"strings"
"code.cloudfoundry.org/lager"
"github.com/cloudfoundry-incubator/etcd-metrics-server/instrumentation"
)
type Store struct {
statsEndpoint string
keysEndpoint string
getter getter
logger lager.Logger
}
func NewStore(getter getter, etcdAddr string, logger lager.Logger) *Store {
return &Store{
statsEndpoint: fmt.Sprintf("%s/v2/stats/store", etcdAddr),
keysEndpoint: fmt.Sprintf("%s/v2/keys/", etcdAddr),
getter: getter,
logger: logger,
}
}
func (store *Store) Emit() instrumentation.Context {
context := instrumentation.Context{
Name: "store",
}
var stats map[string]uint64
statsResp, err := store.getter.Get(store.statsEndpoint)
if err != nil {
store.logger.Error("failed-to-collect-store-stats", err)
return context
}
defer statsResp.Body.Close()
err = json.NewDecoder(statsResp.Body).Decode(&stats)
if err != nil {
store.logger.Error("failed-to-unmarshal-store-stats", err)
return context
}
keysResp, err := store.getter.Get(store.keysEndpoint)
if err != nil {
store.logger.Error("failed-to-read-from-store", err)
return context
}
defer keysResp.Body.Close()
etcdIndexHeader := keysResp.Header.Get("X-Etcd-Index")
raftIndexHeader := keysResp.Header.Get("X-Raft-Index")
raftTermHeader := keysResp.Header.Get("X-Raft-Term")
etcdIndex, err := strconv.ParseUint(etcdIndexHeader, 10, 0)
if err != nil {
store.logger.Error("failed-to-parse-etcd-index", err, lager.Data{
"index": etcdIndexHeader,
})
return context
}
raftIndex, err := strconv.ParseUint(raftIndexHeader, 10, 0)
if err != nil {
store.logger.Error("failed-to-parse-raft-index", err, lager.Data{
"index": raftIndexHeader,
})
return context
}
raftTerm, err := strconv.ParseUint(raftTermHeader, 10, 0)
if err != nil {
store.logger.Error("failed-to-parse-raft-term", err, lager.Data{
"term": raftTermHeader,
})
return context
}
context.Metrics = []instrumentation.Metric{
{
Name: "EtcdIndex",
Value: etcdIndex,
},
{
Name: "RaftIndex",
Value: raftIndex,
},
{
Name: "RaftTerm",
Value: raftTerm,
},
}
for name, val := range stats {
context.Metrics = append(context.Metrics, instrumentation.Metric{
Name: strings.ToUpper(name[0:1]) + name[1:],
Value: val,
})
}
return context
}