Skip to content

Commit

Permalink
Moved filter to events
Browse files Browse the repository at this point in the history
  • Loading branch information
obscuren committed Dec 12, 2014
1 parent 9e1689d commit ed15382
Show file tree
Hide file tree
Showing 3 changed files with 126 additions and 0 deletions.
70 changes: 70 additions & 0 deletions event/filter/filter.go
Original file line number Diff line number Diff line change
@@ -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 changes: 34 additions & 0 deletions event/filter/filter_test.go
Original file line number Diff line number Diff line change
@@ -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 changes: 22 additions & 0 deletions event/filter/generic_filter.go
Original file line number Diff line number Diff line change
@@ -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)
}

0 comments on commit ed15382

Please sign in to comment.