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
|
// Copyright 2016 the Go-FUSE Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package benchmark
import (
"sync"
"time"
)
type latencyMapEntry struct {
count int
dur time.Duration
}
type LatencyMap struct {
sync.Mutex
stats map[string]*latencyMapEntry
}
func NewLatencyMap() *LatencyMap {
m := &LatencyMap{}
m.stats = make(map[string]*latencyMapEntry)
return m
}
func (m *LatencyMap) Get(name string) (count int, dt time.Duration) {
m.Mutex.Lock()
l := m.stats[name]
m.Mutex.Unlock()
if l == nil {
return 0, 0
}
return l.count, l.dur
}
func (m *LatencyMap) Add(name string, dt time.Duration) {
m.Mutex.Lock()
e := m.stats[name]
if e == nil {
e = new(latencyMapEntry)
m.stats[name] = e
}
e.count++
e.dur += dt
m.Mutex.Unlock()
}
func (m *LatencyMap) Counts() map[string]int {
r := make(map[string]int)
m.Mutex.Lock()
for k, v := range m.stats {
r[k] = v.count
}
m.Mutex.Unlock()
return r
}
|
