mirror of https://github.com/poanetwork/quorum.git
Generic filter interface
This commit is contained in:
parent
25cf0c440c
commit
9e1689df9c
|
@ -0,0 +1,70 @@
|
|||
package filter
|
||||
|
||||
import "reflect"
|
||||
|
||||
type Filter interface {
|
||||
Compare(Filter) bool
|
||||
Trigger(data interface{})
|
||||
}
|
||||
|
||||
type FilterEvent struct {
|
||||
filter Filter
|
||||
data interface{}
|
||||
}
|
||||
|
||||
type Filters struct {
|
||||
id int
|
||||
watchers map[int]Filter
|
||||
ch chan FilterEvent
|
||||
|
||||
quit chan struct{}
|
||||
}
|
||||
|
||||
func New() *Filters {
|
||||
return &Filters{
|
||||
ch: make(chan FilterEvent),
|
||||
watchers: make(map[int]Filter),
|
||||
quit: make(chan struct{}),
|
||||
}
|
||||
}
|
||||
|
||||
func (self *Filters) Start() {
|
||||
go self.loop()
|
||||
}
|
||||
|
||||
func (self *Filters) Stop() {
|
||||
close(self.quit)
|
||||
}
|
||||
|
||||
func (self *Filters) Notify(filter Filter, data interface{}) {
|
||||
self.ch <- FilterEvent{filter, data}
|
||||
}
|
||||
|
||||
func (self *Filters) Install(watcher Filter) int {
|
||||
self.watchers[self.id] = watcher
|
||||
self.id++
|
||||
|
||||
return self.id - 1
|
||||
}
|
||||
|
||||
func (self *Filters) Uninstall(id int) {
|
||||
delete(self.watchers, id)
|
||||
}
|
||||
|
||||
func (self *Filters) loop() {
|
||||
out:
|
||||
for {
|
||||
select {
|
||||
case <-self.quit:
|
||||
break out
|
||||
case event := <-self.ch:
|
||||
for _, watcher := range self.watchers {
|
||||
if reflect.TypeOf(watcher) == reflect.TypeOf(event.filter) {
|
||||
if watcher.Compare(event.filter) {
|
||||
watcher.Trigger(event.data)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
|
@ -0,0 +1,34 @@
|
|||
package filter
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestFilters(t *testing.T) {
|
||||
var success bool
|
||||
var failure bool
|
||||
|
||||
fm := New()
|
||||
fm.Start()
|
||||
fm.Install(Generic{
|
||||
Str1: "hello",
|
||||
Fn: func(data interface{}) {
|
||||
success = data.(bool)
|
||||
},
|
||||
})
|
||||
fm.Install(Generic{
|
||||
Str1: "hello1",
|
||||
Str2: "hello",
|
||||
Fn: func(data interface{}) {
|
||||
failure = true
|
||||
},
|
||||
})
|
||||
fm.Notify(Generic{Str1: "hello"}, true)
|
||||
fm.Stop()
|
||||
|
||||
if !success {
|
||||
t.Error("expected 'hello' to be posted")
|
||||
}
|
||||
|
||||
if failure {
|
||||
t.Error("hello1 was triggered")
|
||||
}
|
||||
}
|
|
@ -0,0 +1,22 @@
|
|||
package filter
|
||||
|
||||
type Generic struct {
|
||||
Str1, Str2, Str3 string
|
||||
|
||||
Fn func(data interface{})
|
||||
}
|
||||
|
||||
func (self Generic) Compare(f Filter) bool {
|
||||
filter := f.(Generic)
|
||||
if (len(self.Str1) == 0 || filter.Str1 == self.Str1) &&
|
||||
(len(self.Str2) == 0 || filter.Str2 == self.Str2) &&
|
||||
(len(self.Str3) == 0 || filter.Str3 == self.Str3) {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func (self Generic) Trigger(data interface{}) {
|
||||
self.Fn(data)
|
||||
}
|
Loading…
Reference in New Issue