/
cacherMock.go
159 lines (124 loc) · 3.15 KB
/
cacherMock.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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
package testscommon
import (
"sync"
)
// CacherMock -
type CacherMock struct {
mut sync.RWMutex
dataMap map[string]interface{}
mutAddedDataHandlers sync.RWMutex
addedDataHandlers []func(key []byte, val interface{})
}
// NewCacherMock -
func NewCacherMock() *CacherMock {
return &CacherMock{
dataMap: make(map[string]interface{}),
addedDataHandlers: make([]func(key []byte, val interface{}), 0),
}
}
// Clear -
func (cacher *CacherMock) Clear() {
cacher.mut.Lock()
defer cacher.mut.Unlock()
cacher.dataMap = make(map[string]interface{})
}
// Put -
func (cacher *CacherMock) Put(key []byte, value interface{}, _ int) (evicted bool) {
cacher.mut.Lock()
defer cacher.mut.Unlock()
cacher.dataMap[string(key)] = value
cacher.callAddedDataHandlers(key, value)
return false
}
func (cacher *CacherMock) callAddedDataHandlers(key []byte, val interface{}) {
cacher.mutAddedDataHandlers.RLock()
for _, handler := range cacher.addedDataHandlers {
go handler(key, val)
}
cacher.mutAddedDataHandlers.RUnlock()
}
// Get -
func (cacher *CacherMock) Get(key []byte) (value interface{}, ok bool) {
cacher.mut.RLock()
defer cacher.mut.RUnlock()
val, ok := cacher.dataMap[string(key)]
return val, ok
}
// Has -
func (cacher *CacherMock) Has(key []byte) bool {
cacher.mut.RLock()
defer cacher.mut.RUnlock()
_, ok := cacher.dataMap[string(key)]
return ok
}
// Peek -
func (cacher *CacherMock) Peek(key []byte) (value interface{}, ok bool) {
cacher.mut.RLock()
defer cacher.mut.RUnlock()
val, ok := cacher.dataMap[string(key)]
return val, ok
}
// HasOrAdd -
func (cacher *CacherMock) HasOrAdd(key []byte, value interface{}, _ int) (has, added bool) {
cacher.mut.Lock()
defer cacher.mut.Unlock()
_, has = cacher.dataMap[string(key)]
if has {
return true, false
}
cacher.dataMap[string(key)] = value
cacher.callAddedDataHandlers(key, value)
return false, true
}
// Remove -
func (cacher *CacherMock) Remove(key []byte) {
cacher.mut.Lock()
defer cacher.mut.Unlock()
delete(cacher.dataMap, string(key))
}
// Keys -
func (cacher *CacherMock) Keys() [][]byte {
cacher.mut.RLock()
defer cacher.mut.RUnlock()
keys := make([][]byte, len(cacher.dataMap))
idx := 0
for k := range cacher.dataMap {
keys[idx] = []byte(k)
idx++
}
return keys
}
// Len -
func (cacher *CacherMock) Len() int {
cacher.mut.RLock()
defer cacher.mut.RUnlock()
return len(cacher.dataMap)
}
// SizeInBytesContained -
func (cacher *CacherMock) SizeInBytesContained() uint64 {
return 0
}
// MaxSize -
func (cacher *CacherMock) MaxSize() int {
return 10000
}
// RegisterHandler -
func (cacher *CacherMock) RegisterHandler(handler func(key []byte, value interface{}), _ string) {
if handler == nil {
return
}
cacher.mutAddedDataHandlers.Lock()
cacher.addedDataHandlers = append(cacher.addedDataHandlers, handler)
cacher.mutAddedDataHandlers.Unlock()
}
// UnRegisterHandler -
func (cacher *CacherMock) UnRegisterHandler(string) {
}
// IsInterfaceNil returns true if there is no value under the interface
func (cacher *CacherMock) IsInterfaceNil() bool {
return cacher == nil
}
// Close -
func (cacher *CacherMock) Close() error {
return nil
}