blob: 4594fa06f9b2bdae756a49a396fd6581af9a5dd4 (
plain)
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
|
// SPDX-License-Identifier: GPL-3.0-or-later
package matcher
import "sync"
type (
cachedMatcher struct {
matcher Matcher
mux sync.RWMutex
cache map[string]bool
}
)
// WithCache adds cache to the matcher.
func WithCache(m Matcher) Matcher {
switch m {
case TRUE(), FALSE():
return m
default:
return &cachedMatcher{matcher: m, cache: make(map[string]bool)}
}
}
func (m *cachedMatcher) Match(b []byte) bool {
s := string(b)
if result, ok := m.fetch(s); ok {
return result
}
result := m.matcher.Match(b)
m.put(s, result)
return result
}
func (m *cachedMatcher) MatchString(s string) bool {
if result, ok := m.fetch(s); ok {
return result
}
result := m.matcher.MatchString(s)
m.put(s, result)
return result
}
func (m *cachedMatcher) fetch(key string) (result bool, ok bool) {
m.mux.RLock()
result, ok = m.cache[key]
m.mux.RUnlock()
return
}
func (m *cachedMatcher) put(key string, result bool) {
m.mux.Lock()
m.cache[key] = result
m.mux.Unlock()
}
|
