forked from cerc-io/plugeth
Moved filter to events
This commit is contained in:
parent
9e1689df9c
commit
ed1538248f
70
event/filter/filter.go
Normal file
70
event/filter/filter.go
Normal file
@ -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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
34
event/filter/filter_test.go
Normal file
34
event/filter/filter_test.go
Normal file
@ -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")
|
||||||
|
}
|
||||||
|
}
|
22
event/filter/generic_filter.go
Normal file
22
event/filter/generic_filter.go
Normal file
@ -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
Block a user