forked from sgoware/go-sensitive
/
manager.go
49 lines (38 loc) · 881 Bytes
/
manager.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
package sensitive
import (
"github.com/Kayshen-X/go-sensitive/filter"
"github.com/Kayshen-X/go-sensitive/store"
"sync"
)
type Manager struct {
store store.Model
filter filter.Model
filterMux sync.RWMutex
}
func NewFilter(storeOption StoreOption, filterOption FilterOption) *Manager {
var filterStore store.Model
var myFilter filter.Model
switch storeOption.Type {
case StoreMemory:
filterStore = store.NewMemoryModel()
}
switch filterOption.Type {
case FilterDfa:
dfaModel := filter.NewDfaModel()
go dfaModel.Listen(filterStore.GetAddChan(), filterStore.GetDelChan())
myFilter = dfaModel
}
return &Manager{
store: filterStore,
filter: myFilter,
}
}
func (m *Manager) GetStore() store.Model {
return m.store
}
func (m *Manager) GetFilter() filter.Model {
m.filterMux.RLock()
myFilter := m.filter
m.filterMux.RUnlock()
return myFilter
}