-
Notifications
You must be signed in to change notification settings - Fork 2
/
instrumenting.go
86 lines (70 loc) · 2.18 KB
/
instrumenting.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
package dashboard
import (
"time"
gometrics "github.com/rcrowley/go-metrics"
)
type instrumentingMiddleware struct {
requestCount gometrics.Counter
requestLatency gometrics.Histogram
countResult gometrics.Histogram
next DashboardService
}
func NewInstrumentingMiddleware(requestCount gometrics.Counter,
requestLatency gometrics.Histogram,
countResult gometrics.Histogram,
next DashboardService) DashboardService {
return instrumentingMiddleware{
requestCount: requestCount,
requestLatency: requestLatency,
countResult: countResult,
next: next,
}
}
func (mw instrumentingMiddleware) GetPages() (pages []Page, err error) {
defer func(begin time.Time) {
mw.requestCount.Inc(1)
mw.requestLatency.Update(time.Since(begin).Nanoseconds())
}(time.Now())
pages, err = mw.next.GetPages()
return
}
func (mw instrumentingMiddleware) GetUnregisteredWidgets() (widgets []Widget, err error) {
defer func(begin time.Time) {
mw.requestCount.Inc(1)
mw.requestLatency.Update(time.Since(begin).Nanoseconds())
}(time.Now())
widgets, err = mw.next.GetUnregisteredWidgets()
return
}
func (mw instrumentingMiddleware) Register(widget Widget) (pr RegisterResponse, err error) {
defer func(begin time.Time) {
mw.requestCount.Inc(1)
mw.requestLatency.Update(time.Since(begin).Nanoseconds())
}(time.Now())
pr, err = mw.next.Register(widget)
return
}
func (mw instrumentingMiddleware) RegisterToPage(pageId int64, widgetId string) (pr RegisterResponse, err error) {
defer func(begin time.Time) {
mw.requestCount.Inc(1)
mw.requestLatency.Update(time.Since(begin).Nanoseconds())
}(time.Now())
pr, err = mw.next.RegisterToPage(pageId, widgetId)
return
}
func (mw instrumentingMiddleware) InsertPage(page Page) (response InsertPageResponse, err error) {
defer func(begin time.Time) {
mw.requestCount.Inc(1)
mw.requestLatency.Update(time.Since(begin).Nanoseconds())
}(time.Now())
response, err = mw.next.InsertPage(page)
return
}
func (mw instrumentingMiddleware) Init() (errs []error, ok bool) {
defer func(begin time.Time) {
mw.requestCount.Inc(1)
mw.requestLatency.Update(time.Since(begin).Nanoseconds())
}(time.Now())
errs, ok = mw.next.Init()
return
}