-
Notifications
You must be signed in to change notification settings - Fork 51
/
counters.go
53 lines (39 loc) · 1.16 KB
/
counters.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
package counters
import (
"sync/atomic"
"go.aporeto.io/trireme-lib/collector"
)
// NewCounters initializes new counters handler. Thread safe.
func NewCounters() *Counters {
return &Counters{
counters: make([]uint32, errMax+1),
}
}
// CounterNames returns an array of names
func CounterNames() []string {
names := make([]string, errMax+1)
var ct CounterType
for ct = 0; ct <= errMax; ct++ {
names[ct] = ct.String()
}
return names[:errMax]
}
// CounterError is a convinence function which returns error as well as increments the counter.
func (c *Counters) CounterError(t CounterType, err error) error {
atomic.AddUint32(&c.counters[int(t)], 1)
return err
}
// IncrementCounter increments counters for a given PU
func (c *Counters) IncrementCounter(t CounterType) {
atomic.AddUint32(&c.counters[int(t)], 1)
}
// GetErrorCounters returns the error counters and resets the counters to zero
func (c *Counters) GetErrorCounters() []collector.Counters {
c.Lock()
defer c.Unlock()
report := make([]collector.Counters, totalCounters)
for index := range c.counters {
report[index] = collector.Counters(atomic.SwapUint32(&c.counters[index], 0))
}
return report
}